more debug options

This commit is contained in:
Bui 2022-03-24 16:09:01 +00:00
parent 7f511abb4f
commit 584aee13da
2 changed files with 11 additions and 2 deletions

10
app.py
View File

@ -1,5 +1,6 @@
#!/usr/bin/env python3
import argparse
import base64
import bcrypt
import hashlib
@ -23,6 +24,12 @@ from ffmpy import FFmpeg
from pymongo import MongoClient
from redis import Redis
parser = argparse.ArgumentParser(description='Run the taiko-web development server.')
parser.add_argument('port', type=int, metavar='PORT', nargs='?', default=34801, help='Port to listen on.')
parser.add_argument('-b', '--bind-address', default='localhost', help='Bind server to address.')
parser.add_argument('-d', '--debug', action='store_true', help='Enable debug mode.')
args = parser.parse_args()
def take_config(name, required=False):
if hasattr(config, name):
return getattr(config, name)
@ -717,4 +724,5 @@ if __name__ == '__main__':
def send_assets(path):
return send_from_directory('public/assets', path)
app.run(port=34801)
app.run(host=args.bind_address, port=args.port, debug=args.debug)

View File

@ -9,6 +9,7 @@ import sys
parser = argparse.ArgumentParser(description='Run the taiko-web multiplayer server.')
parser.add_argument('port', type=int, metavar='PORT', nargs='?', default=34802, help='Port to listen on.')
parser.add_argument('-b', '--bind-address', default='localhost', help='Bind server to address.')
parser.add_argument('-o', '--allow-origin', action='append', help='Limit incoming connections to the specified origin. Can be specified multiple times.')
args = parser.parse_args()
@ -382,7 +383,7 @@ port = args.port
print('Starting server on port %d' % port)
loop = asyncio.get_event_loop()
tasks = asyncio.gather(
websockets.serve(connection, "localhost", port, origins=args.allow_origin)
websockets.serve(connection, args.bind_address, port, origins=args.allow_origin)
)
try:
loop.run_until_complete(tasks)