用户登录

service/user/login.py

@@ -1,15 +1,63 @@
 from models.setting import Setting
+from models.response import TokenModel
+from models.user import User
+from pkg.log import log
 
 async def login(
     username: str,
     password: str
-):
+) -> TokenModel | int | None:
     """
+    根据账号密码进行登录。
+
+    如果登录成功，返回一个 TokenModel 对象，包含访问令牌和刷新令牌以及它们的过期时间。
+    如果登录异常，返回 `int` 状态码，`1` 为未完成注册，`2` 为账号被封禁。
+    如果登录失败，返回 `None`。
+
+    :param username: 用户名或邮箱
+    :type username: str
+    :param password: 用户密码
+    :type password: str
+
+    :return: TokenModel 对象或状态码或 None
+    :rtype: TokenModel | int | None
     """
+    from pkg.password.pwd import Password
     
-    isCaptchaRequired = await Setting.get(type='auth', name='login_captcha', type=bool)
-    captchaType = await Setting.get(type='auth', name='captcha_type', type=str)
+    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=username)
+
+    if not user:
+        log.debug(f"Cannot find user with email: {username}")
+        return None
+    
+    # 验证密码是否正确
+    if not Password.verify(user.password, password):
+        log.debug(f"Password verification failed for user: {username}")
+        return None
+    
+    # 验证用户是否可登录
+    if user.status == 1:
+        # 未完成注册
+        return 1
+    elif user.status == 2:
+        # 账号已被封禁
+        return 2
+    
+    # 创建令牌
+    from pkg.JWT.JWT import create_access_token, create_refresh_token
+    
+    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,
+    )