mealie/mealie/services/image_services.py

import shutil
from pathlib import Path

import requests

CWD = Path(__file__).parent
IMG_DIR = CWD.parent.joinpath("data", "img")


def read_image(recipe_slug: str) -> Path:
    if IMG_DIR.joinpath(recipe_slug).is_file():
        return IMG_DIR.joinpath(recipe_slug)
    else:
        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:
    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