Yuerchu
b364b740ca
- Updated README to include KodBox in project vision. - Added model descriptions for clarity in Download, File, Folder, Group, Node, Order, Policy, Redeem, Report, Request, Response, Setting, Share, SourceLink, StoragePack, Tag, Task, User, and WebDAV. - Changed optional fields from Optional[...] to the new union type syntax (e.g., str | None). - Improved foreign key references in models for consistency. - Refactored relationships in models to use singular forms where appropriate. - Updated login service to reflect changes in request model types.
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from pkg.JWT.jwt import create_access_token, create_refresh_token
|
|
from models.setting import Setting
|
|
from models.request import LoginRequest
|
|
from models.response import TokenModel
|
|
from models.user import User
|
|
from loguru import logger as log
|
|
|
|
async def Login(LoginRequest: LoginRequest) -> TokenModel | bool | None:
|
|
"""
|
|
根据账号密码进行登录。
|
|
|
|
如果登录成功，返回一个 TokenModel 对象，包含访问令牌和刷新令牌以及它们的过期时间。
|
|
如果登录异常，返回 `int` 状态码，`1` 为未完成注册，`2` 为账号被封禁。
|
|
如果登录失败，返回 `None`。
|
|
|
|
:param username: 用户名或邮箱
|
|
:type username: str
|
|
:param password: 用户密码
|
|
:type password: str
|
|
:param captcha: 验证码
|
|
:type captcha: str | None
|
|
:param twoFaCode: 两步验证代码
|
|
:type twoFaCode: str | None
|
|
|
|
:return: TokenModel 对象或状态码或 None
|
|
:rtype: TokenModel | int | None
|
|
"""
|
|
from pkg.password.pwd import Password
|
|
|
|
isCaptchaRequired = await Setting.get(type='auth', name='login_captcha', format='bool')
|
|
captchaType = await Setting.get(type='auth', name='captcha_type', format='str')
|
|
|
|
# [TODO] 验证码校验
|
|
|
|
# 获取用户信息
|
|
user = await User.get(email=LoginRequest.username)
|
|
|
|
# 验证用户是否存在
|
|
if not user:
|
|
log.debug(f"Cannot find user with email: {LoginRequest.username}")
|
|
return None
|
|
|
|
# 验证密码是否正确
|
|
if not Password.verify(user.password, LoginRequest.password):
|
|
log.debug(f"Password verification failed for user: {LoginRequest.username}")
|
|
return None
|
|
|
|
# 验证用户是否可登录
|
|
if not user.status:
|
|
# 未完成注册 or 账号已被封禁
|
|
return False
|
|
|
|
# 创建令牌
|
|
access_token, access_expire = create_access_token(data={'sub': user.email})
|
|
refresh_token, refresh_expire = create_refresh_token(data={'sub': user.email})
|
|
|
|
return TokenModel(
|
|
access_token=access_token,
|
|
access_expires=access_expire,
|
|
refresh_token=refresh_token,
|
|
refresh_expires=refresh_expire,
|
|
) |