7fbdcb3b0f
1. 新增：微信小程序端新增微信手机号登录功能（必须为企业认证小程序） 2. 新增：加入动态更新常见问题 3. 新增：新增小程序分享功能 4. 新增：小程序新增第一次登录需要修改密码 5. 新增：新增接口权限控制 6. 新增：用户新增is_staff用来判断是否为工作人员 7. 新增：软删除新增is_delete字段来判断，delete_datetime当前主要来记录时间 8. 更新：部分接口删除功能已更新，需要试用软删除的才会试用软删除 9. 更新：更新系统配置缓存功能 10. 更新：接口认证依赖项更新 11. 更新：获取系统基础配置信息与用户协议与隐私协议更新 12. 优化：优化接口与数据库操作
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Creaet Time : 2022/11/9 10:15
|
||
# @File : login.py
|
||
# @IDE : PyCharm
|
||
# @desc : 登录验证装饰器
|
||
|
||
from fastapi import Request, Depends
|
||
from pydantic import BaseModel, validator
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from apps.vadmin.auth import models, crud, schemas
|
||
from core.database import db_getter
|
||
from core.validator import vali_telephone
|
||
from typing import Optional
|
||
|
||
|
||
class LoginForm(BaseModel):
|
||
telephone: str
|
||
password: str
|
||
method: str = '0' # 认证方式，0：密码登录，1：短信登录，2：微信一键登录
|
||
platform: str = '0' # 登录平台，0：PC端管理系统，1：移动端管理系统
|
||
|
||
# validators
|
||
_normalize_telephone = validator('telephone', allow_reuse=True)(vali_telephone)
|
||
|
||
|
||
class WXLoginForm(BaseModel):
|
||
telephone: Optional[str] = None
|
||
code: str
|
||
method: str = '2' # 认证方式，0：密码登录，1：短信登录，2：微信一键登录
|
||
platform: str = '1' # 登录平台，0：PC端管理系统，1：移动端管理系统
|
||
|
||
|
||
class LoginResult(BaseModel):
|
||
status: Optional[bool] = False
|
||
user: Optional[schemas.UserOut] = None
|
||
msg: Optional[str] = None
|
||
|
||
class Config:
|
||
arbitrary_types_allowed = True
|
||
|
||
|
||
class LoginValidation:
|
||
|
||
"""
|
||
验证用户登录时提交的数据是否有效
|
||
"""
|
||
|
||
def __init__(self, func):
|
||
self.func = func
|
||
|
||
async def __call__(self, data: LoginForm, db: AsyncSession, request: Request) -> LoginResult:
|
||
self.result = LoginResult()
|
||
if data.platform not in ["0", "1"]:
|
||
self.result.msg = "错误平台"
|
||
return self.result
|
||
user = await crud.UserDal(db).get_data(telephone=data.telephone, v_return_none=True)
|
||
if not user:
|
||
self.result.msg = "该手机号不存在！"
|
||
return self.result
|
||
|
||
result = await self.func(self, data=data, user=user, request=request)
|
||
|
||
if not result.status:
|
||
self.result.msg = result.msg
|
||
elif not user.is_active:
|
||
self.result.msg = "此手机号已被冻结！"
|
||
elif user:
|
||
self.result.msg = "OK"
|
||
self.result.status = True
|
||
self.result.user = schemas.UserSimpleOut.from_orm(user)
|
||
await user.update_login_info(db, request.client.host)
|
||
return self.result
|