blob: de498027c6cb8ef3fa68e844b1a8227094ba6b97 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
#!/usr/bin/env python3
""" common functions and stuff """
import logging
import os
import pickle
from litequeue import SQLQueue
QUEUE_DB = os.environ.get("el_QUEUE_db", "litequeue.db")
QUEUE_DIR = os.environ.get("el_QUEUE_dir", "queue")
QUEUE_DB = QUEUE_DIR + "/" + QUEUE_DB
log = logging.getLogger(__name__)
# Initialize queue
q = SQLQueue(QUEUE_DB, maxsize=None)
def dbi(sql, values, **kwargs):
"""insert into queue"""
log.debug("function dbi()")
verbose = bool(kwargs["verbose"]) if "verbose" in kwargs else False
log.debug("Inserting into sqlite-queue")
q.put(pickle.dumps([sql, values]))
if verbose is True:
log.debug("verbose = True")
table = sql.split(" ")[2]
num = 1 if isinstance(values, tuple) else len(values)
print("Inserted " + str(num) + " item(s) into queue for " + str(table))
return True
|
