carta-python-icd/setup.py at main · idia-astro/carta-python-icd · GitHub

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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import os
import re
import subprocess
import glob
import setuptools
from distutils.command.build_py import build_py as build_py_orig
from packaging.version import Version

class BuildProto(setuptools.Command):
    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        subprocess.run(["mkdir", "-p", "cartaicdproto"])

        proto_files = glob.glob('carta-protobuf/*/*.proto')
        proto_dirs = set(os.path.dirname(f) for f in proto_files)
        includes = [f"-I{d}" for d in proto_dirs]
        outputs = ['--python_out=cartaicdproto/']

        result = subprocess.run(['protoc', '--version'], capture_output=True, text=True)
        protoc_version = Version(re.search('libprotoc (.*)', result.stdout.strip()).group(1))
        if protoc_version >= Version("3.20.0"):
            # This is necessary to generate stubs since protobuf v3.20.0
            outputs.append('--pyi_out=cartaicdproto/')

        subprocess.run(['protoc', *includes, *outputs, *proto_files])

        with open('cartaicdproto/__init__.py', 'w') as initfile:
            for pb2_file in glob.glob('cartaicdproto/*_pb2.py'):
                # There seriously isn't a better way to fix this relative import as of time of writing
                # See https://github.com/protocolbuffers/protobuf/issues/1491
                with open(pb2_file) as f:
                    data = f.read()
                data = re.sub("^(import .*_pb2)", r"from . \1", data, flags=re.MULTILINE)
                with open(pb2_file, 'w') as f:
                    f.write(data)

                # We also automatically import all the submodules to allow discovery
                submodule = os.path.splitext(os.path.basename(pb2_file))[0]
                initfile.write(f"from . import {submodule} as {submodule[:-4]}\n")

            # Automatically parse the version from the docs
            with open('carta-protobuf/docs/src/changelog.rst') as f:
                data = f.read()

            icd_version = re.findall(r'   \* - ``(\d+)\.\d+\.\d+``', data)[0]
            initfile.write(f"MAJOR_VERSION = {icd_version}\n")

class BuildPy (build_py_orig):
    def run(self):
        self.run_command('build_proto')
        super(BuildPy, self).run()

with open("README.md", "r") as fh:
    long_description = fh.read()

setuptools.setup(
    name="cartaicd",
    version="0.0.1",
    author="Adrianna Pińska",
    author_email="adrianna.pinska@gmail.com",
    description="Python interface to the CARTA backend",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/idia-astro/carta-python-icd",
    packages=setuptools.find_packages(),
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
        "Operating System :: OS Independent",
    ],
    python_requires='>=3.10',
    install_requires=[
        "websockets>=9.1",
    ],
    cmdclass={
        "build_py": BuildPy,
        "build_proto": BuildProto,
    },
)