From fe424e9d9dac8731d3d200d2883f5be830881409 Mon Sep 17 00:00:00 2001
From: Christophe Siraut <d@tobald.eu.org>
Date: Thu, 7 Nov 2024 08:09:13 +0100
Subject: [PATCH] misc: enable adding new episodes

---
 aircox/templates/aircox/program_form.html | 2 ++
 aircox/urls.py                            | 3 +++
 aircox/views/episode.py                   | 9 +++++++++
 3 files changed, 14 insertions(+)

diff --git a/aircox/templates/aircox/program_form.html b/aircox/templates/aircox/program_form.html
index dd89aca..c18e15b 100644
--- a/aircox/templates/aircox/program_form.html
+++ b/aircox/templates/aircox/program_form.html
@@ -8,6 +8,8 @@
 
 {% block page-form-actions %}
 <a class="button secondary" href="{% url 'episode-list' parent_slug=object.slug %}">{% trans "Episodes" %}</a>
+&nbsp;
+<a class="button secondary" href="{% url 'program-add-episode' object.pk %}">{% trans "New episode" %}</a>
 {% if object and object.pk and request.user.is_superuser %}
 &nbsp;
 <button type="button"
diff --git a/aircox/urls.py b/aircox/urls.py
index 172401e..2fc509c 100755
--- a/aircox/urls.py
+++ b/aircox/urls.py
@@ -127,6 +127,9 @@ urls = [
     # ---- dashboard
     path(_("dashboard/"), views.dashboard.DashboardView.as_view(), name="dashboard"),
     path(_("dashboard/program/<pk>/"), views.program.ProgramUpdateView.as_view(), name="program-edit"),
+    path(
+        _("dashboard/program/<pk>/add-episode/"), views.episode.EpisodeCreateView.as_view(), name="program-add-episode"
+    ),
     path(_("dashboard/episodes/<pk>/"), views.episode.EpisodeUpdateView.as_view(), name="episode-edit"),
     path(_("dashboard/statistics/"), views.dashboard.StatisticsView.as_view(), name="dashboard-statistics"),
     path(_("dashboard/statistics/<date:date>/"), views.dashboard.StatisticsView.as_view(), name="dashboard-statistics"),
diff --git a/aircox/views/episode.py b/aircox/views/episode.py
index b9b42cd..f9d2dfc 100644
--- a/aircox/views/episode.py
+++ b/aircox/views/episode.py
@@ -1,5 +1,7 @@
 from django.contrib.auth.mixins import UserPassesTestMixin
+from django.shortcuts import get_object_or_404
 from django.urls import reverse
+from django.utils.translation import gettext_lazy as _
 
 from aircox.models import Episode, Program, StaticPage, Track
 from aircox import forms, filters, permissions
@@ -12,6 +14,7 @@ __all__ = (
     "EpisodeDetailView",
     "EpisodeListView",
     "PodcastListView",
+    "EpisodeCreateView",
     "EpisodeUpdateView",
 )
 
@@ -139,3 +142,9 @@ class EpisodeUpdateView(UserPassesTestMixin, VueFormDataMixin, PageUpdateView):
         if invalid:
             return self.get(request, **formsets)
         return resp
+
+
+class EpisodeCreateView(EpisodeUpdateView):
+    def get_object(self):
+        program = get_object_or_404(Program, pk=self.kwargs["pk"])
+        return Episode.objects.create(program=program, title="%s (%s)" % (program.title, _("draft")))