chore: add CronJob to recreate Django superuser for dev/test namespaces after DB reset

Author: Sepehr-Sobhani

dags/bc_obps_reset_data.py

@@ -74,7 +74,8 @@ def should_cycle_backend(**context):
 
 
 ROLLOUT_DAG_DOC = """
-DAG to wait for the backend rollout to complete. Intended to be triggered after the DB reset.
+DAG to wait for the backend rollout to complete, then recreate the django superuser that was
+dropped along with the database. Intended to be triggered after the DB reset.
 """
 
 
@@ -99,7 +100,12 @@ def wait_for_backend_rollout():
         wait_until_job_complete=True,
     )
 
-    wait_for_backend_rollout_task  # NOSONAR
+    # Must run after the rollout completes, since it hits the backend over its route
+    @task
+    def add_django_admin_task():
+        trigger_k8s_cronjob("add-django-admin", BCIERS_NAMESPACE)
+
+    wait_for_backend_rollout_task >> add_django_admin_task()
 
 
 reset_data()  # NOSONAR

helm/cas-bciers/templates/backend/job/add-django-admin.yaml

@@ -0,0 +1,44 @@
+{{- if or (hasSuffix "-dev" .Release.Namespace) (hasSuffix "-test" .Release.Namespace) }}
+
+apiVersion: batch/v1
+kind: CronJob
+metadata:
+  labels: {{- include "cas-bciers.labels" . | nindent 4 }}
+    component: job-with-database-access
+  name: add-django-admin
+  namespace: {{ .Release.Namespace }}
+  annotations:
+    "helm.sh/hook": "pre-install,pre-upgrade"
+    "helm.sh/hook-weight": "-10"
+spec:
+  suspend: true # This cron job is intended to be triggered manually by Airflow
+  schedule: "0 * * * *"
+  jobTemplate:
+    spec:
+      template:
+        metadata:
+          name: {{ template "cas-bciers.fullname" . }}-add-django-admin
+        spec:
+          activeDeadlineSeconds: 300
+          containers:
+            - name: {{ template "cas-bciers.fullname" . }}-add-django-admin
+              image: curlimages/curl
+              env:
+                - name: BACKEND_HOST
+                  value: {{ .Values.backend.route.host }}
+                - name: DJANGO_SUPERUSER_USERNAME
+                  valueFrom:
+                    secretKeyRef:
+                      key: username
+                      name: django-superuser
+                - name: DJANGO_SUPERUSER_PASSWORD
+                  valueFrom:
+                    secretKeyRef:
+                      key: password
+                      name: django-superuser
+              command:
+                - /bin/sh
+                - -ec
+                - "curl --fail-with-body https://$(BACKEND_HOST)/api/common/create-superuser"
+          restartPolicy: Never
+{{ end }}