feat: 修改作业权限

This commit is contained in:
Aaron3S
2022-12-15 17:25:21 +08:00
parent 9d80abadd8
commit e969a01689
12 changed files with 79 additions and 47 deletions

View File

@@ -2,11 +2,7 @@ import os
import zipfile
from django.conf import settings
from rest_framework_bulk import BulkModelViewSet
from common.mixins import CommonApiMixin
from orgs.mixins.api import OrgBulkModelViewSet
from .base import SelfBulkModelViewSet
from ..exception import PlaybookNoValidEntry
from ..models import Playbook
from ..serializers.playbook import PlaybookSerializer
@@ -20,11 +16,16 @@ def unzip_playbook(src, dist):
fz.extract(file, dist)
class PlaybookViewSet(SelfBulkModelViewSet):
class PlaybookViewSet(OrgBulkModelViewSet):
serializer_class = PlaybookSerializer
permission_classes = ()
model = Playbook
def get_queryset(self):
queryset = super().get_queryset()
queryset = queryset.filter(creator=self.request.user)
return queryset
def perform_create(self, serializer):
instance = serializer.save()
src_path = os.path.join(settings.MEDIA_ROOT, instance.path.name)