import psycopg2 

# MERK: M? kj?res med Python 3

user = "osjohans" # Sett inn ditt UiO-brukernavn ("_priv" blir lagt til under)
pwd = "Iizah3zein" # 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, firstname, lastname, mail " + \
                            "FROM teacherr as t" + \
                            "WHERE t.id = " + id
    
    cur = conn.cursor() #Trengs for ? kunne kj?re sp?rringen

    sporring = gruppel?rer
    cur.execute(sporring) #Kj?rer sp?rringen

    rows = cur.fetchall() #Henter resultatet 

    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")

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

    #SQL Sp?rring:

    gruppe = "SELECT id, teacher, subject, count " + \
                            "FROM teachinggroup as tg" + \
                            "WHERE tg.teacher = " + id
    
    cur = conn.cursor()

    sporring = gruppe
    cur.execute(sporring)

    rows = cur.fetchall()

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

    print("--[ Gruppe ]--")
    print("Vi har ID: " + id)

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

        print("\n")


if __name__ == "__main__":
    main()