Du kan skriva en kontexthanterare och använda med-satsen. Se till exempel detta blogginlägg:
http://jessenoller. com/2009/02/03/get-with-the-program-as-contextmanager-completely-different/
Python-dokumentationen har också ett exempel som i stort sett matchar dina behov. Se avsnitt 8.1 på den här sidan, särskilt utdraget som börjar:
db_connection = DatabaseConnection()
with db_connection as cursor:
cursor.execute('insert into ...')
cursor.execute('delete from ...')
# ... more operations ...