Custom-Gemini / debug.py
Moonfanz's picture
Upload debug.py
d92a175 verified
import logging
from flask import Flask
from werkzeug.middleware.proxy_fix import ProxyFix
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
app = Flask(__name__)
app.wsgi_app = ProxyFix(app.wsgi_app)
# 创建自定义的日志过滤器
class StaticFilter(logging.Filter):
def filter(self, record):
# 如果是静态资源的请求并且状态码是304,则不记录
return not (
('/static/' in record.getMessage() and '304' in record.getMessage()) or
('.css' in record.getMessage() and '304' in record.getMessage()) or
('.js' in record.getMessage() and '304' in record.getMessage())
)
# 配置werkzeug的日志
logging.getLogger('werkzeug').addFilter(StaticFilter())
# 设置基本的日志格式
logging.basicConfig(
level=logging.INFO,
format='%(name)s:%(message)s'
)
def log_prompt(messages, preset_name):
logger.info(f"Preset: {preset_name}")
for message in messages:
logger.info(message)
def log_message(message):
logger.info(message)