diff --git a/server.py b/server.py index 32ad722..7cc858d 100644 --- a/server.py +++ b/server.py @@ -1,4 +1,5 @@ from argparse import ArgumentParser +import logging from pathlib import Path import urllib.error import urllib.request @@ -8,10 +9,12 @@ from flask import Flask, send_file def main(): parser = ArgumentParser() + parser.add_argument('--host', default='0.0.0.0', help='Server host (default: 0.0.0.0)') parser.add_argument('--port', type=int, default=8000, help='Server port (default: 8000)') parser.add_argument('--debug', action='store_true', help='Run in debug mode (auto-reload on change)') arguments = parser.parse_args() + host: str = arguments.host port: int = arguments.port debug_mode: bool = arguments.debug del arguments @@ -19,6 +22,9 @@ def main(): app = Flask('map_server', static_folder='public', static_url_path='/') osm_timeout = 3 http_code_ok = 200 + if not debug_mode: + log = logging.getLogger('werkzeug') + log.setLevel(logging.WARNING) @app.route('/') def _index(): @@ -46,7 +52,7 @@ def main(): except urllib.error.URLError as error: return f'Cannot retrieve tile: {error}', 501 - app.run(host='0.0.0.0', port=port, debug=debug_mode) + app.run(host=host, port=port, debug=debug_mode) if __name__ == '__main__':