mirror of
https://github.com/cnlimiter/codex-register.git
synced 2026-08-11 07:33:43 +08:00
feat(config): 合并上传配置并修复debug模式下数据库初始化提示错误
This commit is contained in:
@@ -647,4 +647,68 @@ def delete_sub2api_service(db: Session, service_id: int) -> bool:
|
||||
return False
|
||||
db.delete(svc)
|
||||
db.commit()
|
||||
return True
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# Team Manager 服务 CRUD
|
||||
# ============================================================================
|
||||
|
||||
def create_tm_service(
|
||||
db: Session,
|
||||
name: str,
|
||||
api_url: str,
|
||||
api_key: str,
|
||||
enabled: bool = True,
|
||||
priority: int = 0,
|
||||
):
|
||||
"""创建 Team Manager 服务配置"""
|
||||
from .models import TeamManagerService
|
||||
svc = TeamManagerService(
|
||||
name=name,
|
||||
api_url=api_url,
|
||||
api_key=api_key,
|
||||
enabled=enabled,
|
||||
priority=priority,
|
||||
)
|
||||
db.add(svc)
|
||||
db.commit()
|
||||
db.refresh(svc)
|
||||
return svc
|
||||
|
||||
|
||||
def get_tm_service_by_id(db: Session, service_id: int):
|
||||
"""按 ID 获取 Team Manager 服务"""
|
||||
from .models import TeamManagerService
|
||||
return db.query(TeamManagerService).filter(TeamManagerService.id == service_id).first()
|
||||
|
||||
|
||||
def get_tm_services(db: Session, enabled=None):
|
||||
"""获取 Team Manager 服务列表"""
|
||||
from .models import TeamManagerService
|
||||
q = db.query(TeamManagerService)
|
||||
if enabled is not None:
|
||||
q = q.filter(TeamManagerService.enabled == enabled)
|
||||
return q.order_by(TeamManagerService.priority.asc(), TeamManagerService.id.asc()).all()
|
||||
|
||||
|
||||
def update_tm_service(db: Session, service_id: int, **kwargs):
|
||||
"""更新 Team Manager 服务配置"""
|
||||
svc = get_tm_service_by_id(db, service_id)
|
||||
if not svc:
|
||||
return None
|
||||
for k, v in kwargs.items():
|
||||
setattr(svc, k, v)
|
||||
db.commit()
|
||||
db.refresh(svc)
|
||||
return svc
|
||||
|
||||
|
||||
def delete_tm_service(db: Session, service_id: int) -> bool:
|
||||
"""删除 Team Manager 服务配置"""
|
||||
svc = get_tm_service_by_id(db, service_id)
|
||||
if not svc:
|
||||
return False
|
||||
db.delete(svc)
|
||||
db.commit()
|
||||
return True
|
||||
@@ -158,6 +158,20 @@ class Sub2ApiService(Base):
|
||||
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
|
||||
class TeamManagerService(Base):
|
||||
"""Team Manager 服务配置表"""
|
||||
__tablename__ = 'tm_services'
|
||||
|
||||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||||
name = Column(String(100), nullable=False) # 服务名称
|
||||
api_url = Column(String(500), nullable=False) # API URL
|
||||
api_key = Column(Text, nullable=False) # X-API-Key
|
||||
enabled = Column(Boolean, default=True)
|
||||
priority = Column(Integer, default=0) # 优先级
|
||||
created_at = Column(DateTime, default=datetime.utcnow)
|
||||
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
|
||||
class Proxy(Base):
|
||||
"""代理列表表"""
|
||||
__tablename__ = 'proxies'
|
||||
|
||||
Reference in New Issue
Block a user