Initial commit

src/services/user/preference.py

@@ -0,0 +1,137 @@
+"""
+用户偏好设置服务
+"""
+
+from typing import Optional
+
+from sqlalchemy.orm import Session
+
+from src.core.exceptions import NotFoundException
+from src.core.logger import logger
+from src.models.database import Provider, User, UserPreference
+
+
+
+class PreferenceService:
+    """用户偏好设置服务"""
+
+    @staticmethod
+    def get_or_create_preferences(db: Session, user_id: str) -> UserPreference:  # UUID
+        """获取或创建用户偏好设置"""
+        preferences = db.query(UserPreference).filter(UserPreference.user_id == user_id).first()
+
+        if not preferences:
+            # 创建默认偏好设置
+            preferences = UserPreference(
+                user_id=user_id,
+                theme="light",
+                language="zh-CN",
+                timezone="Asia/Shanghai",
+                email_notifications=True,
+                usage_alerts=True,
+                announcement_notifications=True,
+            )
+            db.add(preferences)
+            db.commit()
+            db.refresh(preferences)
+            logger.info(f"Created default preferences for user {user_id}")
+
+        return preferences
+
+    @staticmethod
+    def update_preferences(
+        db: Session,
+        user_id: str,  # UUID
+        avatar_url: Optional[str] = None,
+        bio: Optional[str] = None,
+        default_provider_id: Optional[str] = None,  # UUID
+        theme: Optional[str] = None,
+        language: Optional[str] = None,
+        timezone: Optional[str] = None,
+        email_notifications: Optional[bool] = None,
+        usage_alerts: Optional[bool] = None,
+        announcement_notifications: Optional[bool] = None,
+    ) -> UserPreference:
+        """更新用户偏好设置"""
+        preferences = PreferenceService.get_or_create_preferences(db, user_id)
+
+        # 更新提供的字段
+        if avatar_url is not None:
+            preferences.avatar_url = avatar_url
+        if bio is not None:
+            preferences.bio = bio
+        if default_provider_id is not None:
+            # 验证提供商是否存在且活跃
+            provider = (
+                db.query(Provider)
+                .filter(Provider.id == default_provider_id, Provider.is_active == True)
+                .first()
+            )
+            if not provider:
+                raise NotFoundException("Provider not found or inactive")
+            preferences.default_provider_id = default_provider_id
+        if theme is not None:
+            if theme not in ["light", "dark", "auto"]:
+                raise ValueError("Invalid theme. Must be 'light', 'dark', or 'auto'")
+            preferences.theme = theme
+        if language is not None:
+            preferences.language = language
+        if timezone is not None:
+            preferences.timezone = timezone
+        if email_notifications is not None:
+            preferences.email_notifications = email_notifications
+        if usage_alerts is not None:
+            preferences.usage_alerts = usage_alerts
+        if announcement_notifications is not None:
+            preferences.announcement_notifications = announcement_notifications
+
+        db.commit()
+        db.refresh(preferences)
+        logger.info(f"Updated preferences for user {user_id}")
+
+        return preferences
+
+    @staticmethod
+    def get_user_with_preferences(db: Session, user_id: str) -> dict:  # UUID
+        """获取用户信息及其偏好设置"""
+        user = db.query(User).filter(User.id == user_id).first()
+        if not user:
+            raise NotFoundException("User not found")
+
+        preferences = PreferenceService.get_or_create_preferences(db, user_id)
+
+        # 构建返回数据
+        user_data = {
+            "id": user.id,
+            "email": user.email,
+            "username": user.username,
+            "role": user.role.value,
+            "is_active": user.is_active,
+            "created_at": user.created_at,
+            "last_login_at": user.last_login_at,
+            "preferences": {
+                "avatar_url": preferences.avatar_url,
+                "bio": preferences.bio,
+                "default_provider": (
+                    preferences.default_provider.name if preferences.default_provider else None
+                ),
+                "theme": preferences.theme,
+                "language": preferences.language,
+                "timezone": preferences.timezone,
+                "notifications": {
+                    "email": preferences.email_notifications,
+                    "usage_alerts": preferences.usage_alerts,
+                    "announcements": preferences.announcement_notifications,
+                },
+            },
+            # 配额信息
+            "quota_usd": user.quota_usd,
+            "used_usd": user.used_usd,
+            "stats": {
+                "total_cost": user.used_usd,
+                "total_cost_all_time": user.total_usd,
+                "api_keys_count": len(user.api_keys),
+            },
+        }
+
+        return user_data