blob: f67acbb8b35f3562060432658a49635a1a6a62d9 (
plain) (
tree)
|
|
#!/usr/bin/env python3
""" common functions and stuff """
import logging
import pickle
from litequeue import SQLQueue
from . import env
QUEUE_DB = env("el_QUEUE_db", "litequeue.db")
QUEUE_DIR = env("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
|
