common/queue.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

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

import os
import sys
import pickle
from litequeue import SQLQueue

QUEUE_DB = os.environ.get('el_QUEUE_db', 'litequeue.db')

# Initialize queue
q = SQLQueue(QUEUE_DB, maxsize=None)

def dbi(sql,values,**kwargs):
    verbose = bool(kwargs['verbose']) if 'verbose' in kwargs else False

    q.put(pickle.dumps([sql, values]))
    if verbose is True:
        table = sql.split(' ')[2]
        num = 1 if isinstance(values, tuple) else len(values)
        print("Inserted", num, "item(s) into queue for", table)
    return True