ly_baoxiu_admin/apps/admin/config/views.py

# coding=utf-8
import json
from django.db import transaction

from utils.permission import IsAdministratorUser, IsTenantUser
from apps.tenant.config.serializers import ConfigSerializer
from utils.custom_modelviewset import CustomModelViewSet
from apps.tenant.config.models import Config
from apps.tenant.config.filters import ConfigFilter
from apps.log.models import BizLog
from utils import response_ok, response_error


class ConfigViewSet(CustomModelViewSet):
    # TODO
    permission_classes = [IsTenantUser, ]
    queryset = Config.objects.filter()
    serializer_class = ConfigSerializer

    # def filter_queryset(self, queryset):
    #     queryset = queryset.filter()
    #     f = ConfigFilter(self.request.GET, queryset=queryset)
    #     return f.qs

    # 不分页用list
    def list(self, request, *args, **kwargs):
        queryset = self.filter_queryset(self.get_queryset())
        serializer = self.get_serializer(queryset, many=True)
        return response_ok(serializer.data)

    def create(self, request, *args, **kwargs):
        data = json.loads(request.POST.get('data'))
        with transaction.atomic():
            for item in data:
                config = Config.objects.filter(property=item['key']).first()
                if config:
                    config.value = item['value']
                    config.save()
                else:
                    Config.objects.create(property=item['key'], value=item['value'])
            BizLog.objects.addnew(None, request.user, BizLog.INSERT, u'修改系统设置', data)
        return response_ok()