From 71599e9c6d74674bc67305b5204a329d5f04727e Mon Sep 17 00:00:00 2001 From: Sanster Date: Mon, 18 Apr 2022 15:29:29 +0800 Subject: [PATCH] filter flaskwebgui log --- lama_cleaner/server.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py index 50f4150..1aa00f9 100644 --- a/lama_cleaner/server.py +++ b/lama_cleaner/server.py @@ -48,15 +48,15 @@ if os.environ.get("CACHE_DIR"): BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "app/build") -class InterceptHandler(logging.Handler): - def emit(self, record): - logger_opt = logger.opt(depth=6, exception=record.exc_info) - logger_opt.log(record.levelno, record.getMessage()) +class NoFlaskwebgui(logging.Filter): + def filter(self, record): + return 'GET //flaskwebgui-keep-server-alive' not in record.getMessage() +logging.getLogger("werkzeug").addFilter(NoFlaskwebgui()) + app = Flask(__name__, static_folder=os.path.join(BUILD_DIR, "static")) app.config["JSON_AS_ASCII"] = False -app.logger.addHandler(InterceptHandler()) CORS(app, expose_headers=["Content-Disposition"]) model: ModelManager = None