from avto import avto

def menu():
    print "1. Dodaj avto"
    print "2. Izpisi avte"
    print "3. Izbrisi avto"
    print "4. Shrani avte"
    print "5. Preberi avte"
    print "6. Izhod"

    return int(raw_input())

# Od uporabnika pridobi podatke o avtu, ustvari nov objekt in ga vrni
def dodaj():
    from avto import avto

    model = raw_input("Model: ")
    znamka = raw_input("Znamka: ")
    prevozeni_km = int(raw_input("Prevozeni km: "))
    zadnji_servis = raw_input("Zadnji servis: ")

    avto = avto(model, znamka, prevozeni_km, zadnji_servis)

    return avto

# Sprehodi se čez celotno listov (ki jih dobiš v funkcijo kot paratemer) in jih izpiši
def izpis_avtov(avti_list):
    i = 0
    for avto in avti_list:
        print str(i) + "."
        avto.izpis()
        i += 1

# Shrani celotno listo avtomobilov v text (.txt) datoteko, podatke o avtu loči z ; različne avte pa zapiši v svojo vrstico
def shrani(avti_list):
    file = open("avti.txt", "w+")

    for avto in avti_list:
        file.write(avto.model + ";")
        file.write(avto.znamka + ";")
        file.write(str(avto.prevozeni_km) + ";")
        file.write(avto.zadnji_servis + ";")

# Preberi vse avte iz text datoteke, jih shrani v liso in vrni
def preberi_avte():
    file = open("avti.txt", "r+")
    temp_avti = []
    for line in file:
        a = line.split(";")
        av = avto(a[0], a[1], int(a[2]), a[3])
        temp_avti.append(av)
    return temp_avti

avti = []

while True:
    x = menu()

    if x == 1:
        avto = dodaj()
        avti.append(avto)
    if x == 2:
        izpis_avtov(avti)
    if x == 3:
        izpis_avtov(avti)
        st_avta = int(raw_input("Stevilka avta za izpis: "))
        avti.remove(avti[st_avta])
    if x == 4:
        shrani(avti)
    if x == 5:
        avti = preberi_avte()
    if x == 6:
        break