import Karte
import Oprema
import Radnici
from Karte import godisnjaKarta
from Oprema import oMain
from Karte import mesecnaKarta
from Karte import dnevnaKarta
from Karte import studentskaGodisnja
from Karte import suma


def main():
    print()
    print("BAZEN")
    print("_____________________________")
    print()
    if not login():
        print("Niste se uneli dobar username ili password")
        return main()
    else:
        print("Dobrodosli")

    komanda = menu()
    if komanda != 'X':
        if komanda == '1':
            godisnjaKarta()
        elif komanda == '2':
            mesecnaKarta()
        elif komanda == '3':
            dnevnaKarta()
        elif komanda == '4':
            studentskaGodisnja()
        elif komanda == '5':
            oMain()
        elif komanda == '6':
            suma()
        return main()

    print("KRAJ")
def menu():
    printMenu()
    command = input(">>")
    while command.upper() not in ('1', '2', '3', '4', '5', '6', 'X'):
        print("\nUneli sta pogresnu komandu. \n")
        printMenu()
        command = input(">>")
    return command.upper()

def printMenu():
    print("Izaberite opciju.")
    print("1.) Izdavanje godisnje karte.")
    print("2.) Izdavanje mesecne karte.")
    print("3.) Izdavanje dnevne karte.")
    print("4.) Studentski popust na godisnju kartu.")
    print("5.) Pomocna oprema.")
    print("6.) Zbir zarade od karata.")
    print("X ) Izlaz")


def login():
    username = input("Korisnicko ime: ")
    password = input("Lozinka: ")
    return Radnici.login(username, password)
if __name__ == '__main__':
    main()