#!/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)