feat(system):优化后端
1.新增后端测试 2.增加了后端的加密 3.增加了i18n(国际化)
This commit is contained in:
@@ -17,10 +17,37 @@ from typing import List
|
||||
from kilostar.core.postgres_database.model.provider import ProviderModel
|
||||
from sqlalchemy import select
|
||||
from kilostar.core.postgres_database.database_exception import database_exception
|
||||
from kilostar.utils.crypto import (
|
||||
CryptoError,
|
||||
decrypt_secret,
|
||||
encrypt_secret,
|
||||
is_encrypted,
|
||||
)
|
||||
from kilostar.utils.logger import get_logger
|
||||
|
||||
logger = get_logger("provider_dao")
|
||||
|
||||
|
||||
def _decrypt_apikey(value):
|
||||
if not value:
|
||||
return value
|
||||
if not is_encrypted(value):
|
||||
return value
|
||||
try:
|
||||
return decrypt_secret(value)
|
||||
except CryptoError as e:
|
||||
logger.error(f"Provider apikey 解密失败: {e}")
|
||||
return value
|
||||
|
||||
|
||||
def _encrypt_apikey(value):
|
||||
if not value or is_encrypted(value):
|
||||
return value
|
||||
return encrypt_secret(value)
|
||||
|
||||
|
||||
class ProviderDatabase:
|
||||
"""Provider 表的 DAO:模型 Provider 的增删查改。"""
|
||||
"""Provider 表的 DAO:模型 Provider 的增删查改;``provider_apikey`` 透明 Fernet 加解密。"""
|
||||
|
||||
def __init__(self, async_session_maker):
|
||||
self.async_session_maker = async_session_maker
|
||||
@@ -37,11 +64,10 @@ class ProviderDatabase:
|
||||
provider_id=provider.provider_id,
|
||||
provider_title=provider.provider_title,
|
||||
provider_url=provider.provider_url,
|
||||
provider_apikey=provider.provider_apikey,
|
||||
provider_apikey=_decrypt_apikey(provider.provider_apikey),
|
||||
provider_models=provider.provider_models,
|
||||
provider_type=provider.provider_type,
|
||||
provider_owner=provider.provider_owner,
|
||||
provider_status=provider.provider_status,
|
||||
is_active=provider.is_active,
|
||||
)
|
||||
for provider in results
|
||||
@@ -50,7 +76,9 @@ class ProviderDatabase:
|
||||
|
||||
@database_exception
|
||||
async def add_provider(self, **kwargs) -> None:
|
||||
"""新建一条 Provider 记录;字段通过 kwargs 直接传给 ProviderModel。"""
|
||||
"""新建一条 Provider 记录;``provider_apikey`` 写入前自动加密。"""
|
||||
if "provider_apikey" in kwargs:
|
||||
kwargs["provider_apikey"] = _encrypt_apikey(kwargs["provider_apikey"])
|
||||
async with self.async_session_maker() as session:
|
||||
provider = ProviderModel(**kwargs)
|
||||
session.add(provider)
|
||||
@@ -67,7 +95,9 @@ class ProviderDatabase:
|
||||
|
||||
@database_exception
|
||||
async def update_provider(self, provider_id: str, **kwargs) -> None:
|
||||
"""部分更新指定 Provider 的字段;不存在时返回 None,否则返回刷新后的对象。"""
|
||||
"""部分更新指定 Provider 的字段;``provider_apikey`` 写入前自动加密。"""
|
||||
if "provider_apikey" in kwargs:
|
||||
kwargs["provider_apikey"] = _encrypt_apikey(kwargs["provider_apikey"])
|
||||
async with self.async_session_maker() as session:
|
||||
provider = await session.get(ProviderModel, provider_id)
|
||||
if provider is not None:
|
||||
@@ -76,5 +106,7 @@ class ProviderDatabase:
|
||||
session.add(provider)
|
||||
await session.commit()
|
||||
await session.refresh(provider)
|
||||
# 解密返回,方便上游使用
|
||||
provider.provider_apikey = _decrypt_apikey(provider.provider_apikey)
|
||||
return provider
|
||||
return None
|
||||
|
||||
Reference in New Issue
Block a user