Add helpers for retrieving the images from picnic and add a test for this

Author: daviddemeij

integration_tests/test_helper.py

@@ -0,0 +1,33 @@
+from python_picnic_api.helper import get_image, get_recipe_image
+import requests
+
+def test_get_image():
+    id = "8560e1f1c2d2811dfefbbb2342ef0d95250533f2131416aca459bde35d73e901"
+    size = "tile-medium"
+    suffix = "webp"
+    url = get_image(id, size=size, suffix=suffix)
+
+    response = requests.get(url)
+
+    # Check if the response status code indicates success
+    assert response.status_code == 200, "Failed to fetch URL"
+
+    # Check if the response content type is an image format
+    content_type = response.headers.get("content-type")
+    assert content_type.startswith("image/"), "URL does not return an image"
+
+
+def test_get_recipe_image():
+    id = "5c4cc7cb7a0429695da708394eb0cae1bd9b92935ac76c8fda63bbc57ad5b826"
+    size = "medium"
+    url = get_recipe_image(id, size=size)
+    print(url)
+
+    response = requests.get(url)
+
+    # Check if the response status code indicates success
+    assert response.status_code == 200, "Failed to fetch URL"
+
+    # Check if the response content type is an image format
+    content_type = response.headers.get("content-type")
+    assert content_type.startswith("image/"), "URL does not return an image"

python_picnic_api/helper.py

@@ -5,6 +5,8 @@
 tee = "├── "
 last = "└── "
 
+IMAGE_SIZES = ["small", "medium", "regualr", "large", "extra-large"]
+IMAGE_BASE_URL = "https://storefront-prod.nl.picnicinternational.com/static/images"
 
 def _tree_generator(response: list, prefix: str = ""):
     """A recursive tree generator,
@@ -23,3 +25,19 @@ def _tree_generator(response: list, prefix: str = ""):
 
 def _url_generator(url: str, country_code: str, api_version: str):
     return url.format(country_code.lower(), api_version)
+
+
+def get_recipe_image(id: str, size="regular"):
+    sizes = IMAGE_SIZES + ["1250x1250"]
+    assert size in sizes, "size must be one of: " + ", ".join(sizes)
+    return f"{IMAGE_BASE_URL}/recipes/{id}/{size}.png"
+
+
+def get_image(id: str, size="regular", suffix="webp"):
+    assert suffix in ["webp", "png"], "suffix must be webp or png"
+    sizes = IMAGE_SIZES + [f"tile-{size}" for size in IMAGE_SIZES]
+
+    assert size in sizes, (
+        "size must be one of: " + ", ".join(sizes)
+    )
+    return f"{IMAGE_BASE_URL}/{id}/{size}.{suffix}"