Closes #36: Execute all branch provisioning operations inside a Postg…

…reSQL transaction
netboxlabs · Aug 9, 2024 · f971fb0 · f971fb0
1 parent 3dac852
commit f971fb0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/netbox_branching/models/branches.py b/netbox_branching/models/branches.py
@@ -392,6 +392,9 @@ def provision(self, user):
             with connection.cursor() as cursor:
                 schema = self.schema_name
 
+                # Start a transaction
+                cursor.execute('BEGIN')
+
                 # Create the new schema
                 logger.debug(f'Creating schema {schema}')
                 try:
@@ -435,6 +438,9 @@ def provision(self, user):
                         f"ALTER TABLE {schema}.{table} ALTER COLUMN id SET DEFAULT nextval('public.{table}_id_seq')"
                     )
 
+                # Commit the transaction
+                cursor.execute('COMMIT')
+
         except Exception as e:
             logger.error(e)
             Branch.objects.filter(pk=self.pk).update(status=BranchStatusChoices.FAILED)