# Inspired by https://github.com/BradyAJohnston/MolecularNodes/blob/main/build.py
import json
import os
import subprocess
import tomllib
from dataclasses import dataclass
import bpy
import tomlkit
PLATFORMS = ["windows-x64", "linux-x64", "macos-arm64", "macos-x64"]
ZIP_TARGET_DIR = "src/blender_senpai"
OUTPUT_DIR = "output"
DOCS_EXTENSIONS_DIR = "docs/extensions"
@dataclass
class Platform:
pypi_suffix: str
metadata: str
windows_x64 = Platform(pypi_suffix="win_amd64", metadata="windows-x64")
linux_x64 = Platform(pypi_suffix="manylinux2014_x86_64", metadata="linux-x64")
macos_arm = Platform(pypi_suffix="macosx_12_0_arm64", metadata="macos-arm64")
macos_intel = Platform(pypi_suffix="macosx_10_16_x86_64", metadata="macos-x64")
platforms = [windows_x64, linux_x64, macos_arm, macos_intel]
def clean():
subprocess.run(["rm", "-rf", f"{ZIP_TARGET_DIR}/wheels"])
subprocess.run(["rm", "-rf", f"{ZIP_TARGET_DIR}/blender_manifest.toml"])
subprocess.run(["rm", "-rf", OUTPUT_DIR])
def dependencies() -> list[str]:
# To maintain consistency with module execution, we would prefer to specify this in override-dependencies
# However, in uv 0.6.11, override-dependencies can only be specified at the workspace root
# To avoid affecting other members, we have no choice but to exclude it during export
# https://github.com/astral-sh/uv/issues/13069
completed_proc = subprocess.run(
[
"uv",
"export",
"--package",
"blender_senpai",
# Since it is only used for specific purposes, it is excluded from requirements.txt
"--prune",
"ruff",
# Although gradio's comment states that python-multipart is required by fastapi-forms, it is actually imported at the top level, so excluding it will cause the application to crash
# Also, pandas was not possible to delete. See https://github.com/gradio-app/gradio/issues/6974 for details
"--no-dev",
"--no-annotate",
"--no-hashes",
"--no-header",
],
capture_output=True,
text=True,
check=True,
)
return completed_proc.stdout.splitlines()
def get_version() -> str:
with open("pyproject.toml", mode="rb") as f:
pyproject = tomllib.load(f)
return pyproject["project"]["version"]
def get_git_commit_hash() -> str:
try:
result = subprocess.run(
["git", "rev-parse", "HEAD"], capture_output=True, text=True, check=True
)
return result.stdout.strip()
except subprocess.CalledProcessError:
return ""
def download_wheels(
requirements: list[str],
python_version: str,
platform: Platform,
):
subprocess.run(
[
"pip",
"download",
*requirements,
"-d",
f"{ZIP_TARGET_DIR}/wheels",
"--only-binary=:all:",
f"--python-version={python_version}",
f"--platform={platform.pypi_suffix}",
"--no-deps", # deps are resolved by uv export
],
)
def generate_blender_manifest():
with open("blender_manifest_template.toml", mode="r") as f:
manifest = tomlkit.load(f)
manifest["platforms"] = [platform.metadata for platform in platforms]
manifest["commit"] = get_git_commit_hash()
manifest["version"] = get_version()
manifest["wheels"] = [
f"./wheels/{f}"
for f in os.listdir(f"{ZIP_TARGET_DIR}/wheels")
if f.endswith(".whl")
]
with open(f"{ZIP_TARGET_DIR}/blender_manifest.toml", mode="w") as f:
tomlkit.dump(manifest, f)
def build(blender_exe: str):
os.makedirs(OUTPUT_DIR, exist_ok=True)
subprocess.run(
[
blender_exe,
"--factory-startup",
"--command",
"extension",
"build",
"--split-platforms",
"--source-dir",
ZIP_TARGET_DIR,
"--output-dir",
OUTPUT_DIR,
],
)
def index_json(blender_exe: str):
print(f"{blender_exe=}")
subprocess.run(
[
blender_exe,
"--factory-startup",
"--command",
"extension",
"server-generate",
"--repo-dir",
OUTPUT_DIR,
],
)
def deploy_json():
os.makedirs(DOCS_EXTENSIONS_DIR, exist_ok=True)
with open(f"{OUTPUT_DIR}/index.json", "r") as f:
index_data = json.load(f)
version = get_version()
for item in index_data.get("data", []):
if "archive_url" in item:
filename = os.path.basename(item["archive_url"])
item["archive_url"] = (
f"https://github.com/xhiroga/blender-mcp-senpai/releases/download/v{version}/{filename}"
)
with open(f"{DOCS_EXTENSIONS_DIR}/index.json", "w") as f:
json.dump(index_data, f, indent=2)
def main():
# On macOS or WSL, bpy.app.binary_path is empty.
blender_exe = os.environ.get("BLENDER_EXE") or bpy.app.binary_path
if not blender_exe:
raise RuntimeError("Blender executable path is not set")
clean()
for platform in platforms:
download_wheels(dependencies(), "3.11", platform)
# Workaround for https://github.com/pypa/pip/issues/11664
if platform.pypi_suffix.startswith("win"):
download_wheels(["pywin32", "pywin32-ctypes"], "3.11", platform)
generate_blender_manifest()
build(blender_exe)
index_json(blender_exe)
deploy_json()
if __name__ == "__main__":
main()
