updated with list comprehension and method get()

mate-academy · Dec 4, 2024 · 60a0344 · 60a0344
1 parent 1dbd313
commit 60a0344
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -15,9 +15,9 @@ def create_person_list(people: list) -> list:
         name = person["name"]
         instance = Person.people[name]
 
-        if "wife" in person and person["wife"]:
+        if person.get("wife"):
             instance.wife = instance.people[person["wife"]]
-        elif "husband" in person and person["husband"]:
+        elif person.get("husband"):
             instance.husband = instance.people[person["husband"]]
 
     return person_list