path: root/scripts/common/queue.py

#!/usr/bin/env python3
''' common functions and stuff '''

import os
import pickle
import logging
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