import os
import psycopg2

if os.name =='nt':
    root = "//WDsentinel/share/CorpCDOs"
elif os.name == 'posix':
    root = '/home/share/CorpCDOs'

conn = psycopg2.connect(database="ET",
                        user="et_user",
                        password="Serenitas1",
                        host="debian")

def query_db(conn, sqlstr, **kwargs):

    params = kwargs.get('params', None)
    one = kwargs.get('one', True)
    try:
        with conn.cursor() as c:
            if params:
                c.execute(sqlstr, params)
            else:
                c.execute(sqlstr)
            if one:
                return c.fetchone()
            else:
                return c.fetchall()
    except psycopg2.Error:
        c.close()
        conn.rollback()
        return None