fix(migrations): remove usage of models

PnX-SI · Nov 20, 2024 · e672283 · e672283
1 parent 35e508c
commit e672283
Showing 1 changed file with 33 additions and 28 deletions.
diff --git a/backend/geonature/migrations/versions/95acee9f0452_add_comment_notification.py b/backend/geonature/migrations/versions/95acee9f0452_add_comment_notification.py
@@ -9,12 +9,6 @@
 import sqlalchemy as sa
 from alembic import op
 
-from geonature.core.notifications.models import (
-    NotificationCategory,
-    NotificationRule,
-    NotificationTemplate,
-)
-
 # revision identifiers, used by Alembic.
 revision = "95acee9f0452"
 down_revision = "e2a94808cf76"
@@ -36,34 +30,45 @@
 
 
 def upgrade():
-    bind = op.get_bind()
-    session = sa.orm.Session(bind=bind)
+    conn = op.get_bind()
+    metadata = sa.MetaData(bind=conn)
 
     # Add category
-    category = NotificationCategory(
-        code=CATEGORY_CODE,
-        label="Nouveau commentaire sur une observation",
-        description=(
-            "Se déclenche lorsqu'un nouveau commentaire est ajouté à une de vos observations, ou une observation que vous avez commenté"
-        ),
+    notification_category = sa.Table(
+        "bib_notifications_categories", metadata, schema="gn_notifications", autoload_with=conn
+    )
+    op.execute(
+        sa.insert(notification_category).values(
+            {
+                "code": CATEGORY_CODE,
+                "label": "Nouveau commentaire sur une observation",
+                "description": "Se déclenche lorsqu'un nouveau commentaire est ajouté à une de vos observations, ou une observation que vous avez commenté",
+            }
+        )
     )
 
-    session.add(category)
-
-    for method, content in (("EMAIL", EMAIL_CONTENT), ("DB", DB_CONTENT)):
-        template = NotificationTemplate(category=category, code_method=method, content=content)
-        session.add(template)
-
-    session.commit()
+    notification_template = sa.Table(
+        "bib_notifications_templates", metadata, schema="gn_notifications", autoload_with=conn
+    )
+    op.execute(
+        sa.insert(notification_template).values(
+            [
+                {"code_category": CATEGORY_CODE, "code_method": "EMAIL", "content": EMAIL_CONTENT},
+                {"code_category": CATEGORY_CODE, "code_method": "DB", "content": DB_CONTENT},
+            ]
+        )
+    )
 
+    notification_rule = sa.Table(
+        "bib_notifications_rules", metadata, schema="gn_notifications", autoload_with=conn
+    )
     op.execute(
-        f"""
-        INSERT INTO 
-            gn_notifications.t_notifications_rules (code_category, code_method)
-        VALUES
-            ('{CATEGORY_CODE}', 'DB'),
-            ('{CATEGORY_CODE}', 'EMAIL')
-        """
+        sa.insert(notification_rule).values(
+            [
+                {"code_category": CATEGORY_CODE, "code_method": "EMAIL"},
+                {"code_category": CATEGORY_CODE, "code_method": "DB"},
+            ]
+        )
     )