postgres_testing.py

from time import sleep

import psycopg2

CONN_INFO = {
    'host': '',
    'port': 5432,
    'dbname': '',
    'user': '',
    'password': '',
}


def log(*args):
    print('\x1b[1K\r', end='')
    print(*args, end='', flush=True)

def query():
    conn = psycopg2.connect(**CONN_INFO)
    cursor = conn.cursor()
    cursor.execute('SELECT * FROM people;')
    cursor.fetchall()
    log('Queries operating normally')

def test():
    try:
        while True:
            try:
                query()
            except psycopg2.errors.Error as e:
                log('Error running query')
            sleep(1)
    except KeyboardInterrupt:
        print('\nexiting test')