103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
|
import importlib
|
||
|
|
import pkgutil
|
||
|
|
|
||
|
|
from fastapi import APIRouter
|
||
|
|
from pydantic import BaseModel, ConfigDict
|
||
|
|
|
||
|
|
from common.exception_handler import CustomException
|
||
|
|
from common.result import Result
|
||
|
|
from models import Admin, Student
|
||
|
|
|
||
|
|
|
||
|
|
class Account(BaseModel):
|
||
|
|
model_config = ConfigDict(from_attributes=True)
|
||
|
|
|
||
|
|
id: int | None = None
|
||
|
|
username: str | None = None
|
||
|
|
password: str | None = None
|
||
|
|
newPassword: str | None = None
|
||
|
|
role: str | None = None
|
||
|
|
name: str | None = None
|
||
|
|
avatar: str | None = None
|
||
|
|
clazzId: int | None = None
|
||
|
|
majorId: int | None = None
|
||
|
|
score: int | None = None
|
||
|
|
|
||
|
|
|
||
|
|
api_router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
# 登录
|
||
|
|
@api_router.post("/login")
|
||
|
|
async def login(account: Account):
|
||
|
|
if account.role == '管理员':
|
||
|
|
admin = await Admin.get_or_none(username=account.username)
|
||
|
|
if admin is None:
|
||
|
|
raise CustomException("账号或密码错误")
|
||
|
|
if admin.password != account.password:
|
||
|
|
raise CustomException("账号或密码错误")
|
||
|
|
account = Account.model_validate(admin)
|
||
|
|
elif account.role == '学生':
|
||
|
|
student = await Student.get_or_none(username=account.username).prefetch_related("clazz__major")
|
||
|
|
if student is None:
|
||
|
|
raise CustomException("账号或密码错误")
|
||
|
|
if student.password != account.password:
|
||
|
|
raise CustomException("账号或密码错误")
|
||
|
|
account = Account.model_validate(student)
|
||
|
|
account.clazzId = student.clazz.id if student and student.clazz else None
|
||
|
|
account.majorId = student.clazz.major.id if student and student.clazz and student.clazz.major else None
|
||
|
|
else:
|
||
|
|
raise CustomException("角色错误")
|
||
|
|
return Result.success(account)
|
||
|
|
|
||
|
|
|
||
|
|
# 注册
|
||
|
|
@api_router.post("/register")
|
||
|
|
async def register(account: Account):
|
||
|
|
if account.username is None:
|
||
|
|
raise CustomException("账号不能为空")
|
||
|
|
if account.password is None:
|
||
|
|
raise CustomException("密码不能为空")
|
||
|
|
# 设置默认的name
|
||
|
|
if account.name is None:
|
||
|
|
account.name = account.username
|
||
|
|
student = await Student.get_or_none(username=account.username)
|
||
|
|
if student is not None:
|
||
|
|
raise CustomException("账号已存在")
|
||
|
|
create_data = account.model_dump(exclude_unset=True, exclude={"id"})
|
||
|
|
create_data['score'] = 0
|
||
|
|
await Student.create(**create_data)
|
||
|
|
return Result.success()
|
||
|
|
|
||
|
|
|
||
|
|
# 修改密码
|
||
|
|
@api_router.put("/updatePassword")
|
||
|
|
async def update_password(account: Account):
|
||
|
|
if account.role == '管理员':
|
||
|
|
admin = await Admin.get_or_none(id=account.id)
|
||
|
|
if admin is None:
|
||
|
|
raise CustomException("未找到用户")
|
||
|
|
if admin.password != account.password:
|
||
|
|
raise CustomException("原密码错误")
|
||
|
|
if admin.password == account.newPassword:
|
||
|
|
raise CustomException("新密码不能原密码跟相同")
|
||
|
|
await Admin.filter(id=admin.id).update(password=account.newPassword)
|
||
|
|
if account.role == '学生':
|
||
|
|
student = await Student.get_or_none(id=account.id)
|
||
|
|
if student is None:
|
||
|
|
raise CustomException("未找到用户")
|
||
|
|
if student.password != account.password:
|
||
|
|
raise CustomException("原密码错误")
|
||
|
|
if student.password == account.newPassword:
|
||
|
|
raise CustomException("新密码不能原密码跟相同")
|
||
|
|
await Student.filter(id=student.id).update(password=account.newPassword)
|
||
|
|
return Result.success(account)
|
||
|
|
|
||
|
|
|
||
|
|
# 自动导入当前目录下的所有模块
|
||
|
|
for _, module_name, _ in pkgutil.iter_modules(__path__, __name__ + "."):
|
||
|
|
module = importlib.import_module(module_name)
|
||
|
|
if hasattr(module, "router"):
|
||
|
|
# 假设每个端点文件都有一个 router 变量
|
||
|
|
api_router.include_router(module.router)
|