osm2city-scripts/set-status.py

#! /usr/bin/python3
# Copyright (C) 2018-2020 Merspieler, merspieler _at_ airmail.cc
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

import socket
import re
import sys
from common import send_status

host = socket.gethostname()
port = 12345
api = None
apt_token = None
tile = ""
status = ""

argc = len(sys.argv)
first = 1
i = 1
while i < argc:
	if sys.argv[i] == "--port":
		i += 1
		port = int(sys.argv[i])
	elif sys.argv[i] == "--host":
		i += 1
		host = sys.argv[i]
	elif sys.argv[i] == "-a" or sys.argv[i] == "--api":
		i += 1
		api = sys.argv[i]
	elif sys.argv[i] == "-t" or sys.argv[i] == "--api-token":
		i += 1
		api_token = sys.argv[i]
	elif sys.argv[i] == "-h" or sys.argv[i] == "--help":
		print("usage: set-status.py <tile> <status> [OPTIONS]")
		print("Manually set a tile status")
		print("")
		print("  <tile>            Is the tile name")
		print("  <status>          Is the new status to set the tile to")
		print("                    Can be:")
		print("                     - pending")
		print("                     - skip")
		print("                     - rebuild")
		print("                     - started")
		print("                     - done")
		print("                     - packaged")
		print("OPTIONS")
		print("    , --host        Manager host")
		print("    , --port        Manager port")
		print("  -a, --api         Manager api url")
		print("  -t, --api-token   Manager api token")
		print("  -h, --help        Shows this help and exit")
		sys.exit(0)
	else:
		if first == 1:
			match = re.match(r"([ew]\d{3}[ns]\d{2}|[0-9]{1,7})", sys.argv[i])
			if match == None:
				print("ERROR: Invalid tile name " + sys.argv[i])
				sys.exit(1)
			first = 2
			tile = match.group(0)
		elif first == 2:
			match = re.match(r"(done|pending|rebuild|skip|packaged|started)", sys.argv[i])
			if match == None:
				print("ERROR: Invalid status " + sys.argv[i])
				sys.exit(1)
			first == 0
			status = match.group(0)
		else:
			print("Unknown option " + sys.argv[i])
			sys.exit(1)
	i += 1

if tile == "":
	print("ERROR: No tile given")
	sys.exit(1)

if status == "":
	print("ERROR: No status given")
	sys.exit(1)

if api != None and api_token == None:
	print("Error: API given but no token")
	sys.exit(1)

if api == None:
	send_status(tile, status, host, port)
else:
	api_send_status(tile, status, api, api_token)

print("Status set successfully")