mirror of
https://github.com/Cccc-owo/CheckInApp.git
synced 2026-06-17 14:06:28 +00:00
81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
import logging
|
|
from datetime import datetime, timedelta
|
|
from typing import List, Dict, Any
|
|
from sqlalchemy.orm import Session
|
|
|
|
from backend.models import User
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class AdminService:
|
|
"""管理员服务"""
|
|
|
|
@staticmethod
|
|
def get_pending_users(db: Session) -> List[User]:
|
|
"""获取待审批用户列表"""
|
|
users = (
|
|
db.query(User)
|
|
.filter(User.is_approved == False, User.role == "user")
|
|
.order_by(User.created_at.desc())
|
|
.all()
|
|
)
|
|
|
|
return users
|
|
|
|
@staticmethod
|
|
def approve_user(user_id: int, db: Session) -> Dict[str, Any]:
|
|
"""审批通过用户"""
|
|
user = db.query(User).filter(User.id == user_id).first()
|
|
|
|
if not user:
|
|
return {"success": False, "message": "用户不存在"}
|
|
|
|
if user.is_approved:
|
|
return {"success": False, "message": "用户已经通过审批"}
|
|
|
|
user.is_approved = True
|
|
user.updated_at = datetime.now()
|
|
db.commit()
|
|
|
|
logger.info(f"管理员审批通过用户: {user.alias} (ID: {user.id})")
|
|
|
|
return {"success": True, "message": "审批成功", "user_id": user.id}
|
|
|
|
@staticmethod
|
|
def reject_user(user_id: int, db: Session) -> Dict[str, Any]:
|
|
"""拒绝并删除用户"""
|
|
user = db.query(User).filter(User.id == user_id).first()
|
|
|
|
if not user:
|
|
return {"success": False, "message": "用户不存在"}
|
|
|
|
alias = user.alias
|
|
db.delete(user)
|
|
db.commit()
|
|
|
|
logger.info(f"管理员拒绝用户: {alias} (ID: {user_id})")
|
|
|
|
return {"success": True, "message": "已拒绝并删除用户"}
|
|
|
|
@staticmethod
|
|
def delete_expired_pending_users(db: Session) -> int:
|
|
"""删除24小时未审批的用户"""
|
|
cutoff_time = datetime.now() - timedelta(hours=24)
|
|
|
|
expired_users = (
|
|
db.query(User)
|
|
.filter(User.is_approved == False, User.role == "user", User.created_at < cutoff_time)
|
|
.all()
|
|
)
|
|
|
|
count = len(expired_users)
|
|
|
|
for user in expired_users:
|
|
logger.info(f"删除过期未审批用户: {user.alias} (ID: {user.id})")
|
|
db.delete(user)
|
|
|
|
db.commit()
|
|
|
|
return count
|