import random from bestilling import Bestilling class BookingSystem: def __init__(self, filnavn): self._bookinger = {} self.lesFraFil(filnavn) def lesFraFil(self, filnavn): fil = open(filnavn) for linje in fil: data = linje.strip().split(";") ref = self.lagBookingref(data[0]) nyBestilling = Bestilling(data[0], data[1], data[2], ref) self._bookinger[ref] = nyBestilling fil.close() def lagBookingref(self, navn): tilfeldigTall = random.randint(1, 99) ref = "ref" + navn[0:3] + str(tilfeldigTall) # hva om den allerede finnes? while ref in self._bookinger: tilfeldigTall += 1 ref = "ref" + navn[0:3] + str(tilfeldigTall) return ref def skrivUt(self): for bookingref in self._bookinger: print(self._bookinger[bookingref]) def skrivBookingRef(self, bookingref): print(self._bookinger[bookingref]) def skrivBookingNavn(self, navn): for bookingref in self._bookinger: booking = self._bookinger[bookingref] if booking.hentNavn() == navn: print(booking) def meny(self): menystr= """Meny 1 - se booking på bookingref 2 - se booking på navn 3 - legg inn ny booking (kommer) 0 - avslutt""" print(menystr) brukervalg = input("Ditt valg:\n> ") while brukervalg != "0": if brukervalg == "1": ref = input("Oppgi booking referanse:\n> ") self.skrivBookingRef(ref) elif brukervalg == "2": navn = input("Oppgi navn:\n> ") self.skrivBookingNavn(navn) elif brukervalg == "3": print("Kommer snart...") elif brukervalg == "0": print("Avslutter") print(menystr) brukervalg = input("Ditt valg:\n> ")