mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-07 08:43:13 +08:00
169488851f
主要变更：
1. **数据库集成**:
* 引入 MySQL 数据库支持，使用 SQLAlchemy 和 `databases` 库持久化存储应用程序设置。
* 添加了 `app/database` 目录，包含数据库连接、模型和初始化逻辑。
* 更新 `requirements.txt` 添加数据库相关依赖 (`pymysql`, `sqlalchemy`, `aiomysql`, `databases`, `python-dotenv`)。
2. **配置管理重构**:
* 重构 `ConfigService` (`app/service/config/config_service.py`)，使其从数据库加载和保存设置，并支持从 `.env` 文件同步初始配置到数据库。
* 修改 `Settings` 模型 (`app/config/config.py`) 以包含数据库连接信息，并添加了从数据库加载/同步配置的逻辑。
* 配置相关的路由 (`app/router/config_routes.py`) 更新为异步，并调用新的 `ConfigService` 方法。
* `KeyManager` (`app/service/key/key_manager.py`) 现在可以在配置更新后重置和重新初始化。
3. **错误日志查看器**:
* 新增 `/logs` 页面 (`app/templates/error_logs.html`) 用于展示应用程序错误日志。
* 添加了相应的路由 (`app/router/log_routes.py`)、静态资源 (`app/static/css/error_logs.css`, `app/static/js/error_logs.js`) 和日志记录器 (`app/log/logger.py`)。
* 在配置页面和密钥管理页面的导航栏中添加了指向日志页面的链接。
4. **异步操作**:
* 将配置服务和相关路由转换为异步 (`async def`) 以支持异步数据库操作。
5. **其他**:
* 更新了应用程序初始化逻辑 (`app/core/application.py`, `app/core/initialization.py`) 以包含数据库连接的建立和关闭。
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
"""
|
|
数据库初始化模块
|
|
"""
|
|
from dotenv import dotenv_values
|
|
|
|
from sqlalchemy import inspect
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.database.connection import engine, Base
|
|
from app.database.models import Settings
|
|
from app.log.logger import get_database_logger
|
|
|
|
logger = get_database_logger()
|
|
|
|
|
|
def create_tables():
|
|
"""
|
|
创建数据库表
|
|
"""
|
|
try:
|
|
# 创建所有表
|
|
Base.metadata.create_all(engine)
|
|
logger.info("Database tables created successfully")
|
|
except Exception as e:
|
|
logger.error(f"Failed to create database tables: {str(e)}")
|
|
raise
|
|
|
|
|
|
def import_env_to_settings():
|
|
"""
|
|
将.env文件中的配置项导入到t_settings表中
|
|
"""
|
|
try:
|
|
# 获取.env文件中的所有配置项
|
|
env_values = dotenv_values(".env")
|
|
|
|
# 获取检查器
|
|
inspector = inspect(engine)
|
|
|
|
# 检查t_settings表是否存在
|
|
if "t_settings" in inspector.get_table_names():
|
|
# 使用Session进行数据库操作
|
|
with Session(engine) as session:
|
|
# 获取所有现有的配置项
|
|
current_settings = {setting.key: setting for setting in session.query(Settings).all()}
|
|
|
|
# 遍历所有配置项
|
|
for key, value in env_values.items():
|
|
# 检查配置项是否已存在
|
|
if key not in current_settings:
|
|
# 插入配置项
|
|
new_setting = Settings(key=key, value=value)
|
|
session.add(new_setting)
|
|
logger.info(f"Inserted setting: {key}")
|
|
|
|
# 提交事务
|
|
session.commit()
|
|
|
|
logger.info("Environment variables imported to settings table successfully")
|
|
except Exception as e:
|
|
logger.error(f"Failed to import environment variables to settings table: {str(e)}")
|
|
raise
|
|
|
|
|
|
def initialize_database():
|
|
"""
|
|
初始化数据库
|
|
"""
|
|
try:
|
|
# 创建表
|
|
create_tables()
|
|
|
|
# 导入环境变量
|
|
import_env_to_settings()
|
|
except Exception as e:
|
|
logger.error(f"Failed to initialize database: {str(e)}")
|
|
raise
|