osm2city-scripts/worldbuild-manager.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 sys
import os
import re
import pymysql

from common import norm

host = socket.gethostname()
port = 12345
verbose = False
dbuser = ""
dbpw = ""

argc = len(sys.argv)
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] == "-u" or sys.argv[i] == "--user":
		i += 1
		dbuser = sys.argv[i]
	elif sys.argv[i] == "-p" or sys.argv[i] == "--password":
		i += 1
		dbpw = sys.argv[i]
	elif sys.argv[i] == "-v":
		verbose = True
	elif sys.argv[i] == "-h" or sys.argv[i] == "--help":
		print("usage: worldbuild-manager.py [OPTIONS]")
		print("Handles job asignments and keeps track of the status")
		print("")
		print("OPTIONS")
		print("  -u, --user        Database user")
		print("  -p, --password    Database password")
		print("  -v, --verbose     Verbose printouts")
		print("    , --host        hostname or interface to bind to")
		print("    , --port        Port to bind to")
		print("  -h, --help        Shows this help and exit")
		sys.exit(0)
	else:
		print("Unknown option " + sys.argv[i])
		sys.exit(1)
	i += 1

if dbuser == "":
	print("ERROR: No database user given")
	sys.exit(1)

if dbpw == "":
	print("ERROR: No database password given")
	sys.exit(1)

db = pymysql.connect("localhost", dbuser, dbpw, "worldbuild")

cursor = db.cursor()

def set_state(name, status):
	if verbose:
		print(name + " set status to " + status)

	match = re.match(r"^[ew]\d{3}[ns]\d{2}|[0-9]{1,7}$", name)
	if match == None:
		print("WARNING: Recived status " + status + " from invalid tile '" + name + "'")
	else:
		if status != "started" and status != "done" and status != "rebuild" and status != "skip" and status != "pending" and status != "packaged":
			print("WARNING: Invalid status '" + status + "' recived from " + name)
		else:
			match = re.match(r"[ew]\d{3}[ns]\d{2}", name)
			if match != None:
				sql ="SELECT id FROM secondLevel WHERE name = '" + name + "'"
				cursor.execute(sql)
				pid = cursor.fetchone()
				sql = "SELECT id FROM status WHERE name = '" + status + "'"
				cursor.execute(sql)
				sid = cursor.fetchone()
				sql = "UPDATE tile SET status_id = " + str(sid[0]) + " WHERE parent_id = " + str(pid[0])
				cursor.execute(sql)
			else:
				sql = "UPDATE tile SET status_id = (SELECT id FROM status WHERE name = '" + status + "') WHERE id = " + name
				cursor.execute(sql)
			db.commit()

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind((host, port))

sock.listen(5)
print("Up and running...")
try:
	while True:
		try:
			c, addr = sock.accept()
			msg = c.recv(128)
			msg = msg.decode()

			db.ping(reconnect=True)

			match = re.match(r"(set) ([ew]\d{3}[ns]\d{2}|[0-9]{1,7}) (done|started|rebuild|skip|pending|packaged)", msg)
			if match != None:
				action = "set"
				name = match.group(2)
				status = match.group(3)
				set_state(name, status)
			else:
				match = re.match(r"(get) (started|rebuild|skip|pending|done)", msg)
				if match != None:
					if match.group(2) == "done":
						action = "get-done"
					else:
						action = "get-job"
						get = match.group(2)
				else:
					match = re.match(r"(status) ([ew]\d{3}[ns]\d{2}|[0-9]{1,7})", msg)
					if match != None:
						action = "status"
						name = match.group(2)
					else:
						print("WARNING: Recived invalid package")

			if action == "get-job":
				additional_type = ""
				if get != "pending":
					additional_type = "OR status_id = (SELECT id FROM status WHERE name = '" + get + "')"
				sql = "SELECT id FROM tile WHERE status_id = (SELECT id FROM status WHERE name = 'pending') " + additional_type + " ORDER BY parent_id LIMIT 1"
				cursor.execute(sql)
				result = cursor.fetchone()
				if result == None:
					tile = "None"
				else:
					tile = result[0]
					sql = "UPDATE tile SET status_id = (SELECT id FROM status WHERE name = 'started') WHERE id = " + str(tile)
					cursor.execute(sql)
					db.commit()


				if verbose:
					print("Asigning job on tile " + str(tile))

				c.send(str(tile).encode())
			elif action == "get-done":
				sql = "SELECT name FROM secondLevel WHERE status_id = (SELECT id FROM status WHERE name = 'done') LIMIT 1"
				cursor.execute(sql)
				result = cursor.fetchone()
				if result == None:
					tile = "None"
				else:
					tile = result[0]

				c.send(str(tile).encode())
			elif action == "status":
				match = re.match(r"([ew])(\d{3})([ns])(\d{2})", name)
				if match != None:
					sql = "SELECT name FROM status WHERE id = (SELECT status_id FROM secondLevel WHERE name = '" + name + "')"
					cursor.execute(sql)
					result = cursor.fetchone()

					c.send(str(result[0]).encode())
				else:
					sql = "SELECT name FROM status WHERE id = (SELECT status_id FROM tile WHERE id = '" + name + "')"
					cursor.execute(sql)
					result = cursor.fetchone()

					c.send(str(result[0]).encode())
			c.close()
		except IOError:
			print("WARNING: Recived invalid package")
			c.close()

except KeyboardInterrupt:
	sock.close()
	db.close()
	print("Exiting...")
	sys.exit(0)