1.celery配置与基本使用
1.1 安装celery
pip install celery==5.0.0
1.2 新建celery/main.py
配置celery
import os
from celery import Celery
app = Celery('mycelery',
broker='redis://127.0.0.1:6379/14',
backend='redis://127.0.0.1:6379/15')
@app.task
def add(x, y):
return x + y
2.测试celery
2.1 启动celery
'''1.启动celery'''
#1.1 单进程启动celery
celery -A main worker -l INFO
#1.2 celery管理
celery multi start celery_test -A celery_test -l debug --autoscale=50,5 # celery并发数:最多50个,最少5个
ps auxww|grep "celery worker"|grep -v grep|awk '{print $2}'|xargs kill -9 # 关闭所有celery进程
3.使用celery异步发送短信
3.1 在celery_task/mian.py
中添加发送短信函数
import os
import sys
import time
from celery import Celery
app = Celery('mycelery',
broker='redis://127.0.0.1:6379/14',
backend='redis://127.0.0.1:6379/15')
@app.task
def add(x, y):
return x + y
"""
'''1.启动celery'''
#1.1 单进程启动celery
celery -A main worker -l INFO
#1.2 celery管理
celery multi start celery_test -A celery_test -l debug --autoscale=50,5 # celery并发数:最多50个,最少5个
ps auxww|grep "celery worker"|grep -v grep|awk '{print $2}'|xargs kill -9 # 关闭所有celery进程
"""
CELERY_BASE_DIR = os.path.dirname(os.path.abspath(__file__))
print(sys.path.insert(0, os.path.join(CELERY_BASE_DIR, '../syl')))
try:
@app.task(bind=True)
def send_sms_code(self, mobile, datas):
sys.path.insert(0, os.path.join(CELERY_BASE_DIR, '../syl'))
from libs.rl_sms import send_message
time.sleep(5)
try:
res = send_message(mobile, datas)
except Exception as e:
print(e)
res = '-1'
if res == '-1':
self.retry(countdown=5, max_retries=3, exc=Exception('短信发送失败'))
except Exception as e:
print(e)
3.2 在verifications/views.py
中添加celery发送短信试图函数
import os
import random
import re
import sys
from django.shortcuts import render
from django.http import HttpResponse, HttpResponseForbidden
from django.views import View
from django_redis import get_redis_connection
from libs.captcha.captcha import captcha
from rest_framework.permissions import AllowAny
from rest_framework.views import APIView
from rest_framework.response import Response
class ImageCodeView(View):
def get(self, request):
uuid = request.GET.get('uuid')
if not uuid:
return HttpResponseForbidden('uuid无效')
text, image = captcha.generate_captcha()
redis_client = get_redis_connection('img_code')
redis_client.setex(uuid, 60 * 5, text)
return HttpResponse(image, content_type='image/jpg')
class SmsCodeView(APIView):
"""使用apiview的限流"""
permission_classes = (AllowAny,)
def post(self, request):
phone = request.data.get('phone')
image_code = request.data.get('image_code')
image_code_uuid = request.data.get('image_code_uuid')
print(phone, image_code)
if not all([phone, image_code, image_code_uuid]):
return Response({"code": 999, "msg": "参数不全"})
if not re.match(r'^1[3456789]\d{9}$', phone):
return Response({"code": 999, "msg": "手机号码不正确"})
redis_client = get_redis_connection('img_code')
phone_exists = redis_client.get(phone)
if phone_exists:
return Response({"code": 999, "msg": "频繁发送, 请稍后再试"})
redis_image_code = redis_client.get(image_code_uuid)
if redis_image_code:
redis_image_code = redis_image_code.decode()
if image_code.upper() != redis_image_code:
return Response({'code': 999, 'msg': '图片验证码不正确'})
code = '%06d' % random.randint(0, 999999)
from syl.settings import BASE_DIR
sys.path.insert(0, os.path.join(BASE_DIR, '../celery_task'))
from main import send_sms_code
send_sms_code.delay(phone, (code, "5"))
print(code)
pl = redis_client.pipeline()
pl.setex(phone, 60 * 5, code)
pl.delete(image_code_uuid)
pl.execute()
return Response({"code": 0, "msg": "短信发送成功"})
3.3 添加路由verifications/urls
from django.urls import path
from . import views
urlpatterns = [
path('image_codes/', views.ImageCodeView.as_view()),
path('sms_codes/', views.SmsCodeView.as_view()),
]
123456789
3.4.测试接口
http://192.168.56.100:8888/verify/sms_codes/
请求携带参数
{
"phone": 18538752511,
"image_code":"aed3", # 前端生成的 图形验证码
"image_code_uuid":"de8edce2-fc9f-11ea-9325-005056c00008" # 前端生成的uuid
}
4.0 django添加检查用户名和手机号是否存在接口和注册接口
4.1在user/urls.py
中添加
urlpatterns = [
path('count/', views.RegCountView.as_view()),
path('register/', views.RegisterView.as_view()),
]
4.2 在user/views.py
中添加检查视图函数
import datetime
import random
from django.http import HttpResponse
from django.views import View
from django_filters.rest_framework import DjangoFilterBackend
from rest_framework import viewsets
from rest_framework.authentication import BasicAuthentication, SessionAuthentication
from rest_framework.decorators import action
from rest_framework.filters import OrderingFilter
from rest_framework.permissions import AllowAny, IsAdminUser, IsAuthenticated, IsAuthenticatedOrReadOnly
from rest_framework.response import Response
from rest_framework.throttling import UserRateThrottle
from rest_framework.pagination import PageNumberPagination
from rest_framework.views import APIView
from rest_framework.permissions import BasePermission, SAFE_METHODS
from .models import User
from .serializers import UserSerializer, UserUnActiveSerializer, UserInfoSerializer
import redis
redis_client = redis.Redis(db=2)
class RegCountView(APIView):
permission_classes = (AllowAny,)
def post(self, request):
datatype = request.data.get('type')
data = request.data.get('data')
if not all([data, datatype]):
return Response({'code': 999, 'msg': '参数不完整'})
if datatype == 'username':
count = User.objects.filter(username=data).count()
if datatype == 'phone':
count = User.objects.filter(phone=data).count()
return Response({'code': 0, 'msg': '查询成功', 'data': {'type': datatype, 'count': count}})
4.2 在user/views.py
中添注册视图函数
class RegisterView(APIView):
"""
用户注册, 权限是: 匿名用户可访问
"""
permission_classes = (AllowAny,)
def post(self, request):
"""
接收用户名,密码,手机号和验证码, 前端校验两遍一致性, 注册成功后返回成功, 然后用户自行登录获取token
1. 用户名
2. 密码
3. 手机号
4. 验证码
:param request:
:return: {'code':0,'msg':'注册成功'}
code: "260361"
password: "123123"
phone: "13303479527"
username: "liangxuepeng"
"""
username = request.data.get('username')
phone = request.data.get('phone')
code = request.data.get('code')
passwrod = request.data.get('password')
if all([username, passwrod, phone, code]):
print(username, passwrod, phone, code)
else:
return Response({'code': 999, 'msg': '参数不全'})
code_redis = redis_client.get(phone)
if code_redis:
code_redis = code_redis.decode()
if not code == code_redis:
return Response({'code': 999, 'msg': '手机验证码错误'})
user = User(username=username, phone=phone)
user.set_password(passwrod)
user.save()
return Response({'code': 0, 'msg': '注册成功'})
4.3测试检查接口
http://192.168.56.100:8888/user/count/
if not code == code_redis:
return Response({'code': 999, 'msg': '手机验证码错误'})
user = User(username=username, phone=phone)
user.set_password(passwrod)
user.save()
return Response({'code': 0, 'msg': '注册成功'})
## 4.3测试检查接口
```python
http://192.168.56.100:8888/user/count/