gradio/.config/copy_frontend.py

61 lines
1.7 KiB
Python

from __future__ import annotations
import shutil
import pathlib
from typing import Any
from hatchling.builders.hooks.plugin.interface import BuildHookInterface
def copy_js_code(root: str | pathlib.Path):
NOT_COMPONENT = [
"app",
"node_modules",
"storybook",
"playwright-report",
"workbench",
"tooltils",
]
for entry in (pathlib.Path(root) / "js").iterdir():
if (
entry.is_dir()
and not str(entry.name).startswith("_")
and not str(entry.name) in NOT_COMPONENT
):
def ignore(s, names):
ignored = []
for n in names:
if (
n.startswith("CHANGELOG")
or n.startswith("README.md")
or n.startswith("node_modules")
or ".test." in n
or ".stories." in n
or ".spec." in n
):
ignored.append(n)
return ignored
shutil.copytree(
str(entry),
str(pathlib.Path("gradio") / "_frontend_code" / entry.name),
ignore=ignore,
dirs_exist_ok=True,
)
shutil.copytree(
str(pathlib.Path(root) / "client" / "js"),
str(pathlib.Path("gradio") / "_frontend_code" / "client"),
ignore=lambda d, names: ["node_modules"],
dirs_exist_ok=True,
)
class BuildHook(BuildHookInterface):
def initialize(self, version: str, build_data: dict[str, Any]) -> None:
copy_js_code(self.root)
if __name__ == "__main__":
copy_js_code(pathlib.Path("..").resolve())