#!/usr/bin/python

import psycopg2

def getconnection(dbname, user, password, host, port):
    try:
        conn = psycopg2.connect("dbname="+dbname+" user="+user+" password="+password+" host="+host+" port="+port)
        print('CONNECTION SUCCESSFUL')
        return conn
    except:
        print('CONNECTION FAILED')

def insert(conn, sql):
    cur = conn.cursor()
    try:
        cur.execute(sql)
        conn.commit()
        print('INSERT SUCCESSFUL')
    except:
        print('INSERT FAILED')

def delete(conn,sql):
    cur = conn.cursor()
    try:
        cur.execute(sql)
        conn.commit()
        print('DELETE SUCCESSFUL')
    except:
        print('DELETE FAILED')

def update(conn,sql):
    cur = conn.cursor()
    try:
        cur.execute(sql)
        conn.commit()
        print('UPDATE SUCCESSFUL')
    except:
        print('UPDATE FAILED')

def select(conn, sql):
    cur = conn.cursor()
    try:
        cur.execute(sql)
        print('SELECT SUCCESSFUL')
        return cur
    except:
        print('SELECT FAILED')

def get(cur,limit=500):
    try:
        results=cur.fetchmany(limit)
        print('GET SUCCESSFUL')
        yield results
    except:
        print('GET FAILED')

def close(conn):
    try:
        conn.close()
        print('CLOSE SUCCESSFUL')
    except:
        print('CLOSE FAILED')