2

src/database/crud.py

@@ -0,0 +1,372 @@
+"""
+数据库 CRUD 操作
+"""
+
+from typing import List, Optional, Dict, Any, Union
+from datetime import datetime, timedelta
+from sqlalchemy.orm import Session
+from sqlalchemy import and_, or_, desc, asc, func
+
+from .models import Account, EmailService, RegistrationTask, Setting
+
+
+# ============================================================================
+# 账户 CRUD
+# ============================================================================
+
+def create_account(
+    db: Session,
+    email: str,
+    email_service: str,
+    email_service_id: Optional[str] = None,
+    account_id: Optional[str] = None,
+    workspace_id: Optional[str] = None,
+    access_token: Optional[str] = None,
+    refresh_token: Optional[str] = None,
+    id_token: Optional[str] = None,
+    proxy_used: Optional[str] = None,
+    metadata: Optional[Dict[str, Any]] = None
+) -> Account:
+    """创建新账户"""
+    db_account = Account(
+        email=email,
+        email_service=email_service,
+        email_service_id=email_service_id,
+        account_id=account_id,
+        workspace_id=workspace_id,
+        access_token=access_token,
+        refresh_token=refresh_token,
+        id_token=id_token,
+        proxy_used=proxy_used,
+        metadata=metadata or {},
+        registered_at=datetime.utcnow()
+    )
+    db.add(db_account)
+    db.commit()
+    db.refresh(db_account)
+    return db_account
+
+
+def get_account_by_id(db: Session, account_id: int) -> Optional[Account]:
+    """根据 ID 获取账户"""
+    return db.query(Account).filter(Account.id == account_id).first()
+
+
+def get_account_by_email(db: Session, email: str) -> Optional[Account]:
+    """根据邮箱获取账户"""
+    return db.query(Account).filter(Account.email == email).first()
+
+
+def get_accounts(
+    db: Session,
+    skip: int = 0,
+    limit: int = 100,
+    email_service: Optional[str] = None,
+    status: Optional[str] = None,
+    search: Optional[str] = None
+) -> List[Account]:
+    """获取账户列表（支持分页、筛选）"""
+    query = db.query(Account)
+
+    if email_service:
+        query = query.filter(Account.email_service == email_service)
+
+    if status:
+        query = query.filter(Account.status == status)
+
+    if search:
+        search_filter = or_(
+            Account.email.ilike(f"%{search}%"),
+            Account.account_id.ilike(f"%{search}%"),
+            Account.workspace_id.ilike(f"%{search}%")
+        )
+        query = query.filter(search_filter)
+
+    query = query.order_by(desc(Account.created_at)).offset(skip).limit(limit)
+    return query.all()
+
+
+def update_account(
+    db: Session,
+    account_id: int,
+    **kwargs
+) -> Optional[Account]:
+    """更新账户信息"""
+    db_account = get_account_by_id(db, account_id)
+    if not db_account:
+        return None
+
+    for key, value in kwargs.items():
+        if hasattr(db_account, key) and value is not None:
+            setattr(db_account, key, value)
+
+    db.commit()
+    db.refresh(db_account)
+    return db_account
+
+
+def delete_account(db: Session, account_id: int) -> bool:
+    """删除账户"""
+    db_account = get_account_by_id(db, account_id)
+    if not db_account:
+        return False
+
+    db.delete(db_account)
+    db.commit()
+    return True
+
+
+def delete_accounts_batch(db: Session, account_ids: List[int]) -> int:
+    """批量删除账户"""
+    result = db.query(Account).filter(Account.id.in_(account_ids)).delete(synchronize_session=False)
+    db.commit()
+    return result
+
+
+def get_accounts_count(
+    db: Session,
+    email_service: Optional[str] = None,
+    status: Optional[str] = None
+) -> int:
+    """获取账户数量"""
+    query = db.query(func.count(Account.id))
+
+    if email_service:
+        query = query.filter(Account.email_service == email_service)
+
+    if status:
+        query = query.filter(Account.status == status)
+
+    return query.scalar()
+
+
+# ============================================================================
+# 邮箱服务 CRUD
+# ============================================================================
+
+def create_email_service(
+    db: Session,
+    service_type: str,
+    name: str,
+    config: Dict[str, Any],
+    enabled: bool = True,
+    priority: int = 0
+) -> EmailService:
+    """创建邮箱服务配置"""
+    db_service = EmailService(
+        service_type=service_type,
+        name=name,
+        config=config,
+        enabled=enabled,
+        priority=priority
+    )
+    db.add(db_service)
+    db.commit()
+    db.refresh(db_service)
+    return db_service
+
+
+def get_email_service_by_id(db: Session, service_id: int) -> Optional[EmailService]:
+    """根据 ID 获取邮箱服务"""
+    return db.query(EmailService).filter(EmailService.id == service_id).first()
+
+
+def get_email_services(
+    db: Session,
+    service_type: Optional[str] = None,
+    enabled: Optional[bool] = None,
+    skip: int = 0,
+    limit: int = 100
+) -> List[EmailService]:
+    """获取邮箱服务列表"""
+    query = db.query(EmailService)
+
+    if service_type:
+        query = query.filter(EmailService.service_type == service_type)
+
+    if enabled is not None:
+        query = query.filter(EmailService.enabled == enabled)
+
+    query = query.order_by(
+        asc(EmailService.priority),
+        desc(EmailService.last_used)
+    ).offset(skip).limit(limit)
+
+    return query.all()
+
+
+def update_email_service(
+    db: Session,
+    service_id: int,
+    **kwargs
+) -> Optional[EmailService]:
+    """更新邮箱服务配置"""
+    db_service = get_email_service_by_id(db, service_id)
+    if not db_service:
+        return None
+
+    for key, value in kwargs.items():
+        if hasattr(db_service, key) and value is not None:
+            setattr(db_service, key, value)
+
+    db.commit()
+    db.refresh(db_service)
+    return db_service
+
+
+def delete_email_service(db: Session, service_id: int) -> bool:
+    """删除邮箱服务配置"""
+    db_service = get_email_service_by_id(db, service_id)
+    if not db_service:
+        return False
+
+    db.delete(db_service)
+    db.commit()
+    return True
+
+
+# ============================================================================
+# 注册任务 CRUD
+# ============================================================================
+
+def create_registration_task(
+    db: Session,
+    task_uuid: str,
+    email_service_id: Optional[int] = None,
+    proxy: Optional[str] = None
+) -> RegistrationTask:
+    """创建注册任务"""
+    db_task = RegistrationTask(
+        task_uuid=task_uuid,
+        email_service_id=email_service_id,
+        proxy=proxy,
+        status='pending'
+    )
+    db.add(db_task)
+    db.commit()
+    db.refresh(db_task)
+    return db_task
+
+
+def get_registration_task_by_uuid(db: Session, task_uuid: str) -> Optional[RegistrationTask]:
+    """根据 UUID 获取注册任务"""
+    return db.query(RegistrationTask).filter(RegistrationTask.task_uuid == task_uuid).first()
+
+
+def get_registration_tasks(
+    db: Session,
+    status: Optional[str] = None,
+    skip: int = 0,
+    limit: int = 100
+) -> List[RegistrationTask]:
+    """获取注册任务列表"""
+    query = db.query(RegistrationTask)
+
+    if status:
+        query = query.filter(RegistrationTask.status == status)
+
+    query = query.order_by(desc(RegistrationTask.created_at)).offset(skip).limit(limit)
+    return query.all()
+
+
+def update_registration_task(
+    db: Session,
+    task_uuid: str,
+    **kwargs
+) -> Optional[RegistrationTask]:
+    """更新注册任务状态"""
+    db_task = get_registration_task_by_uuid(db, task_uuid)
+    if not db_task:
+        return None
+
+    for key, value in kwargs.items():
+        if hasattr(db_task, key):
+            setattr(db_task, key, value)
+
+    db.commit()
+    db.refresh(db_task)
+    return db_task
+
+
+def append_task_log(db: Session, task_uuid: str, log_message: str) -> bool:
+    """追加任务日志"""
+    db_task = get_registration_task_by_uuid(db, task_uuid)
+    if not db_task:
+        return False
+
+    if db_task.logs:
+        db_task.logs += f"\n{log_message}"
+    else:
+        db_task.logs = log_message
+
+    db.commit()
+    return True
+
+
+def delete_registration_task(db: Session, task_uuid: str) -> bool:
+    """删除注册任务"""
+    db_task = get_registration_task_by_uuid(db, task_uuid)
+    if not db_task:
+        return False
+
+    db.delete(db_task)
+    db.commit()
+    return True
+
+
+# 为 API 路由添加别名
+get_account = get_account_by_id
+get_registration_task = get_registration_task_by_uuid
+
+
+# ============================================================================
+# 设置 CRUD
+# ============================================================================
+
+def get_setting(db: Session, key: str) -> Optional[Setting]:
+    """获取设置"""
+    return db.query(Setting).filter(Setting.key == key).first()
+
+
+def get_settings_by_category(db: Session, category: str) -> List[Setting]:
+    """根据分类获取设置"""
+    return db.query(Setting).filter(Setting.category == category).all()
+
+
+def set_setting(
+    db: Session,
+    key: str,
+    value: str,
+    description: Optional[str] = None,
+    category: str = 'general'
+) -> Setting:
+    """设置或更新配置项"""
+    db_setting = get_setting(db, key)
+    if db_setting:
+        db_setting.value = value
+        db_setting.description = description or db_setting.description
+        db_setting.category = category
+        db_setting.updated_at = datetime.utcnow()
+    else:
+        db_setting = Setting(
+            key=key,
+            value=value,
+            description=description,
+            category=category
+        )
+        db.add(db_setting)
+
+    db.commit()
+    db.refresh(db_setting)
+    return db_setting
+
+
+def delete_setting(db: Session, key: str) -> bool:
+    """删除设置"""
+    db_setting = get_setting(db, key)
+    if not db_setting:
+        return False
+
+    db.delete(db_setting)
+    db.commit()
+    return True