import psycopg2 

# MERK: M? kj?res med Python 3

user = "" # Sett inn ditt UiO-brukernavn ("_priv" blir lagt til under)
pwd = "" # Sett inn passordet for _priv-brukeren du fikk i en mail

connection = \
    "dbname='" + user + "' " +  \
    "user='" + user + "_priv' " + \
    "port='5432' " +  \
    "host='dbpg-ifi-kurs03.uio.no' " + \
    "password='" + pwd + "'"

def main():
    conn = psycopg2.connect(connection) # Kobler til databasen
    
    valg = 0
    while (valg != 3):
        print("--[ Velkommen ]--")
        print("Vennligst velg et alternativ:\n 1. S?k etter en gruppel?rer\n \n 2. S?k etter en gruppe\n 3. Avslutt")
        valg = int(input("Valg: "))

        if (valg == 1):
            finn_gruppel?rer(conn)
        elif (valg == 2):
            finn_gruppe(conn)
    
def finn_gruppel?rer(conn):
    #Input
    id = input("Gi en id: ")

    #SQL Sp?rring:

    gruppel?rer = "SELECT ID, Fornavn, Etternavn, Email " + \
                "FROM Gruppel?rer AS g " + \
                "WHERE g.ID = %(id)s;" #1; DROP TABLE gruppel?rer CASCADE;--

    cur = conn.cursor() #Trengs for ? kunne kj?re sp?rringen

    sporring = gruppel?rer

    cur.execute(sporring, {'id' : id}) #Kj?rer sp?rringen
    conn.commit()

    try:
        rows = cur.fetchall()

        if (rows == []):
            print("No results.")
            return

        print("--[ GRUPPEL?RER ]--")
        print("Vi har ID: " + id)

        for row in rows:
            print("--Gruppel?rer--")
            print("ID: " + str(row[0]) + "\n" +  \
                "Navn: " + str(row[1]) + "\n" + \
                "Etternavn: " + str(row[2]) + "\n" + \
                "Mail: " + str(row[3]))

            print("\n")

    except Exception as e:
        print("Error occurred:", e)

def finn_gruppe(conn):
    #Input
    A = input("Gi en gruppel?rer: ")

    #SQL Sp?rring:

    gruppetime = "SELECT ID, Gruppel?rer, Emne, Antall " + \
            "FROM Gruppetime AS gt " + \
            "WHERE gt.ID = %(B)s;" #1; DROP TABLE gruppetime CASCADE;--
    
    cur = conn.cursor()

    sporring = gruppetime

    cur.execute(sporring, {'B' : A}) #Kj?rer sp?rringen
    conn.commit()

    try:
        rows = cur.fetchall()

        if (rows == []):
            print("No results.")
            return

        print("--[ GRUPPEL?RER ]--")
        print("Vi har ID: " + id)

        for row in rows:
            print("--Gruppel?rer--")
            print("ID: " + str(row[0]) + "\n" +  \
                "Navn: " + str(row[1]) + "\n" + \
                "Etternavn: " + str(row[2]) + "\n" + \
                "Mail: " + str(row[3]))

            print("\n")

    except Exception as e:
        print("Error occurred:", e)


if __name__ == "__main__":
    main()