voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add argument transaction_mode to begin.

Browse source
This commit is contained in:
voussoir 2022-11-12 12:38:50 -08:00
parent 953ebe4951
commit 355c02cf7e
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
voussoirkit/worms.py
View file

@ -99,7 +99,7 @@ class TransactionContextManager:
def __enter__(self): def __enter__(self):
log.loud('Entering transaction.') log.loud('Entering transaction.')
self.database.begin() self.database.begin(transaction_mode='IMMEDIATE')
return self return self
def __exit__(self, exc_type, exc_value, exc_traceback): def __exit__(self, exc_type, exc_value, exc_traceback):
@ -235,9 +235,12 @@ class Database(metaclass=abc.ABCMeta):
if table not in self.COLUMN_INDEX: if table not in self.COLUMN_INDEX:
raise BadTable(f'Table {table} does not exist.') raise BadTable(f'Table {table} does not exist.')
def begin(self): def begin(self, transaction_mode='DEFERRED'):
if transaction_mode not in {'DEFERRED', 'IMMEDIATE', 'EXCLUSIVE'}:
raise ValueError(transaction_mode)
self.acquire_transaction_lock() self.acquire_transaction_lock()
self.execute('BEGIN') self.execute(f'BEGIN {transaction_mode}')
def close(self): def close(self):
# Wrapped in hasattr because if the object fails __init__, Python will # Wrapped in hasattr because if the object fails __init__, Python will