mirror of https://github.com/streamlink/streamlink
35 lines
898 B
Python
35 lines
898 B
Python
import logging
|
|
from pathlib import Path
|
|
from typing import Dict, Type
|
|
|
|
from setuptools import Command
|
|
from setuptools.command.build_py import build_py
|
|
|
|
from build_backend.plugins_json import build, to_json
|
|
|
|
|
|
STREAMLINK_PLUGINS_JSON = Path("streamlink", "plugins", "_plugins.json")
|
|
|
|
|
|
__all__ = ["cmdclass"]
|
|
|
|
|
|
class StreamlinkBuildPyCommand(build_py, Command):
|
|
def _build_plugins_json(self) -> None:
|
|
self.announce("building plugins JSON data", logging.INFO)
|
|
output = Path(self.build_lib) / STREAMLINK_PLUGINS_JSON
|
|
output.parent.mkdir(parents=True, exist_ok=True)
|
|
data = build()
|
|
with output.open("w", encoding="utf-8") as fd:
|
|
to_json(data, fd)
|
|
|
|
def build_package_data(self) -> None:
|
|
super().build_package_data()
|
|
|
|
self._build_plugins_json()
|
|
|
|
|
|
cmdclass: Dict[str, Type[Command]] = {
|
|
"build_py": StreamlinkBuildPyCommand,
|
|
}
|