aboutsummaryrefslogtreecommitdiffstats
path: root/scripts/common/queue.py
blob: b4fa3690dd848743910da957a7f33ed363adc0b4 (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
#!/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')

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