Spaces:
Running
Running
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) |