add change location

mate-academy · Jun 1, 2024 · 359fb4e · 359fb4e
1 parent 1991b23
commit 359fb4e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/app/main.py b/app/main.py
@@ -43,9 +43,12 @@ def shop_trip() -> None:
         if customer.money >= min_price:
             print(f"{customer.name} rides to "
                   f"{cheapest_shop.name}\n")
+            home_location = customer.location
+            customer.location = cheapest_shop.location
             cheapest_shop.print_receipt(customer)
             print(f"{customer.name} rides home\n{customer.name} now has "
                   f"{round(customer.money - min_price, 2)} dollars\n")
+            customer.location = home_location
         else:
             print(f"{customer.name} doesn't have enough money "
                   f"to make a purchase in any shop")