def multiple_delete(self, request, *args, **kwargs):
selectids = request.query_params.get('selectId', None)
if not selectids:
return CustomResponse(status=status.HTTP_404_NOT_FOUND)
selectid = selectids.split(',')
selectid = [int(x) for x in selectid if x.split()]
models.WebSite.objects.filter(id__in=selectid).delete()
return CustomResponse(status=status.HTTP_204_NO_CONTENT)
def post(self, request, *args, **kwargs):
username = request.data.get('username')
password = request.data.get('password')
user = authenticate(username=username, password=password)
if user:
payload = jwt_payload_handler(user)
payload['is_superuser'] = user.is_superuser
return CustomResponse({'token': jwt_encode_handler(payload)},
status=status.HTTP_200_OK)
else:
return CustomResponse('用户名或密码错误!',
status=status.HTTP_400_BAD_REQUEST)
def change_password(self, request, *args, **kwargs):
id = kwargs.get('id')
password1 = request.data['password1']
password2 = request.data['password2']
user = models.UserInfo.objects.get(id=id)
if password1 == password2:
user.set_password(password2)
user.save()
return CustomResponse(msg="密码修改成功", status=status.HTTP_200_OK)
else:
return CustomResponse(msg="密码两次输入不一致!",
status=status.HTTP_400_BAD_REQUEST)
def create(self, request, *args, **kwargs):
for data in request.data:
serializer = self.get_serializer(data=data)
if not serializer.is_valid():
return CustomResponse(status=status.HTTP_400_BAD_REQUEST,
msg='form表单校验不通过',
data={
'data': serializer.data,
'error': serializer.errors
})
for data in request.data:
serializer = self.get_serializer(data=data)
serializer.is_valid()
self.perform_create(serializer)
return CustomResponse(
status=status.HTTP_201_CREATED,
msg='添加成功',
)