#!/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