diff --git a/src/classes/product_classes.py b/src/classes/product_classes.py index c937f5a..8a33754 100644 --- a/src/classes/product_classes.py +++ b/src/classes/product_classes.py @@ -1,5 +1,6 @@ -from pydantic import BaseModel +from pydantic import BaseModel, HttpUrl class Product(BaseModel): name: str - price: float \ No newline at end of file + price: float + image_url: HttpUrl \ No newline at end of file diff --git a/src/data_connection.py b/src/data_connection.py index ef75441..d9f65d8 100644 --- a/src/data_connection.py +++ b/src/data_connection.py @@ -1,8 +1,9 @@ from src.classes import product_classes -bubble_tea = product_classes.Product(name="Bubble Tea", price=5.0) +apple = product_classes.Product(name="Appel", price=1.0, image_url="https://i.etsystatic.com/16348658/r/il/380709/1340147432/il_570xN.1340147432_p3pc.jpg") +bubble_tea = product_classes.Product(name="Bubble Tea", price=5.0, image_url="https://uxwing.com/wp-content/themes/uxwing/download/food-and-drinks/bubble-tea-icon.png") -product_list = [bubble_tea] +product_list = [apple, bubble_tea] def get_all_products() -> list[product_classes.Product]: return product_list \ No newline at end of file