Develop

README.md

@@ -131,16 +131,20 @@ You design application architecture by yourself, but there are some rules:
   <summary>
     Hint: modules structure example if you have problems with architecture.
   </summary>
-  py-shop-trip/<br>
-  |-- car.py<br>
-  |-- customer.py<br>
-  |-- main.py<br>
-  └-- shop.py<br>
+
+```
+└── py-shop-trip
+  └── app
+    ├── car.py
+    ├── customer.py
+    ├── main.py
+    └── shop.py
+```
 </details>
 
 
 Distance between customer and shop is a distance between their locations in km. 
 Round printed value to two decimal places.
 
 
-### Note: Check your code using this [checklist](checklist.md) before pushing your solution.
+### Note: Check your code using this [checklist](checklist.md) before pushing your solution.

app/car.py

@@ -0,0 +1,16 @@
+class Car:
+    def __init__(
+            self,
+            brand: str,
+            fuel_consumption: float,
+    ) -> None:
+        self.brand = brand
+        self.fuel_consumption = fuel_consumption
+
+    def calculate_trip_cost(
+            self,
+            distance: int,
+            fuel_price: float
+    ) -> float:
+        cost_trip = (distance * (self.fuel_consumption / 100)) * fuel_price
+        return cost_trip

app/customer.py

@@ -0,0 +1,37 @@
+from typing import Dict, List
+from app.car import Car
+from app.shop import Shop
+
+
+class Customer:
+    def __init__(
+            self,
+            name: str,
+            product_cart: List,
+            location: List,
+            car: Car,
+            money: int,
+    ) -> None:
+        self.name = name
+        self.product_cart = product_cart
+        self.location = location
+        self.car = car
+        self.money = money
+        self.total_cost = None
+
+    def calculate_trip_cost(self, shop: Dict, fuel_price: float) -> float:
+        distance_to_shop = self.calculate_distance(shop.location)
+        fuel_cost = self.car.calculate_trip_cost(distance_to_shop, fuel_price)
+        shopping_cost = shop.shopping_cost(self.product_cart)
+        total_cost = fuel_cost + shopping_cost + fuel_cost
+        return round(total_cost, 2)
+
+    def calculate_distance(self, shop_location: List) -> float:
+        x1, y1 = self.location
+        x2, y2 = shop_location
+        distance = ((x2 - x1) ** 2 + (y2 - y1) ** 2) ** 0.5
+        return distance
+
+    def calculate_money(self, shop: Shop) -> bool:
+        shopping_cost = shop.shopping_cost(self.product_cart)
+        return self.money >= shopping_cost

app/main.py

@@ -1,3 +1,67 @@
-def shop_trip():
-    # write your code here
-    pass
+import json
+import datetime
+from app.customer import Customer
+from app.car import Car
+from app.shop import Shop
+
+
+def shop_trip() -> str:
+    with open("app/config.json", "r") as config_file:
+        config_data = json.load(config_file)
+
+    fuel_price = config_data["FUEL_PRICE"]
+    customers_data = config_data["customers"]
+    shops_data = config_data["shops"]
+
+    customers = []
+    shops = []
+
+    for customer_data in customers_data:
+        car_data = customer_data["car"]
+        car = Car(car_data["brand"], car_data["fuel_consumption"])
+
+        customer = Customer(
+            customer_data["name"],
+            customer_data["product_cart"],
+            customer_data["location"],
+            car,
+            customer_data["money"],
+        )
+        customers.append(customer)
+
+    for shop_data in shops_data:
+        shop = Shop(
+            shop_data["name"],
+            shop_data["location"],
+            shop_data["products"],
+            fuel_price
+        )
+        shops.append(shop)
+
+    for customer in customers:
+        print(f"{customer.name} has {customer.money} dollars")
+        cheapest_shop = None
+        cheapest_cost = float("inf")
+
+        for shop in shops:
+            trip_cost = customer.calculate_trip_cost(shop, fuel_price)
+            print(f"{customer.name}'s trip to the {shop.name}"
+                  f" costs {trip_cost}")
+            if customer.calculate_money(shop):
+                if trip_cost < cheapest_cost:
+                    cheapest_cost = trip_cost
+                    cheapest_shop = shop
+
+        if cheapest_shop is not None:
+            print(f"{customer.name} rides to "
+                  f"{cheapest_shop.name}\n")
+            current_time = datetime.datetime.\
+                now().strftime("%d/%m/%Y %H:%M:%S")
+            receipt = cheapest_shop.generate_receipt(
+                customer, current_time, fuel_price)
+            print(receipt)
+            last_money = customer.money - cheapest_cost
+            print(f"{customer.name} now has {last_money} dollars\n")
+        else:
+            print(f"{customer.name} doesn't have enough money "
+                  f"to make a purchase in any shop")

app/shop.py

@@ -0,0 +1,44 @@
+from typing import Dict, List
+
+
+class Shop:
+    def __init__(
+            self,
+            name: str,
+            location: Dict,
+            product_cart: List,
+            fuel_price: float
+    ) -> None:
+        self.name = name
+        self.location = location
+        self.product_cart = product_cart
+        self.fuel_price = fuel_price
+
+    def shopping_cost(self, product_cart: List) -> float:
+        total_cost = sum(self.product_cart[product] * product_cart[product]
+                         for product in product_cart.keys())
+        return round(total_cost, 2)
+
+    def generate_receipt(
+            self,
+            customer: str,
+            current_time: str,
+            fuel_price: float
+    ) -> str:
+        receipt = f"Date: {current_time}\nThanks, {customer.name}," \
+                  f" for your purchase!\nYou have bought:"
+        total_cost = 0
+        for product, quantity in customer.product_cart.items():
+            if product in self.product_cart:
+                product_cost = self.product_cart[product]
+                product_total_cost = float(product_cost * quantity)
+                if product_total_cost.is_integer():
+                    product_total_cost = int(product_total_cost)
+                receipt += (
+                    f"\n{quantity} {product}s for "
+                    f"{product_total_cost} dollars"
+                )
+                total_cost += product_total_cost
+        receipt += f"\nTotal cost is {total_cost} dollars\nSee " \
+                   f"you again!\n\n{customer.name} rides home"
+        return receipt