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.
60 lines
2.3 KiB
60 lines
2.3 KiB
3 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding:utf-8 -*-
|
||
|
|
# project: fir_ser
|
||
|
|
# filename: getip
|
||
|
|
# author: liuyu
|
||
|
|
# data: 2022/3/28
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from rest_framework.response import Response
|
||
|
|
from rest_framework.views import APIView
|
||
|
|
|
||
|
|
from api.models import UserPersonalConfig
|
||
|
|
from api.utils.response import BaseResponse
|
||
|
|
from api.utils.serializer import PersonalConfigSerializer
|
||
|
|
from common.core.auth import ExpiringTokenAuthentication, SuperSignPermission
|
||
|
|
from common.core.sysconfig import UserConfig, Config
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class PersonalConfigView(APIView):
|
||
|
|
authentication_classes = [ExpiringTokenAuthentication, ]
|
||
|
|
permission_classes = [SuperSignPermission, ]
|
||
|
|
|
||
|
|
def get(self, request):
|
||
|
|
res = BaseResponse()
|
||
|
|
|
||
|
|
config_key = request.query_params.get("config_key", None)
|
||
|
|
developer_personal_config = Config.DEVELOPER_STATUS_CONFIG
|
||
|
|
personal_config_queryset = UserPersonalConfig.objects.filter(user_id=request.user,
|
||
|
|
key__in=developer_personal_config)
|
||
|
|
if personal_config_queryset.count() != len(developer_personal_config):
|
||
|
|
for key in developer_personal_config:
|
||
|
|
UserConfig(request.user).set_default_value(key)
|
||
|
|
if config_key:
|
||
|
|
personal_config_queryset = personal_config_queryset.filter(key=config_key)
|
||
|
|
|
||
|
|
personal_config_serializer = PersonalConfigSerializer(personal_config_queryset, many=True)
|
||
|
|
res.data = personal_config_serializer.data
|
||
|
|
res.count = personal_config_queryset.count()
|
||
|
|
|
||
|
|
return Response(res.dict)
|
||
|
|
|
||
|
|
def put(self, request):
|
||
|
|
res = BaseResponse()
|
||
|
|
config_key = request.data.get("config_key", None)
|
||
|
|
config_value = request.data.get("config_value", None)
|
||
|
|
if config_key is not None and config_value is not None and config_key in Config.DEVELOPER_STATUS_CONFIG:
|
||
|
|
UserPersonalConfig.objects.filter(user_id=request.user, key=config_key).update(value=config_value)
|
||
|
|
return Response(res.dict)
|
||
|
|
|
||
|
|
def delete(self, request):
|
||
|
|
res = BaseResponse()
|
||
|
|
developer_personal_config = Config.DEVELOPER_STATUS_CONFIG
|
||
|
|
for key in developer_personal_config:
|
||
|
|
UserConfig(request.user).del_value(key)
|
||
|
|
Config.invalid_config_cache(f'{key}_DES')
|
||
|
|
return Response(res.dict)
|