add rollback method in migration

gip-inclusion · Nov 21, 2024 · caf89b4 · caf89b4
1 parent e5daefd
commit caf89b4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...he/tenders/migrations/0094_add_templatetransactional_tender_author_commercial_partners.py b/...he/tenders/migrations/0094_add_templatetransactional_tender_author_commercial_partners.py
@@ -10,11 +10,16 @@ def create_template(apps, schema_editor):
     )
 
 
+def delete_template(apps, schema_editor):
+    TemplateTransactional = apps.get_model("conversations", "TemplateTransactional")
+    TemplateTransactional.objects.get(code="TENDERS_AUTHOR_CONFIRMATION_VALIDATED_COMMERCIAL_PARTNERS").delete()
+
+
 class Migration(migrations.Migration):
     dependencies = [
         ("tenders", "0093_alter_tender_send_to_commercial_partners_only"),
     ]
 
     operations = [
-        migrations.RunPython(create_template),
+        migrations.RunPython(create_template, reverse_code=delete_template),
     ]