File size: 1,741 Bytes
a03b3ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
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())
|