import psycopg2


def connect_starrs(db, user, host, password):
    try:
        starrs = psycopg2.connect("dbname='{}' user='{}' host='{}' password='{}'".format(db, user, host, password))
    except:
        print("Unable to connect to STARRS database.")
        raise
    return starrs


def get_next_ip(starrs, range_name):
    c = starrs.cursor()
    try:
        c.execute("BEGIN")
        c.callproc("api.initialize", ('root',))
        c.callproc("api.get_address_from_range", (range_name,))
        results = c.fetchall()
        c.execute("COMMIT")
    finally:
        c.close()
    return results


def register_starrs(starrs, name, owner, mac, addr):
    c = starrs.cursor()
    try:
        c.execute("BEGIN")
        c.callproc("api.initialize", ('root',))
        c.callproc("api.create_system_quick", (name, owner, 'members', mac, addr, 'csh.rit.edu', 'dhcp', True))
        results = c.fetchall()
        c.execute("COMMIT")
    finally:
        c.close()
    return results


def delete_starrs(starrs, name):
    c = starrs.cursor()
    try:
        c.execute("BEGIN")
        c.callproc("api.initialize", ('root',))
        c.callproc("api.remove_system", (name,))
        results = c.fetchall()
        c.execute("COMMIT")
    finally:
        c.close()
    return results