You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
flyapps/fir_ser/api/views/login.py

174 lines
6.5 KiB

from django.contrib import auth
from api.models import Token, UserInfo
from rest_framework.response import Response
from rest_framework.viewsets import ViewSetMixin
from api.utils.serializer import UserInfoSerializer
from rest_framework.viewsets import generics
from api.utils.randomstrings import make_from_user_uuid
from rest_framework.views import APIView
from fir_ser import settings
import binascii
import os,datetime
from api.utils.app.TokenManager import DownloadToken,generateNumericTokenOfLength
from api.utils.auth import ExpiringTokenAuthentication
from api.utils.app.analyze import delete_apps_icon_storage
from api.utils.response import BaseResponse
from api.utils.captcha_verify import verify
from django.middleware import csrf
def get_token(request):
token = csrf.get_token(request)
return {'csrf_token': token}
class LoginView(APIView):
def generate_key(self):
return binascii.hexlify(os.urandom(20)).decode()
def post(self, request):
response = BaseResponse()
receive = request.data
if request.method == 'POST':
print(receive)
# is_valid = verify(receive)
is_valid = True
print("is_valid", is_valid)
if is_valid:
username = receive.get("username")
password = receive.get("password")
user = auth.authenticate(username=username, password=password)
if user is not None:
# update the token
key = self.generate_key()
now = datetime.datetime.now()
Token.objects.update_or_create(user=user, defaults={"access_token": key, "created": now})
user_info = UserInfo.objects.get(pk=user.pk)
serializer = UserInfoSerializer(user_info)
data = serializer.data
response.msg = "验证成功!"
response.userinfo = data
response.token = key
else:
try:
UserInfo.objects.get(username=username)
response.msg = "密码错误!"
response.code = 1002
except UserInfo.DoesNotExist:
response.msg = "用户不存在!"
response.code = 1003
else:
response.code = 1001
response.msg = "请完成滑动验证!"
return Response(response.dict)
def get(self,request):
response = BaseResponse()
csrf=get_token(request)
response.data=csrf
return Response(response.dict)
class UserInfoView(APIView):
authentication_classes = [ExpiringTokenAuthentication, ]
def get(self,request):
res = BaseResponse()
serializer = UserInfoSerializer(request.user)
res.data = serializer.data
act = request.query_params.get("act",None)
if act and act == "sms":
sms_token_obj=DownloadToken()
sms_code=generateNumericTokenOfLength(6)
sms_token=sms_token_obj.make_token([sms_code])
res.data["sms_token"] = sms_token
res.data["sms_code"] = sms_code
return Response(res.dict)
def put(self,request):
res = BaseResponse()
request.user.qq=request.data.get("qq",request.user.qq)
request.user.job=request.data.get("job", request.user.job)
request.user.first_name=request.data.get("first_name", request.user.first_name)
oldpassword=request.data.get("oldpassword",None)
surepassword=request.data.get("surepassword",None)
if oldpassword and surepassword:
user = auth.authenticate(username=request.user.username, password=oldpassword)
if user is not None:
user.set_password(surepassword)
user.save()
res.msg="密码修改成功"
return Response(res.dict)
else:
res.code = 1004
res.msg = "老密码校验失败"
else:
sms_token = request.data.get("sms_token",None)
if sms_token:
sms_token_obj=DownloadToken()
if sms_token_obj.verify_token(sms_token,request.data.get("sms_code",None)):
request.user.mobile=request.data.get("mobile", request.user.mobile)
request.user.save()
serializer = UserInfoSerializer(request.user)
res.data = serializer.data
return Response(res.dict)
return Response(res.dict)
def post(self, request):
res = BaseResponse()
# 获取多个file
files = request.FILES.getlist('file', None)
for file_obj in files:
# 将文件缓存到本地后上传
try:
app_type = file_obj.name.split(".")[-1]
if app_type in ['png','jpeg','jpg']:
#上传图片
pass
else:
raise
except Exception as e:
res.code = 1003
res.msg = "错误的类型"
return Response(res.dict)
# img_file_name = request.user.head_img
# if img_file_name == "" or img_file_name == '/files/imgs/head_img.jpeg':
old_head_img = request.user.head_img
random_file_name = make_from_user_uuid(request.user)
head_img = "/".join([settings.MEDIA_URL.strip("/"), "imgs", random_file_name + "." + app_type])
local_file = os.path.join(settings.MEDIA_ROOT,"imgs",random_file_name + "." + app_type)
# 读取传入的文件
try:
destination = open(local_file, 'wb+')
for chunk in file_obj.chunks():
# 写入本地文件
destination.write(chunk)
destination.close()
except Exception as e:
res.code = 1003
res.msg = "数据写入失败"
return Response(res.dict)
try:
request.user.head_img = head_img
request.user.save()
if old_head_img != "" or old_head_img != '/files/imgs/head_img.jpeg':
delete_apps_icon_storage(os.path.basename(old_head_img),'imgs')
except Exception as e:
res.code = 1003
res.msg = "头像保存失败"
return Response(res.dict)
return Response(res.dict)