elections2021: Menu update

elections2021/migrations/0049_auto_20210930_2104.py

+# Generated by Django 3.2.6 on 2021-09-30 19:04
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("wagtailcore", "0062_comment_models_and_pagesubscription"),
+        ("elections2021", "0048_elections2021goingtovotepage_header_text"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="elections2021homepage",
+            name="post_election_strategy",
+            field=models.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.SET_NULL,
+                related_name="+",
+                to="wagtailcore.page",
+                verbose_name="stránka povolební strategie",
+            ),
+        ),
+        migrations.AddField(
+            model_name="elections2021homepage",
+            name="program_for_business",
+            field=models.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.SET_NULL,
+                related_name="+",
+                to="wagtailcore.page",
+                verbose_name="stránka program pro podnikatele",
+            ),
+        ),
+    ]

elections2021/models.py

@@ -231,6 +231,22 @@ class Elections2021HomePage(MetadataPageMixin, RoutablePageMixin, Page):
         on_delete=models.SET_NULL,
         related_name="+",
     )
+    program_for_business = models.ForeignKey(
+        "wagtailcore.Page",
+        verbose_name="stránka program pro podnikatele",
+        null=True,
+        blank=True,
+        on_delete=models.SET_NULL,
+        related_name="+",
+    )
+    post_election_strategy = models.ForeignKey(
+        "wagtailcore.Page",
+        verbose_name="stránka povolební strategie",
+        null=True,
+        blank=True,
+        on_delete=models.SET_NULL,
+        related_name="+",
+    )
     matomo_id = models.IntegerField(
         "Matomo ID pro sledování návštěvnosti", blank=True, null=True
     )
@@ -265,6 +281,12 @@ class Elections2021HomePage(MetadataPageMixin, RoutablePageMixin, Page):
         PageChooserPanel(
             "program_silver_page", "elections2021.Elections2021StrategicPage"
         ),
+        PageChooserPanel(
+            "program_for_business", "elections2021.Elections2021StrategicPage"
+        ),
+        PageChooserPanel(
+            "post_election_strategy", "elections2021.Elections2021StrategicPage"
+        ),
     ]
 
     footer_panels = [
@@ -444,6 +466,18 @@ class Elections2021HomePage(MetadataPageMixin, RoutablePageMixin, Page):
             return self.program_silver_page.url
         return "#"
 
+    @cached_property
+    def program_for_business_url(self):
+        if self.program_for_business:
+            return self.program_for_business.url
+        return "#"
+
+    @cached_property
+    def post_election_strategy_url(self):
+        if self.post_election_strategy:
+            return self.post_election_strategy.url
+        return "#"
+
     @cached_property
     def newsletter_subscribe_url(self):
         return self.url + self.reverse_subpage("newsletter_subscribe")

elections2021/templates/elections2021/base.html

@@ -119,7 +119,11 @@
                       <li><a href="{{ page.root_page.program_strategic_page_url }}" class="navbar-menu__link">Program v 5 minutách</a></li>
                       <li><a href="{{ page.root_page.program_silver_page_url }}" class="navbar-menu__link">Stříbrný program pro seniory</a></li>
                       <li><a href="{{ page.root_page.going_to_vote_page_url }}" class="navbar-menu__link">Program pro mladé</a></li>
+                      <li><a href="{{ page.root_page.program_for_business_url }}" class="navbar-menu__link">Program pro podnikatele</a></li>
                       <li><a href="{{ page.root_page.full_program_page_url }}" class="navbar-menu__link">Program přístupný pro lidi se zrakovým postižením</a></li>
+                      {% if page.root_page.post_election_strategy_url != "#" %}
+                        <li><a href="{{ page.root_page.post_election_strategy_url }}" class="navbar-menu__link">Povolební strategie</a></li>
+                      {% endif %}
                     </ul>
                   </ui-navbar-subtitem>
                 </li>