add banbanmini backend
This commit is contained in:
111
talkingq-url/services/system_config_manager.py
Normal file
111
talkingq-url/services/system_config_manager.py
Normal file
@@ -0,0 +1,111 @@
|
||||
import asyncio
|
||||
from typing import Optional
|
||||
import time
|
||||
from sqlalchemy import select, update, insert, delete, func
|
||||
from database.models import SystemConfig
|
||||
from services.database_service_base import DatabaseServiceBase
|
||||
from utils.logger import session_logger
|
||||
|
||||
class SystemConfigManager(DatabaseServiceBase):
|
||||
def __init__(self):
|
||||
super().__init__(service_name="system_config")
|
||||
|
||||
async def create_config(self, key: str, value: str) -> bool:
|
||||
"""新增系统配置"""
|
||||
await self._init_database()
|
||||
db_session = await self.get_session()
|
||||
try:
|
||||
insert_stmt = insert(SystemConfig).values(
|
||||
config_key=key,
|
||||
config_value=value
|
||||
)
|
||||
await db_session.execute(insert_stmt)
|
||||
await db_session.commit()
|
||||
session_logger.info("system", "system_config", f"新增系统配置: {key}={value}")
|
||||
return True
|
||||
except Exception as e:
|
||||
await db_session.rollback()
|
||||
session_logger.error("system", "system_config", f"新增系统配置失败: {str(e)}")
|
||||
return False
|
||||
finally:
|
||||
await db_session.close()
|
||||
|
||||
async def get_config(self, key: str) -> Optional[SystemConfig]:
|
||||
"""获取系统配置"""
|
||||
await self._init_database()
|
||||
db_session = await self.get_session()
|
||||
try:
|
||||
query = select(SystemConfig).where(SystemConfig.config_key == key)
|
||||
result = await db_session.execute(query)
|
||||
return result.scalars().first()
|
||||
except Exception as e:
|
||||
session_logger.error(key, "system_config", f"查询系统配置失败: {str(e)}")
|
||||
return None
|
||||
finally:
|
||||
await db_session.close()
|
||||
|
||||
async def update_config(self, key: str, value: str) -> bool:
|
||||
"""更新系统配置"""
|
||||
await self._init_database()
|
||||
db_session = await self.get_session()
|
||||
try:
|
||||
query = select(SystemConfig).where(SystemConfig.config_key == key)
|
||||
result = await db_session.execute(query)
|
||||
config = result.scalars().first()
|
||||
|
||||
if config:
|
||||
update_stmt = (
|
||||
update(SystemConfig)
|
||||
.where(SystemConfig.config_key == key)
|
||||
.values(config_value=value, updated_at=func.now())
|
||||
)
|
||||
await db_session.execute(update_stmt)
|
||||
else:
|
||||
insert_stmt = insert(SystemConfig).values(
|
||||
config_key=key,
|
||||
config_value=value
|
||||
)
|
||||
await db_session.execute(insert_stmt)
|
||||
|
||||
await db_session.commit()
|
||||
session_logger.info(key, "system_config", f"系统配置已更新: {value}")
|
||||
return True
|
||||
except Exception as e:
|
||||
await db_session.rollback()
|
||||
session_logger.error(key, "system_config", f"更新系统配置失败: {str(e)}")
|
||||
return False
|
||||
finally:
|
||||
await db_session.close()
|
||||
|
||||
async def delete_config(self, key: str) -> bool:
|
||||
"""删除系统配置"""
|
||||
await self._init_database()
|
||||
db_session = await self.get_session()
|
||||
try:
|
||||
delete_stmt = delete(SystemConfig).where(SystemConfig.config_key == key)
|
||||
await db_session.execute(delete_stmt)
|
||||
await db_session.commit()
|
||||
session_logger.info(key, "system_config", "系统配置已删除")
|
||||
return True
|
||||
except Exception as e:
|
||||
await db_session.rollback()
|
||||
session_logger.error(key, "system_config", f"删除系统配置失败: {str(e)}")
|
||||
return False
|
||||
finally:
|
||||
await db_session.close()
|
||||
|
||||
async def list_configs(self):
|
||||
"""列出所有系统配置"""
|
||||
await self._init_database()
|
||||
db_session = await self.get_session()
|
||||
try:
|
||||
query = select(SystemConfig)
|
||||
result = await db_session.execute(query)
|
||||
return result.scalars().all()
|
||||
except Exception as e:
|
||||
session_logger.error("system_config", "system_config", f"查询所有系统配置失败: {str(e)}")
|
||||
return []
|
||||
finally:
|
||||
await db_session.close()
|
||||
|
||||
system_config_manager = SystemConfigManager()
|
||||
Reference in New Issue
Block a user