lifecycle: migrate: ensure template schema exists before migrating (c…

…herry-pick #8952) (#9022) lifecycle: migrate: ensure template schema exists before migrating (#8952) Co-authored-by: Marc 'risson' Schmitt <marc.schmitt@risson.space>
goauthentik · Mar 25, 2024 · 49d83f1 · 49d83f1
1 parent 5f0af81
commit 49d83f1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lifecycle/migrate.py b/lifecycle/migrate.py
@@ -64,6 +64,7 @@ def release_lock(cursor: Cursor):
     """Release database lock"""
     if not LOCKED:
         return
+    LOGGER.info("releasing database lock")
     cursor.execute("SELECT pg_advisory_unlock(%s)", (ADV_LOCK_UID,))
 
 

diff --git a/lifecycle/system_migrations/template_schema.py b/lifecycle/system_migrations/template_schema.py
@@ -0,0 +1,12 @@
+from lifecycle.migrate import BaseMigration
+
+
+class Migration(BaseMigration):
+    def needs_migration(self) -> bool:
+        self.cur.execute(
+            "SELECT schema_name FROM information_schema.schemata WHERE schema_name = 'template';"
+        )
+        return not bool(self.cur.rowcount)
+
+    def run(self):
+        self.cur.execute("CREATE SCHEMA IF NOT EXISTS template; COMMIT;")