mirror of
https://github.com/spbleadersofdigtal/backend.git
synced 2024-12-01 06:43:43 +03:00
111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
from rest_framework import generics, status
|
|
from rest_framework.generics import get_object_or_404
|
|
from rest_framework.parsers import FormParser, JSONParser, MultiPartParser
|
|
from rest_framework.response import Response
|
|
|
|
from pitch_deck_generator.decks.api.serializers import (
|
|
AnswerSerializer,
|
|
BasePitchDeckSerializer,
|
|
HintSerializer,
|
|
PitchDeckPresentationSerializer,
|
|
PitchDeckSerializer,
|
|
QuestionSerializer,
|
|
)
|
|
from pitch_deck_generator.decks.models import PitchDeck, Question, QuestionDeckHint
|
|
|
|
|
|
class ListDecksApiView(generics.ListCreateAPIView):
|
|
queryset = PitchDeck.objects.all()
|
|
serializer_class = BasePitchDeckSerializer
|
|
|
|
|
|
class RetrievePitchApiView(generics.RetrieveAPIView):
|
|
serializer_class = PitchDeckSerializer
|
|
|
|
def get_object(self):
|
|
return get_object_or_404(PitchDeck, id=self.kwargs["id"])
|
|
|
|
|
|
class GetFirstQuestionApiView(generics.RetrieveAPIView):
|
|
serializer_class = QuestionSerializer
|
|
|
|
def get_object(self):
|
|
return Question.objects.get(order=1)
|
|
|
|
|
|
class GetDeckQuestionApiView(generics.RetrieveAPIView):
|
|
serializer_class = QuestionSerializer
|
|
|
|
def get_object(self):
|
|
return get_object_or_404(Question, id=self.kwargs["question_id"])
|
|
|
|
|
|
class CreateQuestionAnswerApiView(generics.CreateAPIView):
|
|
serializer_class = AnswerSerializer
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
# main thing starts
|
|
file_fields = list(request.FILES.keys()) # list to be passed to the serializer
|
|
serializer = self.get_serializer(data=request.data, file_fields=file_fields)
|
|
# main thing ends
|
|
|
|
serializer.is_valid(raise_exception=True)
|
|
self.perform_create(serializer)
|
|
headers = self.get_success_headers(serializer.data)
|
|
return Response(
|
|
serializer.data, status=status.HTTP_201_CREATED, headers=headers
|
|
)
|
|
|
|
|
|
class GetDeckQuestionHintApiView(generics.GenericAPIView):
|
|
serializer_class = HintSerializer
|
|
parser_classes = [JSONParser, FormParser, MultiPartParser]
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
hint = get_object_or_404(
|
|
QuestionDeckHint,
|
|
question_id=self.kwargs["question_id"],
|
|
deck_id=self.kwargs["deck_id"],
|
|
)
|
|
data = hint.hint
|
|
if data:
|
|
return Response(data)
|
|
return Response(status=404)
|
|
|
|
|
|
class GetDeckPresentationDataApiView(generics.GenericAPIView):
|
|
serializer_class = PitchDeckPresentationSerializer
|
|
|
|
structure = {
|
|
1: ["names", "type"],
|
|
2: ["problems"],
|
|
3: ["actuality", "awards"],
|
|
4: ["solve", "works"],
|
|
5: ["market_values", "users"],
|
|
6: ["competitors", "competitors_strength", "competitors_low", "advantages"],
|
|
7: ["money", "finance_model"],
|
|
8: ["how_much_investments", "financial_indicators"],
|
|
9: ["your_role", "your_teammates", "past_investors"],
|
|
10: ["how_much_investments", "time_to_spend", "investments_sold"],
|
|
11: ["company_value", "future_value", "time_to_spend"],
|
|
12: ["aims"],
|
|
13: ["images"],
|
|
}
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
deck = get_object_or_404(
|
|
PitchDeck,
|
|
id=self.kwargs["deck_id"],
|
|
)
|
|
resp = []
|
|
data = deck.questions
|
|
for slide, tags in self.structure.items():
|
|
slide_data = {"slide": slide, "data": []}
|
|
for tag in tags:
|
|
slide_data["data"].append(
|
|
{"slug": tag, "answer": data[tag] if tag in data else {}}
|
|
)
|
|
resp.append(slide_data)
|
|
|
|
return Response(resp)
|