Skip to content

Commit 78ff572

Browse files
committed
create a helper function to handle the sadness level logic
1 parent 8366a7a commit 78ff572

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

implement-laptop-allocation/main.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,5 +26,16 @@ class Laptop:
2626
operating_system: OperatingSystem
2727

2828

29+
def sadness(person: Person, laptop: Laptop):
30+
try:
31+
return person.preferred_operating_system.index(laptop.operating_system)
32+
except ValueError:
33+
return 100
34+
35+
36+
if laptop.operating_system == person.preferred_operating_system:
37+
return index(Person.preferred_operating_system)
38+
else:
39+
return "Sadness level is 100"
2940

3041
def allocate_laptops(people: List[Person], laptops: List[Laptop]) -> Dict[Person, Laptop]:

0 commit comments

Comments
 (0)