325 lines
11 KiB
Python
325 lines
11 KiB
Python
from django.shortcuts import render
|
||
from rest_framework.views import APIView
|
||
from utils.jsonResponse import SuccessResponse,ErrorResponse,DetailResponse
|
||
from utils.common import get_parameter_dic,getRandomSet
|
||
import re
|
||
from django.db.models import Q,F,Sum
|
||
from rest_framework.serializers import ModelSerializer
|
||
from rest_framework import serializers
|
||
from rest_framework_simplejwt.authentication import JWTAuthentication
|
||
from utils.serializers import CustomModelSerializer
|
||
from utils.viewset import CustomModelViewSet
|
||
from rest_framework.permissions import IsAuthenticated
|
||
from drf_yasg import openapi
|
||
from drf_yasg.utils import swagger_auto_schema
|
||
# 添加七牛云存储工具类导入
|
||
from utils.qiniu_storage import QiniuStorage
|
||
# 移除原有的图片上传工具
|
||
# from utils.imageupload import ImageUpload
|
||
from mysystem.models import Users
|
||
from utils.filters import UsersManageTimeFilter
|
||
from django.contrib.auth.hashers import make_password
|
||
from utils.export_excel import export_excel
|
||
from django.db import transaction
|
||
from apps.oauth.models import OAuthWXUser
|
||
from apps.platformsettings.models import SystemConfig
|
||
# Create your views here.
|
||
|
||
# ================================================= #
|
||
# ************** 后端用户管理 view ************** #
|
||
# ================================================= #
|
||
|
||
class UserManageSerializer(CustomModelSerializer):
|
||
"""
|
||
用户管理-序列化器
|
||
"""
|
||
|
||
class Meta:
|
||
model = Users
|
||
read_only_fields = ["id"]
|
||
exclude = ['password', 'role', 'post', 'dept','user_permissions','groups']
|
||
extra_kwargs = {
|
||
'post': {'required': False},
|
||
'role': {'required': False},
|
||
}
|
||
|
||
class UserManageCreateSerializer(CustomModelSerializer):
|
||
"""
|
||
用户管理-序列化器
|
||
"""
|
||
|
||
# 新增重写
|
||
def create(self, validated_data):
|
||
if "password" in validated_data.keys():
|
||
if validated_data['password']:
|
||
validated_data['password'] = make_password(validated_data['password'])
|
||
validated_data['identity'] = 2
|
||
return super().create(validated_data)
|
||
|
||
class Meta:
|
||
model = Users
|
||
read_only_fields = ["id"]
|
||
exclude = ['role', 'post', 'dept']
|
||
extra_kwargs = {
|
||
'post': {'required': False},
|
||
'role': {'required': False},
|
||
'name': {'required': False},
|
||
'password': {'required': False},
|
||
}
|
||
|
||
class UserManageUpdateSerializer(CustomModelSerializer):
|
||
"""
|
||
用户管理-序列化器
|
||
"""
|
||
# 更新重写
|
||
def update(self, instance, validated_data):
|
||
if "password" in validated_data.keys():
|
||
if validated_data['password']:
|
||
validated_data['password'] = make_password(validated_data['password'])
|
||
else:
|
||
validated_data.pop('password', None)
|
||
return super().update(instance,validated_data)
|
||
|
||
class Meta:
|
||
model = Users
|
||
read_only_fields = ["id"]
|
||
exclude = ['role', 'post', 'dept','identity']
|
||
extra_kwargs = {
|
||
'post': {'required': False},
|
||
'role': {'required': False},
|
||
'name': {'required': False},
|
||
'password': {'required': False},
|
||
}
|
||
class ExportUserManageSerializer(CustomModelSerializer):
|
||
"""
|
||
导出 用户信息 简单序列化器
|
||
"""
|
||
is_active_name = serializers.SerializerMethodField()
|
||
def get_is_active_name(self, obj):
|
||
if obj.is_active:
|
||
return "正常"
|
||
else:
|
||
return "禁用"
|
||
|
||
class Meta:
|
||
model = Users
|
||
fields = ('id', 'nickname','mobile', 'is_active_name','create_datetime')
|
||
|
||
class UserManageViewSet(CustomModelViewSet):
|
||
"""
|
||
后台用户管理 接口:
|
||
"""
|
||
queryset = Users.objects.filter(identity=2).order_by("-create_datetime")#排除管理员
|
||
serializer_class = UserManageSerializer
|
||
create_serializer_class = UserManageCreateSerializer
|
||
update_serializer_class = UserManageUpdateSerializer
|
||
filterset_class = UsersManageTimeFilter
|
||
|
||
def disableuser(self,request,*args, **kwargs):
|
||
"""禁用用户"""
|
||
instance = Users.objects.filter(id=kwargs.get('pk')).first()
|
||
if instance:
|
||
if instance.is_active:
|
||
instance.is_active = False
|
||
else:
|
||
instance.is_active = True
|
||
instance.save()
|
||
return SuccessResponse(data=None, msg="修改成功")
|
||
else:
|
||
return ErrorResponse(msg="未获取到用户")
|
||
|
||
def exportexecl(self, request):
|
||
field_data = ['主键', '昵称', '手机号', '状态', '创建时间']
|
||
queryset = self.filter_queryset(self.get_queryset())
|
||
data = ExportUserManageSerializer(queryset, many=True).data
|
||
return SuccessResponse(data=export_excel(request, field_data, data, '用户数据.xls'), msg='success')
|
||
|
||
# ================================================= #
|
||
# ************** 前端用户中心 view ************** #
|
||
# ================================================= #
|
||
|
||
#前端获取个人信息
|
||
from utils.qiniu_storage import QiniuStorage
|
||
|
||
class GetUserinfoView(APIView):
|
||
authentication_classes = [JWTAuthentication]
|
||
permission_classes = []
|
||
|
||
def get(self, request, *args, **kwargs):
|
||
data = {}
|
||
user = request.user
|
||
if request.user.is_authenticated:
|
||
if user.avatar:
|
||
if 'http' not in user.avatar: # 如果不是完整URL，则生成七牛云链接
|
||
qiniu = QiniuStorage()
|
||
base_url = f'http://{qiniu.domain}/{user.avatar}'
|
||
data['avatar'] = qiniu.q.private_download_url(base_url, expires=3600*24*365)
|
||
else:
|
||
data['avatar'] = user.avatar
|
||
else:
|
||
data['avatar'] = ""
|
||
data['nickname'] = user.nickname
|
||
data['mobile'] = user.mobile
|
||
data['id'] = user.id
|
||
return DetailResponse(data=data)
|
||
# 匿名用户
|
||
data['avatar'] = ""
|
||
data['nickname'] = ""
|
||
data['mobile'] = ""
|
||
data['id'] = ""
|
||
return DetailResponse(data=data)
|
||
|
||
#前端图片上传
|
||
class uploadImagesView(APIView):
|
||
'''
|
||
前端图片上传
|
||
post:
|
||
【功能描述】前端图片上传</br>
|
||
【参数说明】无，需要登录携带token后才能调用</br>
|
||
'''
|
||
authentication_classes = [JWTAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def post(self, request, *args, **kwargs):
|
||
file = request.FILES.get('file')
|
||
if not file:
|
||
return ErrorResponse(msg='请选择文件')
|
||
|
||
qiniu = QiniuStorage()
|
||
result = qiniu.upload_data(file, 'frontendimages')
|
||
|
||
if result['code'] == 200:
|
||
return SuccessResponse(data=result['url'], msg=result['msg'])
|
||
return ErrorResponse(msg=result['msg'])
|
||
|
||
class ChangeAvatarView(APIView):
|
||
'''
|
||
前端app头像修改
|
||
post:
|
||
【功能描述】前端app头像修改</br>
|
||
【参数说明】无，需要登录携带token后才能调用</br>
|
||
'''
|
||
authentication_classes = [JWTAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def post(self, request, *args, **kwargs):
|
||
file = request.FILES.get('file')
|
||
if not file:
|
||
return ErrorResponse(msg='请选择文件')
|
||
|
||
qiniu = QiniuStorage()
|
||
result = qiniu.upload_data(file, 'avatar')
|
||
|
||
if result['code'] == 200:
|
||
user = request.user
|
||
# 只存储文件路径部分，不存储完整URL
|
||
key = result['url']
|
||
user.avatar = key
|
||
user.save()
|
||
return SuccessResponse(data=result['url'], msg=result['msg'])
|
||
return ErrorResponse(msg=result['msg'])
|
||
|
||
class SetUserNicknameView(APIView):
|
||
"""
|
||
修改昵称
|
||
post:
|
||
修改昵称
|
||
【参数】nickname:需要修改的用户新昵称
|
||
"""
|
||
permission_classes = [IsAuthenticated]
|
||
authentication_classes = [JWTAuthentication]
|
||
|
||
#api文档参数
|
||
|
||
@swagger_auto_schema(operation_summary='app回收员修改昵称',
|
||
# manual_parameters=[#GET请求需要
|
||
# # openapi.Parameter("nickname", openapi.IN_QUERY, description="要修改昵称", type=openapi.TYPE_STRING)
|
||
# ],
|
||
request_body=openapi.Schema(#POST请求需要
|
||
type=openapi.TYPE_OBJECT,
|
||
required=['nickname'],
|
||
properties={
|
||
'nickname':openapi.Schema(type=openapi.TYPE_STRING,description="要修改昵称"),
|
||
},
|
||
),
|
||
responses={200:'success'},
|
||
)
|
||
|
||
def post(self, request):
|
||
nickname = get_parameter_dic(request)['nickname']
|
||
if nickname is None:
|
||
return ErrorResponse(msg="昵称不能为空")
|
||
if not isinstance(nickname,str):
|
||
return ErrorResponse(msg='类型错误')
|
||
user = request.user
|
||
user.nickname = nickname
|
||
user.save()
|
||
return SuccessResponse(msg="success")
|
||
|
||
#前端app头像修改
|
||
# class ChangeAvatarView(APIView):
|
||
# '''
|
||
# 前端app头像修改
|
||
# post:
|
||
# 【功能描述】前端app头像修改</br>
|
||
# 【参数说明】无，需要登录携带token后才能调用</br>
|
||
# '''
|
||
# authentication_classes = [JWTAuthentication]
|
||
# permission_classes = [IsAuthenticated]
|
||
|
||
# def post(self, request, *args, **kwargs):
|
||
# result = ImageUpload(request,"avatar")
|
||
# if result['code'] == 200 :
|
||
# user = request.user
|
||
# user.avatar = result['img'][0]
|
||
# user.save()
|
||
# return SuccessResponse(data=result['img'],msg=result['msg'])
|
||
# else:
|
||
# return ErrorResponse(msg=result['msg'])
|
||
|
||
#注销账号(标记已注销)
|
||
class DestroyUserView(APIView):
|
||
'''
|
||
注销账号(标记已注销)
|
||
post:
|
||
【功能描述】注销账号(标记已注销)</br>
|
||
'''
|
||
authentication_classes = [JWTAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def post(self, request, *args, **kwargs):
|
||
user = request.user
|
||
if user.identity not in [2]:
|
||
return ErrorResponse(msg="该用户不支持注销")
|
||
if '(已注销)' in user.username:
|
||
return ErrorResponse(msg="该用户已注销或不支持注销")
|
||
with transaction.atomic():
|
||
randstr = getRandomSet(6)
|
||
user.username = user.username + "(已注销)" + randstr
|
||
user.mobile = user.mobile + "(已注销)" + randstr
|
||
user.is_delete = True
|
||
user.is_active = False
|
||
user.save()
|
||
OAuthWXUser.objects.filter(user=user).delete()
|
||
return SuccessResponse(data={},msg="success")
|
||
|
||
#前端APP下载页面
|
||
def downloadapp(request):
|
||
instance = SystemConfig.objects.filter(key='app').first()
|
||
queryset = SystemConfig.objects.filter(parent=instance)
|
||
print(queryset)
|
||
data = {"name": "", 'logo': "", "apkurl": ""}
|
||
if queryset:
|
||
for m in queryset:
|
||
if m.key == "apk":
|
||
data['apkurl'] = m.value
|
||
elif m.key == "pkglogo":
|
||
data['logo'] = m.value
|
||
elif m.key == "pkgname":
|
||
data['name'] = m.value
|
||
return render(request, "download-app/index.html", {'data': data})
|
||
|
||
#h5端页面
|
||
def h5web(request):
|
||
return render(request,"h5/index.html")
|