-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbriefingsView.py
118 lines (83 loc) · 4.84 KB
/
briefingsView.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
from django.contrib.auth import login, authenticate
from qvsta_server.qvsta_api.serializers.briefingSerializers.briefingsSerializer import BriefingsSerializer
from qvsta_server.qvsta_api.serializers.briefingSerializers.briefingAgenciesSerializer import BriefingDuplicate
from qvsta_server.qvsta_api.models import Briefings, BriefingModelRequests, BriefingAgencies, ModelRequestLooks, ModelRequestDetails, ModelRequestFiles
from rest_framework import status, response, mixins, generics, viewsets, serializers
from rest_framework.response import Response
from rest_framework.decorators import api_view
from rest_framework.views import APIView
from django.http import Http404
from rest_framework.filters import SearchFilter
from django_filters.rest_framework import DjangoFilterBackend
from django.core.mail import send_mail
from rest_framework.authtoken.views import ObtainAuthToken
from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from django.contrib.auth.models import User
from rest_framework.authtoken.models import Token
from rest_framework import permissions
from rest_framework.permissions import IsAuthenticated
from django.http import HttpResponseForbidden
class BriefingsList(generics.ListCreateAPIView):
queryset = Briefings.objects.all()
serializer_class = BriefingsSerializer
filter_backends = (DjangoFilterBackend,)
filter_fields = ('status',)
permission_classes = (IsAuthenticated,)
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
briefing = serializer.save()
briefing.creatingUserID = self.request.user
briefing.clientID = self.request.user.companyID
briefing.save()
headers = self.get_success_headers(serializer.data)
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
def get_queryset(self):
"""
Optionally restricts the returned purchases to a given user,
by filtering against a `username` query parameter in the URL.
"""
queryset = Briefings.objects.all()
queryset = queryset.filter(clientID = self.request.user.companyID.companyID)
return queryset
class BriefingsDetail(generics.RetrieveUpdateDestroyAPIView):
queryset = Briefings.objects.all()
serializer_class = BriefingsSerializer
def destroy(self, request, *args, **kwargs):
instance = self.get_object()
briefingID = instance.briefingsID
BriefingAgencies.objects.filter(briefingID = briefingID).delete()
modelRequestIDs = BriefingModelRequests.objects.filter(briefingID = briefingID).values_list('modelRequestID', flat = True)
modelRequestList = list(set(modelRequestIDs))
for modelRequest in modelRequestList:
ModelRequestDetails.objects.filter(modelRequestID = modelRequest).delete()
ModelRequestLooks.objects.filter(modelRequestID = modelRequest).delete()
ModelRequestFiles.objects.filter(modelRequestID = modelRequest).delete()
BriefingModelRequests.objects.filter(briefingID = briefingID).delete()
Briefings.objects.get(briefingsID = briefingID).delete()
return Response(status=status.HTTP_204_NO_CONTENT)
def patch(self, request, *args, **kwargs):
return HttpResponseForbidden()
# class BriefingsDuplicate(generics.CreateAPIView):
# serializer_class = BriefingDuplicate
# def create(self, request, *args, **kwargs):
# serializer = self.get_serializer(data=request.data)
# serializer.is_valid(raise_exception=True)
# briefing = serializer.validated_data.get('briefingID')
# briefing.briefingsID = None
# briefing.creatingUserID = self.request.user
# briefing.clientID = self.request.user.companyID
# briefing.save()
# briefingID = briefing.briefingsID
# briefingAgenciesID = BriefingAgencies.objects.filter(briefingID = briefing.briefingsID).values_list('briefingAgenciesID', flat = True)
# briefingAgenciesList = list(set(briefingAgenciesID))
# for briefingAgency in briefingAgenciesList:
# briefingAgent = BriefingAgencies.objects.get(briefingAgenciesID = briefingAgency)
# briefingAgent.briefingAgenciesID = None
# briefingAgent.save()
# briefingModelRequestsID = BriefingAgencies.objects.filter(briefingID = briefing.briefingsID).values_list('briefingAgenciesID', flat = True)
# briefingModelRequestsList = list(set(briefingModelRequestsID))
# for briefingModelRequest in briefingModelRequestsList:
# headers = self.get_success_headers(serializer.data)
# return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)