class SMSCodeView(ViewSet): # 验证码模块,设定了一系列的规则 # GET请求,方法作用于整个集合 @action(["GET"], False) def code(self, request): ser = SMSSerializer(data=request.query_params) ser.is_valid(True) code = ser.validated_data.get("code") phone = ser.validated_data.get("phone") method = ser.validated_data.get("method") if code and (code == cache.get(method + phone) or code == "6666"): return APIResponse(response_code.SUCCESS_VALID_CODE, "验证码正确") if code and code != cache.get(method + phone): return APIResponse(response_code.INCORRECT_CODE, "验证码错误") code = numberCode() res = SMS.send_sms(phone, code, method) errmsg = res.get("errmsg") log.info(f"{phone}\t{method}\t{code}\t{errmsg}") if not res: return APIResponse(response_code.FAIL_TO_SEND, "验证码发送失败") if errmsg != "OK": return APIResponse(response_code.SEND_FORBIDDEN, "验证码发送失败", {"errmsg": errmsg}) # 进行缓存 cache.set(method + phone, code, SMS.EXPIRE_TIME * 60) return APIResponse(response_code.SUCCESS_SEND_SMS, "验证码发送成功")
时间: 2024-04-21 18:24:16 浏览: 191
django rest framework get 添加自定义参数的两种方式
这段代码是一个 Django REST Framework 中的视图集类 `SMSCodeView`,它继承自 `ViewSet` 类。
这个视图集定义了一个名为 `code` 的自定义操作,该操作使用 GET 方法访问整个集合。具体的逻辑如下:
1. 首先,使用 `SMSSerializer` 对请求的查询参数进行数据验证和反序列化。
2. 如果验证通过,从验证后的数据中获取 `code`、`phone` 和 `method`。
3. 如果存在验证码 `code`,并且与缓存中的验证码相匹配,或者是固定的测试验证码 "6666",则返回验证码正确的响应。
4. 如果存在验证码 `code`,但与缓存中的验证码不匹配,则返回验证码错误的响应。
5. 生成一个新的随机验证码 `code`,并使用 `SMS.send_sms` 方法发送到指定的手机号码 `phone`。
6. 根据发送结果判断是否发送成功,如果不成功则返回相应的失败响应。
7. 如果发送成功,将生成的验证码 `code` 存入缓存中,以便后续验证。
8. 返回验证码发送成功的响应。
需要注意的是,这段代码中涉及到了一些自定义的类和方法,例如 `SMSSerializer`、`APIResponse`、`SMS.send_sms` 等,这些类和方法可能是根据具体业务需求自定义的。你可能需要根据自己的项目需求进行相应的调整和实现。
阅读全文