init 2

mealie/services/image_services.py

@@ -0,0 +1,62 @@
+import shutil
+from pathlib import Path
+
+import requests
+from fastapi.responses import FileResponse
+
+CWD = Path(__file__).parent
+IMG_DIR = CWD.parent.joinpath("data", "img")
+
+
+def read_image(recipe_slug: str) -> FileResponse:
+    recipe_slug = recipe_slug.split(".")[0]
+    for file in IMG_DIR.glob(f"{recipe_slug}*"):
+        return file
+
+
+def write_image(recipe_slug: str, file_data: bytes, extension: str) -> Path.name:
+    pass
+    delete_image(recipe_slug)
+
+    image_path = Path(IMG_DIR.joinpath(f"{recipe_slug}.{extension}"))
+    with open(image_path, "ab") as f:
+        f.write(file_data)
+
+    return image_path
+
+
+def delete_image(recipe_slug: str) -> str:
+    recipe_slug = recipe_slug.split(".")[0]
+    for file in IMG_DIR.glob(f"{recipe_slug}*"):
+        return file.unlink()
+
+
+def scrape_image(image_url: str, slug: str) -> Path:
+    if isinstance(image_url, str):  # Handles String Types
+        image_url = image_url
+
+    if isinstance(image_url, list):  # Handles List Types
+        image_url = image_url[0]
+
+    if isinstance(image_url, dict):  # Handles Dictionary Types
+        for key in image_url:
+            if key == "url":
+                image_url = image_url.get("url")
+
+    filename = slug + "." + image_url.split(".")[-1]
+    filename = IMG_DIR.joinpath(filename)
+
+    try:
+        r = requests.get(image_url, stream=True)
+    except:
+        return None
+
+    if r.status_code == 200:
+        r.raw.decode_content = True
+
+        with open(filename, "wb") as f:
+            shutil.copyfileobj(r.raw, f)
+
+        return filename
+
+    return None