Added UpdateSeriesView, DeleteSeriesView

src/series/views/__init__.py

@@ -1,2 +1,4 @@
 from .homepage import homepage_view
-from .add_series import AddSeriesView
+from .add_series import AddSeriesView
+from .update_series import UpdateSeriesView
+from .delete_series import SeriesDeleteView

src/series/views/add_series.py

@@ -24,7 +24,7 @@ class AddSeriesView(LoginRequiredMixin, CreateView):
         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.')
+            messages.info(self.request, 'TV Series can not added. Please try again.')
             return redirect('add-series')
         data = raw_data.json()
 
@@ -49,5 +49,4 @@ class AddSeriesView(LoginRequiredMixin, CreateView):
 
         series.save()
         form.save_m2m()
-
         return super().form_valid(form)

src/series/views/delete_series.py

@@ -0,0 +1,17 @@
+from series.models import SeriesModel
+from django.views.generic import DeleteView
+from django.urls import reverse_lazy, reverse
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.contrib import messages
+
+
+class SeriesDeleteView(LoginRequiredMixin, DeleteView):
+    login_url = reverse_lazy('login')
+    template_name = 'article_deletion_confirmation.html'
+
+    def get_success_url(self):
+        messages.success(self.request, 'Series Deleted')
+        return reverse('homepage')
+
+    def get_queryset(self):
+        return SeriesModel.objects.filter(slug=self.kwargs['slug'], user=self.request.user)

src/series/views/update_series.py

@@ -0,0 +1,56 @@
+from series.models import SeriesModel
+from django.views.generic import UpdateView
+from django.urls import reverse, reverse_lazy
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.contrib import messages
+from django.shortcuts import (get_object_or_404,
+                            redirect)
+from requests import get
+
+
+class UpdateSeriesView(LoginRequiredMixin, UpdateView):
+    login_url = reverse_lazy('login')
+    template_name = 'update_series.html'
+    fields = ('title', 'imdb_id', 'last_season', 'last_episode', 'show')
+
+    def get_object(self):
+        return get_object_or_404(SeriesModel, slug=self.kwargs.get('slug'), user=self.request.user)
+
+    def get_success_url(self):
+        messages.success(self.request, 'Series Updated')
+        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 can not updated. 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)
+