v0.1测试

backend/models/admin_role.py

@@ -0,0 +1,60 @@
+# ===========================================
+# 班级操行分管理系统 - 后端服务
+# 
+# 开发者: Canglan
+# 联系方式: admin@sea-studio.top
+# 版权归属: Sea Network Technology Studio
+# 许可证: MIT License
+# 
+# 版权所有 © Sea Network Technology Studio
+# ===========================================
+
+from typing import Optional, Dict, Any, List
+from utils.database import execute_one, execute_query, execute_insert, execute_update
+
+
+class AdminRoleModel:
+    """管理员角色数据模型"""
+    
+    @staticmethod
+    async def get_by_user_id(user_id: int) -> Optional[Dict[str, Any]]:
+        sql = """
+            SELECT ar.*, c.class_name, s.subject_name
+            FROM admin_roles ar
+            LEFT JOIN classes c ON ar.class_id = c.class_id
+            LEFT JOIN subjects s ON ar.subject_id = s.subject_id
+            WHERE ar.user_id = %s
+            LIMIT 1
+        """
+        return await execute_one(sql, (user_id,))
+    
+    @staticmethod
+    async def get_class_id_by_user(user_id: int) -> Optional[int]:
+        sql = "SELECT class_id FROM admin_roles WHERE user_id = %s LIMIT 1"
+        result = await execute_one(sql, (user_id,))
+        return result["class_id"] if result else None
+    
+    @staticmethod
+    async def get_by_class(class_id: int) -> List[Dict[str, Any]]:
+        sql = """
+            SELECT ar.*, u.real_name, u.username
+            FROM admin_roles ar
+            JOIN users u ON ar.user_id = u.user_id
+            WHERE ar.class_id = %s
+            ORDER BY ar.role_type
+        """
+        return await execute_query(sql, (class_id,))
+    
+    @staticmethod
+    async def create(user_id: int, role_type: str, class_id: int, subject_id: int = None) -> int:
+        sql = """
+            INSERT INTO admin_roles (user_id, role_type, class_id, subject_id)
+            VALUES (%s, %s, %s, %s)
+        """
+        return await execute_insert(sql, (user_id, role_type, class_id, subject_id))
+    
+    @staticmethod
+    async def delete(user_id: int) -> bool:
+        sql = "DELETE FROM admin_roles WHERE user_id = %s"
+        result = await execute_update(sql, (user_id,))
+        return result > 0