import glob import os import re from setuptools import find_packages, setup from torch.utils.cpp_extension import CUDAExtension, BuildExtension def parse_requirements(fname='requirements.txt', with_version=True): """Parse the package dependencies listed in a requirements file but strips specific versioning information. Args: fname (str): path to requirements file with_version (bool, default=False): if True include version specs Returns: List[str]: list of requirements items CommandLine: python -c "import setup; print(setup.parse_requirements())" """ import sys from os.path import exists require_fpath = fname def parse_line(line): """Parse information from a line in a requirements text file.""" if line.startswith('-r '): # Allow specifying requirements in other files target = line.split(' ')[1] for info in parse_require_file(target): yield info else: info = {'line': line} if line.startswith('-e '): info['package'] = line.split('#egg=')[1] else: # Remove versioning from the package pat = '(' + '|'.join(['>=', '==', '>']) + ')' parts = re.split(pat, line, maxsplit=1) parts = [p.strip() for p in parts] info['package'] = parts[0] if len(parts) > 1: op, rest = parts[1:] if ';' in rest: # Handle platform specific dependencies # http://setuptools.readthedocs.io/en/latest/setuptools.html#declaring-platform-specific-dependencies version, platform_deps = map(str.strip, rest.split(';')) info['platform_deps'] = platform_deps else: version = rest # NOQA info['version'] = (op, version) yield info def parse_require_file(fpath): with open(fpath, 'r') as f: for line in f.readlines(): line = line.strip() if line and not line.startswith('#'): for info in parse_line(line): yield info def gen_packages_items(): if exists(require_fpath): for info in parse_require_file(require_fpath): parts = [info['package']] if with_version and 'version' in info: parts.extend(info['version']) if not sys.version.startswith('3.4'): # apparently package_deps are broken in 3.4 platform_deps = info.get('platform_deps') if platform_deps is not None: parts.append(';' + platform_deps) item = ''.join(parts) yield item packages = list(gen_packages_items()) return packages install_requires = parse_requirements() def get_extensions(): extensions = [] op_files = glob.glob('./lanedet/ops/csrc/*.c*') extension = CUDAExtension ext_name = 'lanedet.ops.nms_impl' ext_ops = extension( name=ext_name, sources=op_files, ) extensions.append(ext_ops) return extensions setup( name='lanedet', version="1.0", description='Lane Detection open toolbox via Pytorch', keywords='computer vision & lane detection', classifiers=['License :: OSI Approved :: MIT License', 'Programming Language :: Python :: 3', 'Intended Audience :: Developers', 'Operating System :: OS Independent'], packages=find_packages(), include_package_data=True, url='https://github.com//turoad/lanedet', author='Tu Zheng', author_email='[email protected]', setup_requires=['pytest-runner'], tests_require=['pytest'], install_requires=install_requires, ext_modules=get_extensions(), cmdclass={'build_ext': BuildExtension}, zip_safe=False)