From 7a382a35b47ccb3fa25bdf75b30c0e0d3b307a0c Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Thu, 29 May 2025 02:44:30 +0800 Subject: [PATCH] feat(api): simplify gevent setup logic --- api/app.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/api/app.py b/api/app.py index 536a9a17b9..a8b7c95fd2 100644 --- a/api/app.py +++ b/api/app.py @@ -9,12 +9,17 @@ def _setup_gevent(): It should be """ + _FALSE_VALUE_FOR_ENV = frozenset(["false", "0", "no"]) # It seems that JetBrains Python debugger does not work well with gevent, # so we need to disable gevent in debug mode. # If you are using debugpy and set GEVENT_SUPPORT=True, you can debug with gevent. - if (flask_debug := os.environ.get("FLASK_DEBUG", "0")) and flask_debug.lower() not in {"false", "0", "no"}: + flask_debug_env = os.environ.get("FLASK_DEBUG", "0").lower() + if flask_debug_env not in _FALSE_VALUE_FOR_ENV: + print("Flask Debug enabled.", flush=True) return - if os.environ.get("GEVENT_SUPPORT", "0") == "0": + gevent_support_env = os.environ.get("GEVENT_SUPPORT", "false").lower() + if gevent_support_env in _FALSE_VALUE_FOR_ENV: + print("Gevent disabled.", flush=True) return from gevent import monkey @@ -30,6 +35,7 @@ def _setup_gevent(): import psycogreen.gevent # type: ignore psycogreen.gevent.patch_psycopg() + print("Gevnet and psycopg patched", flush=True) _setup_gevent()