helper_sql.py
This commit is contained in:
parent
869d07ff13
commit
745d782ad3
66
src/helper_sql.py
Normal file
66
src/helper_sql.py
Normal file
|
@ -0,0 +1,66 @@
|
||||||
|
import threading
|
||||||
|
import Queue
|
||||||
|
|
||||||
|
sqlSubmitQueue = Queue.Queue() #SQLITE3 is so thread-unsafe that they won't even let you call it from different threads using your own locks. SQL objects can only be called from one thread.
|
||||||
|
sqlReturnQueue = Queue.Queue()
|
||||||
|
sqlLock = threading.Lock()
|
||||||
|
|
||||||
|
def sqlQuery(sqlStatement, *args):
|
||||||
|
sqlLock.acquire()
|
||||||
|
sqlSubmitQueue.put(sqlStatement)
|
||||||
|
|
||||||
|
if args == ():
|
||||||
|
sqlSubmitQueue.put('')
|
||||||
|
else:
|
||||||
|
sqlSubmitQueue.put(args)
|
||||||
|
|
||||||
|
queryreturn = sqlReturnQueue.get()
|
||||||
|
sqlLock.release()
|
||||||
|
|
||||||
|
return queryreturn
|
||||||
|
|
||||||
|
def sqlExecute(sqlStatement, *args):
|
||||||
|
sqlLock.acquire()
|
||||||
|
sqlSubmitQueue.put(sqlStatement)
|
||||||
|
|
||||||
|
if args == ():
|
||||||
|
sqlSubmitQueue.put('')
|
||||||
|
else:
|
||||||
|
sqlSubmitQueue.put(args)
|
||||||
|
|
||||||
|
sqlReturnQueue.get()
|
||||||
|
sqlSubmitQueue.put('commit')
|
||||||
|
sqlLock.release()
|
||||||
|
|
||||||
|
def sqlStoredProcedure(procName):
|
||||||
|
sqlLock.acquire()
|
||||||
|
sqlSubmitQueue.put(procName)
|
||||||
|
sqlLock.release()
|
||||||
|
|
||||||
|
class SqlBulkExecute:
|
||||||
|
def __enter__(self):
|
||||||
|
sqlLock.acquire()
|
||||||
|
return self
|
||||||
|
|
||||||
|
def __exit__(self, type, value, traceback):
|
||||||
|
sqlSubmitQueue.put('commit')
|
||||||
|
sqlLock.release()
|
||||||
|
|
||||||
|
def execute(self, sqlStatement, *args):
|
||||||
|
sqlSubmitQueue.put(sqlStatement)
|
||||||
|
|
||||||
|
if args == ():
|
||||||
|
sqlSubmitQueue.put('')
|
||||||
|
else:
|
||||||
|
sqlSubmitQueue.put(args)
|
||||||
|
sqlReturnQueue.get()
|
||||||
|
|
||||||
|
def query(self, sqlStatement, *args):
|
||||||
|
sqlSubmitQueue.put(sqlStatement)
|
||||||
|
|
||||||
|
if args == ():
|
||||||
|
sqlSubmitQueue.put('')
|
||||||
|
else:
|
||||||
|
sqlSubmitQueue.put(args)
|
||||||
|
return sqlReturnQueue.get()
|
||||||
|
|
Reference in New Issue
Block a user