copy_build_into_docs.py

import importlib.metadata
import os
import shutil
import subprocess as sp
from pathlib import Path


# https://stackoverflow.com/a/1094933
def sizeof_fmt(num, suffix="B"):
    """
    Return a human formatted file size with the given suffix.
    """
    for unit in ("", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi"):
        if abs(num) < 1024.0:
            return f"{num:3.1f}{unit}{suffix}"
        num /= 1024.0
    return f"{num:.1f}Yi{suffix}"


current_build = sp.check_output(["git", "rev-parse", "--short", "HEAD"], text=True)
print(f"Current build: {current_build}")

docs = Path(__file__).parent / "docs"

(docs / ".commit.md").write_text(current_build, encoding="utf-8")

latest_build_folder = docs / "latest-build"
latest_build_folder.mkdir(exist_ok=True)
print(f"Latest build folder: {latest_build_folder}")

latest_build_text = """\
File | Size
---- | ----
"""

version = importlib.metadata.version("songs-dl")

for file in (Path(__file__).parent / "dist").iterdir():
    print(f"Copying {file.name} to latest-build")
    file = Path(shutil.copy(file, latest_build_folder))
    if f"-{version}" in file.name:
        file = file.rename(file.parent / file.name.replace(f"-{version}", ""))
    fname = file.name
    size_formatted = sizeof_fmt(os.path.getsize(file))
    latest_build_text += f"[{fname}](latest-build/{fname}) | {size_formatted}\n"

print("Writing .latest-build.md")
(docs / ".latest-build.md").write_text(latest_build_text, encoding="utf-8")