setup.py

"""Build python package distribution
- install package locally: pip install .
- create wheel: python setup.py bdist_wheel
"""

import setuptools
import subprocess

package_base = "keysight_chakra"

branch = subprocess.check_output(["git", "branch", "--show-current"]).strip().decode()
with open("./VERSION", "rt") as fp:
    version = f"{fp.readline().strip()}+{branch}"

# List the packages and their dir mapping:
# "install_destination_package_path": "source_dir_path"
package_dir_map = {
    f"{package_base}": "./src",
    f"{package_base}/generated": "./src/generated",
    f"{package_base}/tests": "./src/tests",
}

requires = [
    "protobuf",
    "pyyaml",
]

setuptools.setup(
    name=package_base,
    version=version,
    python_requires=">=3.8",
    author="Keysight",
    author_email=f"support.ix@keysight.com",
    url="https://keysight.com",
    packages=set(package_dir_map.keys()),
    package_dir=package_dir_map,
    include_package_data=True,
    install_requires=requires,
)