短信验证码登录业务逻辑

校验账号是否存在 根据前端传过来的值查询数据库,查不到的时候直接抛出异常 提示手机号错误 返回给前端Redis 校验短袖验证码 拼接手机号对应的Redis 验证码缓存key读取缓存里存的验证码 读取缓存key 当没读取缓存时提示验证码过期报错 缓存读取到的验证码需要和用户输入的验证码保持一致 不一致提示验证码错误 。登录通过时生成token令牌 用求职责ID、手机号生成access_token短期访问令牌 refresh_token 刷新令牌打印日志记录两个token清扫缓存并返回信息 删除 Redis 你已使用过验证码 防止重复使用staticmethod async def login_by_mobile_code(login_request: LoginSmsCodeRequest): 1:判断手机号是否存在 2:redis中是否存在验证码并且一样,登录成功,生成token,删除redis中的验证码 3:不一样,登录失败 :return: jobseeker await JobSeeker.get_or_none(mobilelogin_request.mobile) if jobseeker is None: raise Exception(手机号不存在) key fboss-api:login:sms:{login_request.mobile} redis_code redis_client.get(key) if redis_code is None: raise Exception(验证码已过期) if redis_code ! login_request.code: raise Exception(验证码错误) access_token, refresh_token create_tokens(str(jobseeker.id), jobseeker.mobile) logger.info(f生成 access_token:{access_token}) logger.info(f生成 refresh_token:{refresh_token}) redis_client.delete(key) return { jobseeker: jobseeker, access_token: access_token, refresh_token: refresh_token, }将求职责的头像和前端传来的图片转换成二进制调用阿里云存入到求职责的文件夹里面上传成功后判断执行结果 上传成功打印日志 把oss返回的图片生成一个链接赋值给用户头像最终返回上传资源信息 文件上传失败则抛出异常传过前端头像上传失败async def update_avatar(job_seeker_id: str, file:UploadFile): file_content await file.read() filename file.filename oss AliyunOSSTool() is_success, success_res oss.upload_single_file( file_content, filename, oss_pathjob_seeker_avatar/#里面填写阿里云的文件名称 ) if is_success: logger.info(上传成功) jobseeker await JobSeeker.get_or_none(id job_seeker_id) jobseeker.avatar success_res[access_url] await jobseeker.save() return success_res else: raise Exception(上传失败)该静态异步方法实现求职者短信验证码注册功能接收手机号、验证码、密码注册参数。首先读取 Redis 中该手机号缓存的验证码无缓存则提示验证码过期缓存与输入不一致则返回验证码错误接着查询数据库判断手机号是否已注册存在账号则提示直接登录。校验全部通过后自动创建求职者账号默认昵称由手机号后四位生成同时清除 Redis 内已使用的验证码。最后根据新用户 ID 与手机号生成访问令牌和刷新令牌直接返回注册成功提示与双 token用户完成注册后无需再次登录简化注册登录流程。staticmethod async def register(register_req: RegisterRequest): mobile register_req.mobile code register_req.code password register_req.password # 1.校验验证码 key fboss-api:login:sms:{mobile} redis_code await redis_client.get(key) if not redis_code: raise Exception(验证码已过期请重新获取) if redis_code ! code: raise Exception(验证码错误) # 2.校验手机号是否已注册 exist_user await JobSeeker.get_or_none(mobilemobile) if exist_user: raise Exception(该手机号已经注册直接登录) # 3.创建用户 new_user await JobSeeker.create( mobilemobile, passwordpassword, nicknamef用户{mobile[-4:]} ) # 删除验证码 await redis_client.delete(key) # 4.登录直接返回token access_token, refresh_token create_tokens(str(new_user.id), new_user.mobile) return { msg: 注册成功, access_token: access_token, refresh_token: refresh_token }密码重置功能实现分析该代码片段实现了一个基于手机验证码的密码重置功能属于静态异步方法。以下是核心逻辑分解参数接收与验证方法接收ResetPwdRequest类型的参数包含三个关键字段mobile reset_req.mobile # 手机号 code reset_req.code # 短信验证码 new_pwd reset_req.new_password # 新密码验证码校验流程从Redis获取存储的验证码进行比对key fboss-api:login:sms:{mobile} redis_code await redis_client.get(key) if not redis_code: raise Exception(验证码已过期请重新获取) if redis_code ! code: raise Exception(验证码错误)使用get_or_none方法避免未注册用户修改密码。密码更新操作更新用户对象密码字段并持久化user.password new_pwd await user.save() await redis_client.delete(key) # 清除已使用的验证码采用明文存储密码实际生产环境应使用哈希加密如bcrypt。安全增强建议密码加密存储# 使用密码哈希示例 from passlib.hash import bcrypt user.password bcrypt.hash(new_pwd)验证码时效控制 建议在Redis设置时添加过期时间如300秒避免手动清理await redis_client.setex(key, 300, code)异常处理优化 使用自定义异常类代替通用Exception提高代码可读性和可维护性。class MobileNotExistError(Exception): pass class PasswordError(Exception): pass密码安全增强 建议使用哈希加密存储和验证密码而不是明文比较。from passlib.context import CryptContext pwd_context CryptContext(schemes[bcrypt], deprecatedauto) # 密码验证改为 if not pwd_context.verify(login_mobile_request.password, jobseeker.password): raise PasswordError(密码错误)日志记录改进 建议将敏感信息如token进行脱敏处理后再记录。logger.info(f生成 access_token:{access_token[:5]}...) logger.info(f生成 refresh_token:{refresh_token[:5]}...)返回数据结构可以考虑使用Pydantic模型规范返回数据结构。class LoginResponse(BaseModel): jobseeker: JobSeeker access_token: str refresh_token: str完整优化版本staticmethod async def login_by_mobile_password(login_mobile_request: LoginMobileRequest): jobseeker await JobSeeker.get_or_none(mobilelogin_mobile_request.mobile) if jobseeker is None: raise MobileNotExistError(手机号不存在) if not pwd_context.verify(login_mobile_request.password, jobseeker.password): raise PasswordError(密码错误) access_token, refresh_token create_tokens(str(jobseeker.id), jobseeker.mobile) logger.info(f生成 access_token:{access_token[:5]}...) logger.info(f生成 refresh_token:{refresh_token[:5]}...) return LoginResponse( jobseekerjobseeker, access_tokenaccess_token, refresh_tokenrefresh_token )使用fileupload_router.post装饰器定义了一个POST请求的路由路径为/upload添加了summary上传文件作为接口的简要描述上传成功时记录日志并返回JSON响应{ code: 1, message: 上传成功, data: success_res }