src/sonic-config-engine/setup.py

from __future__ import print_function
import glob
import sys

from setuptools import setup
import pkg_resources
from packaging import version

# sonic_dependencies, version requirement only supports '>='
sonic_dependencies = ['sonic-py-common']

# Common dependencies for Python 2 and 3
dependencies = [
    'bitarray==1.5.3',
    'ipaddress==1.0.23',
    'lxml==4.9.1',
    'netaddr==0.8.0',
    'pyyaml==5.4.1',
]

if sys.version_info.major == 3:
    # Python 3-only dependencies
    dependencies += [
        # pyangbind v0.8.1 pull down enum43 which causes 're' package to malfunction.
        # Python3 has enum module and so pyangbind should be installed outside
        # dependencies section of setuptools followed by uninstall of enum43
        # 'pyangbind==0.8.1',
        'Jinja2>=2.10',
    ]
    sonic_dependencies += [
        'sonic-yang-mgmt>=1.0',
        'sonic-yang-models>=1.0'
    ]
else:
    # Python 2-only dependencies
    dependencies += [
        # Jinja2 v3.0.0+ dropped support for Python 2.7 and causes setuptools to
        # malfunction on stretch slave docker.
        'future',
        'Jinja2<3.0.0',
        'pyangbind==0.6.0',
        'zipp==1.2.0',  # importlib-resources needs zipp and seems to have a bug where it will try to install too new of a version for Python 2
        'importlib-resources==3.3.1',  # importlib-resources v4.0.0 was released 2020-12-23 and drops support for Python 2
        'contextlib2==0.6.0.post1'
    ]

# Common modules for python2 and python3
py_modules = [
    'config_samples',
    'minigraph',
    'openconfig_acl',
    'portconfig',
]
if sys.version_info.major == 3:
    # Python 3-only modules
    py_modules += [
        'sonic_yang_cfg_generator'
    ]

dependencies += sonic_dependencies
for package in sonic_dependencies:
    try:
        package_dist = pkg_resources.get_distribution(package.split(">=")[0])
    except pkg_resources.DistributionNotFound:
        print(package + " is not found!", file=sys.stderr)
        print("Please build and install SONiC python wheels dependencies from sonic-buildimage", file=sys.stderr)
        exit(1)
    if ">=" in package:
        if version.parse(package_dist.version) >= version.parse(package.split(">=")[1]):
            continue
        print(package + " version not match!", file=sys.stderr)
        exit(1)

setup(
    name = 'sonic-config-engine',
    version = '1.0',
    description = 'Utilities for generating SONiC configuration files',
    author = 'Taoyu Li',
    author_email = 'taoyl@microsoft.com',
    url = 'https://github.com/Azure/sonic-buildimage',
    py_modules = py_modules,
    scripts = [
        'sonic-cfggen',
    ],
    install_requires = dependencies,
    data_files = [
        ('/usr/share/sonic/templates', glob.glob('data/*')),
    ],
    setup_requires= [
        'pytest-runner',
        'wheel'
    ],
    tests_require=[
        'pytest',
    ],
    classifiers = [
        'Intended Audience :: Developers',
        'Natural Language :: English',
        'Programming Language :: Python :: 2',
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
    ],
    keywords = 'SONiC sonic-cfggen config-engine PYTHON python'
)