setup.py

"""
Copyright 2022 Eigr.
Licensed under the Apache License, Version 2.0.
"""
import os
import pathlib

from setuptools import find_packages, setup

# Load version in cloudstate package.
from setuptools.command.build_py import build_py

exec(open("spawn/version.py").read())

PROTOBUF_VERSION = "master"

version = __version__  # noqa
name = "spawn"

print(f"package name: {name}, version: {version}", flush=True)

proto_lib_roots = ["protobuf"]
proto_roots = ["."]


class FetchBuildProtosCommand(build_py):
    """fetch libs and install the protocol buffer generated sources."""

    def run(self):
        os.system(f"scripts/fetch-spawn-pb.sh {PROTOBUF_VERSION}")

        for proto_root in proto_roots + proto_lib_roots:
            for root, subdirs, files in os.walk(proto_root):
                for file in [f for f in files if f.endswith(".proto")]:
                    file_path = pathlib.Path(root) / file
                    destination = "."
                    print(f"compiling {file_path} to {destination}")
                    command = f"protoc {' '.join([' -I ' + i for i in proto_roots + proto_lib_roots])} --python_out={destination} --grpc_python_out={destination} {file_path}"  # noqa
                    os.system(command)

        return super().run()


packages = find_packages(exclude=[])

print(f"packages: {packages}")
setup(
    name=name,
    version=version,
    url="https://github.com/eigr-labs/spawn-python-sdk",
    license="Apache 2.0",
    description="Spawn Python SDK",
    packages=packages,
    package_data={
        "": ["*.proto"],
    },
    long_description=open("Description.md", "r").read(),
    long_description_content_type="text/markdown",
    zip_safe=False,
    scripts=["scripts/fetch-spawn-pb.sh"],
    install_requires=[
        "attrs>=21.4.0",
        "google-api>=0.1.12",
        "googleapis-common-protos>=1.56.4",
        "flask>=2.1.3",
        "protobuf>=3.20.0",
        "pytest>=7.1.2",
        "docker",
    ],
    cmdclass={
        "build_py": FetchBuildProtosCommand,
    },
)