kqjy/MyFSIO
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 13 Wiki Activity

Cleanup setup

Browse Source
This commit is contained in:
kqjy
2025-11-22 17:41:23 +08:00
parent c8b1c33118
commit 015c9cb52d
2 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docker-entrypoint.sh
View File

@@ -1,8 +1,5 @@
#!/bin/sh
set -e
# Start API server in background
waitress-serve --ident=MyFSIO --listen=*:5000 --call app:create_api_app &
# Start UI server in foreground
waitress-serve --ident=MyFSIO --listen=*:5100 --call app:create_ui_app
# Run both services using the python runner in production mode
exec python run.py --prod

33
run.py
View File

@@ -18,20 +18,28 @@ def _is_debug_enabled() -> bool:
return os.getenv("FLASK_DEBUG", "0").lower() in ("1", "true", "yes")
def serve_api(port: int) -> None:
def serve_api(port: int, prod: bool = False) -> None:
app = create_api_app()
debug = _is_debug_enabled()
if debug:
warnings.warn("DEBUG MODE ENABLED - DO NOT USE IN PRODUCTION", RuntimeWarning)
app.run(host=_server_host(), port=port, debug=debug)
if prod:
from waitress import serve
serve(app, host=_server_host(), port=port, ident="MyFSIO")
else:
debug = _is_debug_enabled()
if debug:
warnings.warn("DEBUG MODE ENABLED - DO NOT USE IN PRODUCTION", RuntimeWarning)
app.run(host=_server_host(), port=port, debug=debug)
def serve_ui(port: int) -> None:
def serve_ui(port: int, prod: bool = False) -> None:
app = create_ui_app()
debug = _is_debug_enabled()
if debug:
warnings.warn("DEBUG MODE ENABLED - DO NOT USE IN PRODUCTION", RuntimeWarning)
app.run(host=_server_host(), port=port, debug=debug)
if prod:
from waitress import serve
serve(app, host=_server_host(), port=port, ident="MyFSIO")
else:
debug = _is_debug_enabled()
if debug:
warnings.warn("DEBUG MODE ENABLED - DO NOT USE IN PRODUCTION", RuntimeWarning)
app.run(host=_server_host(), port=port, debug=debug)
if __name__ == "__main__":
@@ -39,18 +47,19 @@ if __name__ == "__main__":
parser.add_argument("--mode", choices=["api", "ui", "both"], default="both")
parser.add_argument("--api-port", type=int, default=5000)
parser.add_argument("--ui-port", type=int, default=5100)
parser.add_argument("--prod", action="store_true", help="Run in production mode using Waitress")
args = parser.parse_args()
if args.mode in {"api", "both"}:
print(f"Starting API server on port {args.api_port}...")
api_proc = Process(target=serve_api, args=(args.api_port,), daemon=True)
api_proc = Process(target=serve_api, args=(args.api_port, args.prod), daemon=True)
api_proc.start()
else:
api_proc = None
if args.mode in {"ui", "both"}:
print(f"Starting UI server on port {args.ui_port}...")
serve_ui(args.ui_port)
serve_ui(args.ui_port, args.prod)
elif api_proc:
try:
api_proc.join()