add reverse function to permissiontype initial data migration

hackforla · Aug 12, 2024 · d1d7e65 · d1d7e65
1 parent dcabb0a
commit d1d7e65
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/app/data/migrations/0004_permissiontype_seed.py b/app/data/migrations/0004_permissiontype_seed.py
@@ -3,7 +3,7 @@
 from core.models import PermissionType
 
 
-def run(__code__, __reverse_code__):
+def insert_data(__code__, __reverse_code__):
     items = [
         (1, "adminGlobal", "Granted to People Depo Admins. Can CRUD anything."),
         (
@@ -46,8 +46,11 @@ def run(__code__, __reverse_code__):
         PermissionType.objects.create(uuid=uuid, name=name, description=description)
 
 
+def clear_table(__code__, __reverse_code__):
+    PermissionType.objects.all().delete()
+
+
 class Migration(migrations.Migration):
-    initial = True
     dependencies = [("data", "0003_sdg_seed")]
 
-    operations = [migrations.RunPython(run, migrations.RunPython.noop)]
+    operations = [migrations.RunPython(insert_data, clear_table)]