Enable foreign keys when opening the database #11

Activity

1. kotoko commented on Dec 18, 2021

   

   kotoko

   on Dec 18, 2021

   Last edited by kotoko

   More actions

   If someone else is interested I came up with following workaround. First create function decorator:

   from functools import wraps
   def pragmaForeignKeys(func):
   	'''Enable support for foreign keys'''
   
   	@wraps(func)
   	def wrapper(*args, **kwargs):
   		db = args[0]
   		db.execute('PRAGMA foreign_keys = ON')
   		return func(*args, **kwargs)
   
   	return wrapper

   Then instead of this:

   def doSomethingWithSQL(db, param1, param2, param3):
   	db.execute('PRAGMA foreign_keys = ON')
   	sql = 'SELECT ?, ?, ? FROM my_table'
   	db.execute(sql, (param1, param2, param3))
   	# ...

   you can write this:

   @pragmaForeignKeys
   def doSomethingWithSQL(db, param1, param2, param3):
   	sql = 'SELECT ?, ?, ? FROM my_table'
   	db.execute(sql, (param1, param2, param3))
   	# ...

   Each function which is decorated with @pragmaForeignKeys will run PRAGMA foreign_keys = ON first. Only requirement is that connection to sqlite (db) must be first parameter in doSomethingWithSQL.