From 84da4546128525b69a8829aba21209e29c0720a3 Mon Sep 17 00:00:00 2001
From: Jens Langhammer <jens.langhammer@beryju.org>
Date: Fri, 25 Sep 2020 17:43:17 +0200
Subject: [PATCH] stages/otp_: ensure stage.configure_flow is set

---
 .../otp_static/migrations/0003_default_setup_flow.py.py       | 4 ++++
 .../otp_static/templates/stages/otp_static/user_settings.html | 2 +-
 .../stages/otp_time/migrations/0004_default_setup_flow.py     | 4 ++++
 .../otp_time/templates/stages/otp_time/user_settings.html     | 2 +-
 4 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/passbook/stages/otp_static/migrations/0003_default_setup_flow.py.py b/passbook/stages/otp_static/migrations/0003_default_setup_flow.py.py
index fa2e1f129..6bad90c52 100644
--- a/passbook/stages/otp_static/migrations/0003_default_setup_flow.py.py
+++ b/passbook/stages/otp_static/migrations/0003_default_setup_flow.py.py
@@ -29,6 +29,10 @@ def create_default_setup_flow(apps: Apps, schema_editor: BaseDatabaseSchemaEdito
         target=flow, stage=stage, defaults={"order": 0}
     )
 
+    for stage in OTPStaticStage.objects.using(db_alias).filter(configure_flow=None):
+        stage.configure_flow = flow
+        stage.save()
+
 
 class Migration(migrations.Migration):
 
diff --git a/passbook/stages/otp_static/templates/stages/otp_static/user_settings.html b/passbook/stages/otp_static/templates/stages/otp_static/user_settings.html
index a54270c35..66081c593 100644
--- a/passbook/stages/otp_static/templates/stages/otp_static/user_settings.html
+++ b/passbook/stages/otp_static/templates/stages/otp_static/user_settings.html
@@ -26,7 +26,7 @@
         </ul>
         {% if not state %}
             {% if stage.configure_flow %}
-                <a href="{% url 'passbook-flows:configure' stage_uuid=stage.stage_uuid %}" class="pf-c-button pf-m-primary">{% trans "Enable Static Tokens" %}</a>
+                <a href="{% url 'passbook_flows:configure' stage_uuid=stage.stage_uuid %}" class="pf-c-button pf-m-primary">{% trans "Enable Static Tokens" %}</a>
             {% endif %}
         {% else %}
         <a href="{% url 'passbook_stages_otp_static:disable' stage_uuid=stage.stage_uuid %}" class="pf-c-button pf-m-danger">{% trans "Disable Static Tokens" %}</a>
diff --git a/passbook/stages/otp_time/migrations/0004_default_setup_flow.py b/passbook/stages/otp_time/migrations/0004_default_setup_flow.py
index ee685b73b..46f019062 100644
--- a/passbook/stages/otp_time/migrations/0004_default_setup_flow.py
+++ b/passbook/stages/otp_time/migrations/0004_default_setup_flow.py
@@ -30,6 +30,10 @@ def create_default_setup_flow(apps: Apps, schema_editor: BaseDatabaseSchemaEdito
         target=flow, stage=stage, defaults={"order": 0}
     )
 
+    for stage in OTPTimeStage.objects.using(db_alias).filter(configure_flow=None):
+        stage.configure_flow = flow
+        stage.save()
+
 
 class Migration(migrations.Migration):
 
diff --git a/passbook/stages/otp_time/templates/stages/otp_time/user_settings.html b/passbook/stages/otp_time/templates/stages/otp_time/user_settings.html
index 71789eee2..d3d0fd9db 100644
--- a/passbook/stages/otp_time/templates/stages/otp_time/user_settings.html
+++ b/passbook/stages/otp_time/templates/stages/otp_time/user_settings.html
@@ -22,7 +22,7 @@
         <p>
             {% if not state %}
                 {% if stage.configure_flow %}
-                    <a href="{% url 'passbook-flows:configure' stage_uuid=stage.stage_uuid %}" class="pf-c-button pf-m-primary">{% trans "Enable Time-based OTP" %}</a>
+                    <a href="{% url 'passbook_flows:configure' stage_uuid=stage.stage_uuid %}" class="pf-c-button pf-m-primary">{% trans "Enable Time-based OTP" %}</a>
                 {% endif %}
             {% else %}
             <a href="{% url 'passbook_stages_otp_time:disable' stage_uuid=stage.stage_uuid %}" class="pf-c-button pf-m-danger">{% trans "Disable Time-based OTP" %}</a>