2015-06-29 04:43:17 +00:00
|
|
|
|
# coding=utf-8
|
|
|
|
|
from django.contrib import auth
|
|
|
|
|
from django.shortcuts import render
|
2015-08-07 12:25:58 +00:00
|
|
|
|
from django.db.models import Q
|
|
|
|
|
|
2015-06-29 04:43:17 +00:00
|
|
|
|
from rest_framework.views import APIView
|
|
|
|
|
|
2015-08-07 12:25:58 +00:00
|
|
|
|
from utils.shortcuts import serializer_invalid_response, error_response, success_response, paginate
|
2015-06-29 04:43:17 +00:00
|
|
|
|
|
2015-08-12 14:59:27 +00:00
|
|
|
|
from .decorators import login_required
|
2015-06-29 04:43:17 +00:00
|
|
|
|
from .models import User
|
2015-08-12 14:59:27 +00:00
|
|
|
|
from .serializers import (UserLoginSerializer, UsernameCheckSerializer,
|
|
|
|
|
UserRegisterSerializer, UserChangePasswordSerializer,
|
|
|
|
|
EmailCheckSerializer, UserSerializer, EditUserSerializer)
|
2015-06-29 04:43:17 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UserLoginAPIView(APIView):
|
|
|
|
|
def post(self, request):
|
|
|
|
|
"""
|
|
|
|
|
用户登录json api接口
|
|
|
|
|
---
|
|
|
|
|
request_serializer: UserLoginSerializer
|
|
|
|
|
"""
|
2015-08-07 14:01:16 +00:00
|
|
|
|
serializer = UserLoginSerializer(data=request.data)
|
2015-06-29 04:43:17 +00:00
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
data = serializer.data
|
|
|
|
|
user = auth.authenticate(username=data["username"], password=data["password"])
|
|
|
|
|
# 用户名或密码错误的话 返回None
|
|
|
|
|
if user:
|
|
|
|
|
auth.login(request, user)
|
|
|
|
|
return success_response(u"登录成功")
|
|
|
|
|
else:
|
|
|
|
|
return error_response(u"用户名或密码错误")
|
|
|
|
|
else:
|
|
|
|
|
return serializer_invalid_response(serializer)
|
|
|
|
|
|
|
|
|
|
|
2015-08-03 10:45:26 +00:00
|
|
|
|
class UserRegisterAPIView(APIView):
|
2015-06-29 04:43:17 +00:00
|
|
|
|
def post(self, request):
|
2015-08-03 10:45:26 +00:00
|
|
|
|
"""
|
|
|
|
|
用户注册json api接口
|
|
|
|
|
---
|
|
|
|
|
request_serializer: UserRegisterSerializer
|
|
|
|
|
"""
|
2015-08-07 14:01:16 +00:00
|
|
|
|
serializer = UserRegisterSerializer(data=request.data)
|
2015-08-03 10:45:26 +00:00
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
data = serializer.data
|
|
|
|
|
try:
|
|
|
|
|
User.objects.get(username=data["username"])
|
|
|
|
|
return error_response(u"用户名已存在")
|
|
|
|
|
except User.DoesNotExist:
|
2015-08-05 12:11:25 +00:00
|
|
|
|
pass
|
|
|
|
|
try:
|
|
|
|
|
User.objects.get(email=data["email"])
|
|
|
|
|
return error_response(u"该邮箱已被注册,请换其他邮箱进行注册")
|
|
|
|
|
except User.DoesNotExist:
|
|
|
|
|
user = User.objects.create(username=data["username"], real_name=data["real_name"],
|
|
|
|
|
email=data["email"])
|
2015-08-03 10:45:26 +00:00
|
|
|
|
user.set_password(data["password"])
|
|
|
|
|
user.save()
|
|
|
|
|
return success_response(u"注册成功!")
|
|
|
|
|
else:
|
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-07-17 03:00:15 +00:00
|
|
|
|
|
|
|
|
|
|
2015-08-03 11:54:55 +00:00
|
|
|
|
class UserChangePasswordAPIView(APIView):
|
2015-07-17 03:00:15 +00:00
|
|
|
|
def post(self, request):
|
2015-08-03 11:54:55 +00:00
|
|
|
|
"""
|
|
|
|
|
用户修改密码json api接口
|
|
|
|
|
---
|
|
|
|
|
request_serializer: UserChangePasswordSerializer
|
|
|
|
|
"""
|
2015-08-07 14:01:16 +00:00
|
|
|
|
serializer = UserChangePasswordSerializer(data=request.data)
|
2015-08-03 11:54:55 +00:00
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
data = serializer.data
|
|
|
|
|
user = auth.authenticate(username=data["username"], password=data["old_password"])
|
|
|
|
|
if user:
|
|
|
|
|
user.set_password(data["new_password"])
|
|
|
|
|
user.save()
|
|
|
|
|
return success_response(u"用户密码修改成功!")
|
|
|
|
|
else:
|
|
|
|
|
return error_response(u"密码不正确,请重新修改!")
|
|
|
|
|
else:
|
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-08-03 08:52:31 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UsernameCheckAPIView(APIView):
|
|
|
|
|
def post(self, request):
|
|
|
|
|
"""
|
|
|
|
|
检测用户名是否存在,存在返回True,不存在返回False
|
|
|
|
|
---
|
|
|
|
|
request_serializer: UsernameCheckSerializer
|
|
|
|
|
"""
|
2015-08-07 14:01:16 +00:00
|
|
|
|
serializer = UsernameCheckSerializer(data=request.data)
|
2015-08-03 08:52:31 +00:00
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
try:
|
|
|
|
|
User.objects.get(username=serializer.data["username"])
|
|
|
|
|
return success_response(True)
|
|
|
|
|
except User.DoesNotExist:
|
|
|
|
|
return success_response(False)
|
|
|
|
|
else:
|
2015-08-05 12:11:25 +00:00
|
|
|
|
return serializer_invalid_response(serializer)
|
|
|
|
|
|
2015-08-06 11:07:46 +00:00
|
|
|
|
|
2015-08-05 12:11:25 +00:00
|
|
|
|
class EmailCheckAPIView(APIView):
|
|
|
|
|
def post(self, request):
|
|
|
|
|
"""
|
|
|
|
|
检测邮箱是否存在,存在返回True,不存在返回False
|
|
|
|
|
---
|
|
|
|
|
request_serializer: EmailCheckSerializer
|
|
|
|
|
"""
|
2015-08-07 14:01:16 +00:00
|
|
|
|
serializer = EmailCheckSerializer(data=request.data)
|
2015-08-05 12:11:25 +00:00
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
try:
|
|
|
|
|
User.objects.get(email=serializer.data["email"])
|
|
|
|
|
return success_response(True)
|
|
|
|
|
except User.DoesNotExist:
|
|
|
|
|
return success_response(False)
|
|
|
|
|
else:
|
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-08-07 12:25:58 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UserAdminAPIView(APIView):
|
|
|
|
|
def put(self, request):
|
|
|
|
|
"""
|
|
|
|
|
用户编辑json api接口
|
|
|
|
|
---
|
|
|
|
|
request_serializer: EditUserSerializer
|
|
|
|
|
response_serializer: UserSerializer
|
|
|
|
|
"""
|
2015-08-07 14:01:16 +00:00
|
|
|
|
serializer = EditUserSerializer(data=request.data)
|
2015-08-07 12:25:58 +00:00
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
data = serializer.data
|
|
|
|
|
try:
|
|
|
|
|
user = User.objects.get(id=data["id"])
|
|
|
|
|
except User.DoesNotExist:
|
|
|
|
|
return error_response(u"该用户不存在!")
|
2015-08-14 02:21:50 +00:00
|
|
|
|
try:
|
|
|
|
|
user = User.objects.get(username=data["username"])
|
|
|
|
|
if user.id != data["id"]:
|
|
|
|
|
return error_response(u"昵称已经存在")
|
|
|
|
|
except User.DoesNotExist:
|
|
|
|
|
pass
|
2015-08-07 12:25:58 +00:00
|
|
|
|
user.username = data["username"]
|
|
|
|
|
user.real_name = data["real_name"]
|
|
|
|
|
user.email = data["email"]
|
|
|
|
|
user.admin_type = data["admin_type"]
|
|
|
|
|
if data["password"]:
|
|
|
|
|
user.set_password(data["password"])
|
|
|
|
|
user.save()
|
|
|
|
|
return success_response(UserSerializer(user).data)
|
|
|
|
|
else:
|
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-08-12 14:59:27 +00:00
|
|
|
|
|
|
|
|
|
def get(self, request):
|
|
|
|
|
"""
|
|
|
|
|
用户分页json api接口
|
|
|
|
|
---
|
|
|
|
|
response_serializer: UserSerializer
|
|
|
|
|
"""
|
|
|
|
|
user = User.objects.all().order_by("-create_time")
|
|
|
|
|
admin_type = request.GET.get("admin_type", None)
|
|
|
|
|
if admin_type:
|
|
|
|
|
try:
|
|
|
|
|
user = user.filter(admin_type__gte=int(admin_type))
|
|
|
|
|
except ValueError:
|
|
|
|
|
return error_response(u"参数错误")
|
|
|
|
|
keyword = request.GET.get("keyword", None)
|
|
|
|
|
if keyword:
|
|
|
|
|
user = user.filter(Q(username__contains=keyword) |
|
|
|
|
|
Q(real_name__contains=keyword) |
|
|
|
|
|
Q(email__contains=keyword))
|
|
|
|
|
return paginate(request, user, UserSerializer)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UserInfoAPIView(APIView):
|
|
|
|
|
@login_required
|
|
|
|
|
def get(self, request):
|
|
|
|
|
"""
|
|
|
|
|
返回这个用户的个人信息
|
|
|
|
|
---
|
|
|
|
|
response_serializer: UserSerializer
|
|
|
|
|
"""
|
|
|
|
|
return success_response(UserSerializer(request.user).data)
|