28 lines
931 B
Python
28 lines
931 B
Python
from typing import Optional
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from ..models import AdminSetting
|
|
from ..repositories.admin_setting_repository import AdminSettingRepository
|
|
|
|
|
|
class AdminSettingService:
|
|
"""管理员配置项服务，提供简单的 KV 操作。"""
|
|
|
|
def __init__(self, session: AsyncSession):
|
|
self.session = session
|
|
self.repo = AdminSettingRepository(session)
|
|
|
|
async def get(self, key: str, default: Optional[str] = None) -> Optional[str]:
|
|
value = await self.repo.get_value(key)
|
|
return value if value is not None else default
|
|
|
|
async def set(self, key: str, value: str) -> None:
|
|
record = await self.repo.get(key=key)
|
|
if record:
|
|
await self.repo.update_fields(record, value=value)
|
|
else:
|
|
setting = AdminSetting(key=key, value=value)
|
|
await self.repo.add(setting)
|
|
await self.session.commit()
|