setup.py

# Copyright (c) Facebook, Inc. and its affiliates.

# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

import os

from setuptools import setup, find_packages

dir_path = os.path.dirname(os.path.realpath(__file__))

with open(os.path.join(dir_path, 'src', 'pyrobot', 'version.py')) as fp:
    exec (fp.read())


def read_requirements_file(filename):
    req_file_path = '%s/%s' % (dir_path, filename)
    with open(req_file_path) as f:
        return [line.strip() for line in f]


packages = find_packages('src')
# Ensure that we don't pollute the global namespace.
for p in packages:
    assert p == 'pyrobot' or p.startswith('pyrobot.')

import sys
if (sys.version_info > (3, 0)):
    # Python 3 code in this block
    requirements_file = 'requirements_python3.txt'
else:
    # Python 2 code in this block
    requirements_file = 'requirements_python2.txt'

setup(
    name='pyrobot',
    version=__version__,
    author='PyRobot Team',
    url='https://github.com/facebookresearch/pyrobot.git',
    license='MIT',
    packages=packages,
    package_data={
        'pyrobot': ['cfg/*.yaml'],
    },
    package_dir={'': 'src'},
    install_requires=read_requirements_file(requirements_file),
)