From 4b5932383644ef4bb1575b1daf3486882f8d6499 Mon Sep 17 00:00:00 2001 From: Vasyl Date: Sun, 19 Nov 2023 13:30:29 +0200 Subject: [PATCH] used list comprehension and updated the loop --- app/main.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/app/main.py b/app/main.py index e1e6ad0e2..b7bc1f470 100644 --- a/app/main.py +++ b/app/main.py @@ -11,15 +11,15 @@ def __str__(self) -> str: def create_person_list(people: list) -> list: - person_list = [] + person_list = [ + Person(name=person["name"], age=person["age"]) + for person in people + ] for index, person in enumerate(people): - person_list.append(Person(name=person["name"], age=person["age"])) - - for index, person in enumerate(people): - if "wife" in person and person["wife"] is not None: + if person.get("wife"): person_list[index].wife = Person.people[person["wife"]] - elif "husband" in person and person["husband"] is not None: + elif person.get("husband"): person_list[index].husband = Person.people[person["husband"]] return person_list