Replace kirjasto.py

kirjasto.py

@@ -3,7 +3,6 @@ NIMI = 0
 KIRJAILIJA = 1
 VUOSI = 2
 SIVUJA = 3
-
 def lisaa_kirja(kirjat: list, nimi: str, kirjailija: str, vuosi: int, sivuja: int):
     """ Lisää annetun kirjan kirjalistaan """
     kirja = (nimi, kirjailija, vuosi, sivuja)
@@ -14,15 +13,15 @@ def etsi_kirja_nimen_perusteella(kirjat, hakusana: str) -> list:
     lista = []
     for kirja in kirjat:
         if hakusana in kirja[NIMI]:
+            lista.append(kirja)
     return lista
     
 
-
 def kirjailijan_kirjat(kirjat, kirjailija: str) -> list:
     """ Palauttaa kaikki annetun kirjailijan kirjoittamat kirjat """
     lista = []
     for kirja in kirjat:
-        if kirja[NIMI] == kirjailija:
+        if kirja[KIRJAILIJA] == kirjailija:
             lista.append(kirja)
 
     return lista
@@ -31,7 +30,7 @@ def vuoden_kirjat(kirjat, vuosi: int) -> list:
     """ Palauttaa kaikki kirjat, jotka on kirjoitettu annettuna vuonna """
     lista = []
     for kirja in kirjat:
-        if kirja[NIMI] == vuosi:
+        if kirja[VUOSI] == vuosi:
             lista.append(kirja)
 
     return lista
@@ -60,10 +59,9 @@ def tulosta_valikko():
 
 # PÄÄOHJELMA
 kirjat = []
-
-while True:
 tulosta_valikko()
-    valinta = input("Valinta: ")
+while True:
+    valinta = int(input("Valinta: "))
     if valinta == 0:
         break
 
@@ -80,11 +78,11 @@ while True:
     elif valinta == 3:
         kirjailija = input("Anna kirjailja: ")
         lista = kirjailijan_kirjat(kirjat, kirjailija)
-        
+        tulosta_kirjat(lista)
     elif valinta == 4:
         vuosi = int(input("Anna vuosi: "))
         lista = vuoden_kirjat(kirjat, vuosi)
-        tulosta_kirjat(vuosi)
+        tulosta_kirjat(lista)
     elif valinta == 5:
         tulosta_kirjat(kirjat)