get_logo.py

import os
import pathlib

import requests

import conf

DOCS_OUT_FOLDER = pathlib.Path("docsrc") / "source" / "_static" / "images"
DOCS_OUT_PATH = DOCS_OUT_FOLDER / "logo.svg"


def download_logo(logo_url=conf.PACKAGE_LOGO_URL, out_path: str = str(DOCS_OUT_PATH)):
    if not logo_url:
        return

    print(f"Downloading logo from {logo_url}")
    response = requests.get(logo_url)
    if response.status_code != 200:
        raise NoLogoAtUrlException(logo_url)

    content = response.content.decode("utf8")
    with open(out_path, "w") as f:
        f.write(content)
    print(f"Logo outputted to {out_path}")


class NoLogoAtUrlException(Exception):
    pass


if __name__ == "__main__":
    if not os.path.exists(DOCS_OUT_FOLDER):
        os.makedirs(DOCS_OUT_FOLDER)

    download_logo()