From 4d81dea41c94c2e71d0f6a957ff54b09c75af508 Mon Sep 17 00:00:00 2001 From: Joscha Date: Fri, 14 Apr 2017 08:33:11 +0000 Subject: [PATCH] Add command line parameter support to server --- server.py | 26 ++++++++++++++++++++++++-- 1 file changed, 24 insertions(+), 2 deletions(-) diff --git a/server.py b/server.py index d1268c8..b1c3a5c 100644 --- a/server.py +++ b/server.py @@ -1,5 +1,6 @@ # import from chunks, dbchunkpool import json +import sys import threading from SimpleWebSocketServer import SimpleWebSocketServer, WebSocket @@ -82,5 +83,26 @@ class WotServer(WebSocket): while clients and not clients[-1]: clients.pop() -server = SimpleWebSocketServer('', 8000, WotServer) -server.serveforever() +def main(argv): + if len(argv) > 2: + print("Usage:") + print(f" {argv[0]} [port]") + print(" default port: 8000") + return + elif len(argv) > 1: + try: + port = int(argv[1]) + except ValueError: + print("Invalid port") + return + else: + port = 8000 + + server = SimpleWebSocketServer('', port, WotServer) + try: + server.serveforever() + except KeyboardInterrupt: + print("Stopped.") + +if __name__ == "__main__": + main(sys.argv)