tests/delayedquery.py
author Radek Brich <brich.radek@ifortuna.cz>
Tue, 06 May 2014 18:34:38 +0200
changeset 99 245646538743
parent 78 64c62ac8f65d
child 104 d8ff52a0390f
permissions -rwxr-xr-x
Update runquery tool: Add --one-query-per-line parameter.

#!/usr/bin/env python3

import logging
import time

from tests.config import Config
from pgtoolkit import pgmanager
from pgtoolkit.delayedquery import DelayedQuery


if __name__ == '__main__':
    cfg = Config('tests.conf')

    pgm = pgmanager.get_instance()
    pgm.create_conn(isolation_level = 'autocommit', **cfg)

    handler = logging.StreamHandler()
    logger = logging.getLogger('pgmanager')
    logger.addHandler(handler)
    logger.setLevel(logging.DEBUG)

    dq = DelayedQuery(pgm)

    print('add 1')
    dq.add(10, 'INSERT INTO test (name) VALUES (%s)', ['spam1'], queryid=1)
    dq.add(10, 'INSERT INTO test (name) VALUES (%s)', ['spam1'], queryid=1)
    print('after 1')

    time.sleep(5)

    print('add 2')
    dq.add(10, 'INSERT INTO test (name) VALUES (%s)', ['spam2'])

    time.sleep(5)
    print('after 10')

    time.sleep(10)