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.
114 lines
4.5 KiB
114 lines
4.5 KiB
10 months ago
|
# -*- coding: utf-8 -*-
|
||
|
|
# @Author : Sherlock
|
||
|
|
# @Time : 2025/2/21 17:53
|
||
|
|
# @Description : 数据连接
|
||
|
|
from rest_framework.permissions import IsAuthenticated
|
||
|
|
from rest_framework_jwt.authentication import JSONWebTokenAuthentication
|
||
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
||
|
|
from rest_framework.decorators import action
|
||
|
|
from rest_framework.filters import SearchFilter, OrderingFilter
|
||
|
|
|
||
|
|
|
||
|
|
class ConnectorViewSet(CustomViewBase):
|
||
|
|
'''
|
||
|
|
字典详情树
|
||
|
|
'''
|
||
|
|
perms_map = (
|
||
|
|
{'*': 'admin'},
|
||
|
|
{'*': 'connector_all'},
|
||
|
|
{'get': 'connector_list'},
|
||
|
|
{'post': 'connector_create'},
|
||
|
|
{'put': 'connector_edit'},
|
||
|
|
{'delete': 'connector_delete'}
|
||
|
|
)
|
||
|
|
queryset = Connection.objects.all()
|
||
|
|
serializer_class = ConnectionSerializer
|
||
|
|
filter_backends = (SearchFilter, OrderingFilter)
|
||
|
|
ordering_fields = ('id',)
|
||
|
|
authentication_classes = (JSONWebTokenAuthentication,)
|
||
|
|
permission_classes = (IsAuthenticated,)
|
||
|
|
|
||
|
|
def list(self, request, *args, **kwargs):
|
||
|
|
'''
|
||
|
|
重写list方法
|
||
|
|
'''
|
||
|
|
companyMid = request.GET.get("companyMid", None)
|
||
|
|
is_enabled = request.GET.get("is_enabled")
|
||
|
|
label = request.GET.get("label")
|
||
|
|
order = request.GET.get("order")
|
||
|
|
sort = request.GET.get("ordering")
|
||
|
|
order_by = 'id'
|
||
|
|
try:
|
||
|
|
|
||
|
|
queryRows = Connection.objects.filter(companyMid=companyMid).order_by(order_by)
|
||
|
|
for item in queryRows:
|
||
|
|
item.__dict__.pop('_state')
|
||
|
|
results.append(item.__dict__)
|
||
|
|
result = []
|
||
|
|
return CCAIResponse(results)
|
||
|
|
else:
|
||
|
|
logger.error("connector list failed: \n%s" % traceback.format_exc())
|
||
|
|
return CCAIResponse(PARAMS_ERR, SERVER_ERROR)
|
||
|
|
|
||
|
|
def create(self, request, *args, **kwargs):
|
||
|
|
'''添加数据POST'''
|
||
|
|
try:
|
||
|
|
data = req_operate_by_user(request)
|
||
|
|
# data['dict_detail_id'] = uuid.uuid4().__str__()
|
||
|
|
|
||
|
|
serializer = self.get_serializer(data=data)
|
||
|
|
serializer.is_valid(raise_exception=True)
|
||
|
|
self.perform_create(serializer)
|
||
|
|
|
||
|
|
|
||
|
|
return CCAIResponse(data="success")
|
||
|
|
except Exception as e:
|
||
|
|
logger.error("connector create failed: \n%s" % traceback.format_exc())
|
||
|
|
return CCAIResponse("connector create failed", SERVER_ERROR)
|
||
|
|
|
||
|
|
def destroy(self, request, *args, **kwargs):
|
||
|
|
try:
|
||
|
|
instance = self.get_object()
|
||
|
|
# 查询是否存在关联
|
||
|
|
queryRows = Connection.objects.filter(id=instance.id)
|
||
|
|
if len(queryRows > 0):
|
||
|
|
# return CCAIResponse("该数据存在关联,请先删除关联数据", BAD)
|
||
|
|
# is_association = self.check_association(deptId=instance.id)
|
||
|
|
# if is_association:
|
||
|
|
# return CCAIResponse(is_association, BAD)
|
||
|
|
self.perform_destroy(instance)
|
||
|
|
|
||
|
|
return CCAIResponse(data="delete connector success")
|
||
|
|
else:
|
||
|
|
logger.error("connector id failed: \n%s" % traceback.format_exc())
|
||
|
|
return CCAIResponse("delete connector id inexistence ", BAD)
|
||
|
|
except Exception as e:
|
||
|
|
logger.error("user: %s, delete dept failed: \n%s" % (request.user.id, traceback.format_exc()))
|
||
|
|
return CCAIResponse("delete dept failed", SERVER_ERROR)
|
||
|
|
|
||
|
|
|
||
|
|
def update(self, request, *args, **kwargs):
|
||
|
|
from rest_framework.exceptions import ValidationError
|
||
|
|
try:
|
||
|
|
companyMid = request.query_params.get('companyMid', None)
|
||
|
|
data = req_operate_by_user(request)
|
||
|
|
data['companyMid'] = companyMid
|
||
|
|
partial = kwargs.pop('partial', False) # True:所有字段全部更新， False:仅更新提供的字段
|
||
|
|
instance = self.get_object()
|
||
|
|
|
||
|
|
|
||
|
|
serializer = self.get_serializer(instance, data=data, partial=partial)
|
||
|
|
serializer.is_valid(raise_exception=True)
|
||
|
|
self.perform_update(serializer)
|
||
|
|
|
||
|
|
if getattr(instance, '_prefetched_objects_cache', None):
|
||
|
|
# If 'prefetch_related' has been applied to a queryset, we need to
|
||
|
|
# forcibly invalidate the prefetch cache on the instance.
|
||
|
|
instance._prefetched_objects_cache = {}
|
||
|
|
return CCAIResponse(data="success")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
logger.error("user: %s, update connector failed: \n%s" % (request.user.id, traceback.format_exc()))
|
||
|
|
return CCAIResponse("update dept failed", SERVER_ERROR)
|
||
|
|
|