aboutsummaryrefslogtreecommitdiffstats
path: root/scripts/common/queue.py
diff options
context:
space:
mode:
authorDennis Eriksen <d@ennis.no>2023-02-01 20:32:11 +0100
committerDennis Eriksen <d@ennis.no>2023-02-01 20:32:11 +0100
commit8d186d39483beff64a1c11f80c6ca5e56dd7bbc5 (patch)
tree2c5a64ace4bd8eabd4d65014c5313bd7edd76191 /scripts/common/queue.py
parentrun queue in batches (diff)
downloadenergyscripts-8d186d39483beff64a1c11f80c6ca5e56dd7bbc5.tar.gz
moving and renaming/breaking everything
Diffstat (limited to 'scripts/common/queue.py')
-rw-r--r--scripts/common/queue.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/scripts/common/queue.py b/scripts/common/queue.py
new file mode 100644
index 0000000..e567fb4
--- /dev/null
+++ b/scripts/common/queue.py
@@ -0,0 +1,22 @@
+#!/usr/bin/env python3
+''' common functions and stuff '''
+
+import os
+import sys
+import pickle
+from litequeue import SQLQueue
+
+QUEUE_DB = os.environ.get('el_QUEUE_db', 'litequeue.db')
+
+# Initialize queue
+q = SQLQueue(QUEUE_DB, maxsize=None)
+
+def dbi(sql,values,**kwargs):
+ verbose = bool(kwargs['verbose']) if 'verbose' in kwargs else False
+
+ q.put(pickle.dumps([sql, values]))
+ if verbose is True:
+ table = sql.split(' ')[2]
+ num = 1 if isinstance(values, tuple) else len(values)
+ print("Inserted", num, "item(s) into queue for", table)
+ return True