API_clien app removed

src/series/views/__init__.py

@@ -1 +1,2 @@
-from .homepage import homepage_view
+from .homepage import homepage_view
+from .add_series import AddSeriesView

src/series/views/add_series.py

@@ -0,0 +1,53 @@
+from django.views.generic import CreateView
+from series.models import SeriesModel
+from django.urls import reverse_lazy, reverse
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.contrib import messages
+from django.shortcuts import redirect
+from requests import get
+
+
+class AddSeriesView(LoginRequiredMixin, CreateView):
+    login_url = reverse_lazy('login')
+    template_name = 'add_series.html'
+    model = SeriesModel
+    fields = ('title', 'imdb_id', 'last_season', 'last_episode', 'show')
+
+    def get_success_url(self):
+        messages.success(self.request, 'Series Added')
+        return reverse('homepage')
+
+    def form_valid(self, form):
+        series = form.save(commit=False)
+        series.user = self.request.user
+
+        raw_data = get(f"https://imdb-api.com/en/API/Title/{series.user.imdb_api_key}/{series.imdb_id}")
+
+        if raw_data.status_code != 200:
+            messages.info(self.request, 'TV Series no added. Please try again.')
+            return redirect('add-series')
+        data = raw_data.json()
+
+        if data['errorMessage']:
+            form.add_error('imdb_id', 'ID is not correct.')
+            return self.form_invalid(form)
+
+        if not data['tvSeriesInfo']:
+            form.add_error('imdb_id', 'This is not a TV series id.')
+            return self.form_invalid(form)
+        seasons = data['tvSeriesInfo']['seasons']
+
+        if str(series.last_season) not in seasons:
+            form.add_error('last_season', 'The season number is not correct.')
+            return self.form_invalid(form)
+
+        data = get(f"https://imdb-api.com/en/API/SeasonEpisodes/{series.user.imdb_api_key}/{series.imdb_id}/{series.last_season}").json()
+        episodes_count = len(data['episodes'])
+        if series.last_episode > episodes_count:
+            form.add_error('last_episode', 'The episode number is not correct.')
+            return self.form_invalid(form)
+
+        series.save()
+        form.save_m2m()
+
+        return super().form_valid(form)