API_clien app removed

src/series/models.py

@@ -1,6 +1,6 @@
 from django.db import models
 from django.core.exceptions import ValidationError
-from API_client import APIClient
+# from API_client import APIClient
 from django.core.validators import (MaxValueValidator,
                                     MinValueValidator,
                                     MinLengthValidator)
@@ -21,22 +21,5 @@ class SeriesModel(models.Model):
         ], blank=False, null=False)
     show = models.BooleanField(default=True)
 
-    def clean(self) -> None:
-        if len(self.imdb_id) < 9:
-            raise ValidationError(f'Ensure this value has at least 9 characters (it has {len(self.imdb_id)}).')
-
-        if len(self.imdb_id) > 10:
-            raise ValidationError(f'Make sure this value is no more than 10 characters (it has {len(self.imdb_id)}).')
-
-        client = APIClient(self.user.imdb_api_key)
-
-        seasons = client.get_seasons(self.imdb_id)
-        if str(self.last_season) not in seasons:
-            raise ValidationError('The season number you entered is not correct.')
-
-        episodes_count = client.get_episodes_count(self.last_season, self.imdb_id)
-        if self.last_episode > episodes_count:
-            raise ValidationError('The episode number you entered is not correct.')
-
     def __str__(self) -> str:
         return f"{self.user.username} - {self.title}"

src/series/templates/add_series.html

@@ -0,0 +1,22 @@
+{% extends 'base.html' %}
+{% load crispy_forms_tags %}
+
+{% block title %} Add Series {% endblock title %}
+
+{% block content %}
+{% include 'components/message.html' %}
+
+<form enctype="multipart/form-data" method="POST">
+    {% csrf_token %}
+    {{form|crispy}}
+    <style>
+        .btn:hover {
+            background: #fff;
+            color: #218838;
+        }
+    </style>
+    <input type="submit" value="Add" class="btn btn-primary mt-3 mb-5">
+</form>
+
+{% endblock content %}
+    

src/series/urls.py

@@ -4,5 +4,5 @@ from series import views
 
 urlpatterns = [
     path('', views.homepage_view, name='homepage'),
-    # path('series/create', views.homepage_view, name='homepage'),
+    path('series/add', views.AddSeriesView.as_view(), name='add-series'),
 ]

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)