Replace kirjasto.py

kirjasto.py

@@ -4,34 +4,36 @@ 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)
     kirjat.append(kirja)
 
-def etsi_kirja_nimen_perusteella(kirjat, hakusana: str) -> list:
+
+def etsi_kirja_nimen_perusteella(kirjat: list, hakusana: str) -> list:
     """ Etsii annetun hakusana perusteella kirjalistasta kaikki kirjat """
     lista = []
     for kirja in kirjat:
         if hakusana in kirja[NIMI]:
+            lista.append(kirja)
     return lista
     
-    
-
-def kirjailijan_kirjat(kirjat, kirjailija: str) -> list:
+def kirjailijan_kirjat(kirjat: list, 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
 
-def vuoden_kirjat(kirjat, vuosi: int) -> list:
+
+def vuoden_kirjat(kirjat: list, 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
@@ -59,11 +61,13 @@ def tulosta_valikko():
     print("0. Poistu")
 
 # PÄÄOHJELMA
+    
+
 kirjat = []
 
 while True:
     tulosta_valikko()
-    valinta = input("Valinta: ")
+    valinta = int(input("Valinta: "))
     if valinta == 0:
         break
 
@@ -80,11 +84,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)