Update transaction.py

revitron · Oct 15, 2020 · 63c2090 · 63c2090
1 parent ed0c001
commit 63c2090
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/revitron/transaction.py b/revitron/transaction.py
@@ -1,5 +1,5 @@
 """
-The ``transaction`` submodule contains a wrapper class to simplify the usage of transactions:: 
+The ``transaction`` submodule contains a wrapper class to simplify the usage of transactions and subtransactions:: 
 
 	t = revitron.Transaction() 
 	... 
@@ -16,10 +16,15 @@ class Transaction:
 
 	def __init__(self):
 		"""
-		Inits a new transaction.
+		Inits a new transaction. 
+
+		In case there is already an open transaction, a subtransaction will be initialized instead.
 		"""  
 		import revitron
-		self.transaction = revitron.DB.Transaction(revitron.DOC, script.get_button().get_title())
+		if revitron.DOC.IsModifiable:
+			self.transaction = revitron.DB.SubTransaction(revitron.DOC)
+		else:
+			self.transaction = revitron.DB.Transaction(revitron.DOC, script.get_button().get_title())
 		self.transaction.Start()