From 376a7de23e260b1999dc3a4f97f17397c9d7b917 Mon Sep 17 00:00:00 2001 From: Muhamad Aji Wibisono Date: Tue, 10 Jun 2025 21:27:28 +0700 Subject: [PATCH] feat: added proxy as an argument --- surfsense_backend/main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/surfsense_backend/main.py b/surfsense_backend/main.py index 81ef520..8d3f356 100644 --- a/surfsense_backend/main.py +++ b/surfsense_backend/main.py @@ -11,6 +11,7 @@ logging.basicConfig( if __name__ == "__main__": parser = argparse.ArgumentParser(description='Run the SurfSense application') parser.add_argument('--reload', action='store_true', help='Enable hot reloading') + parser.add_argument('--proxy', nargs='?', const='0.0.0.0/0', default=None, help='Enable proxy headers with allowed IPs (e.g., --proxy 0.0.0.0/0)') args = parser.parse_args() uvicorn.run( @@ -18,5 +19,7 @@ if __name__ == "__main__": host="0.0.0.0", log_level="info", reload=args.reload, - reload_dirs=["app"] + reload_dirs=["app"], + proxy_headers=bool(args.proxy), + forwarded_allow_ips=args.proxy if args.proxy else "", )