setup.py

import setuptools
from distutils.dir_util import copy_tree
from pathlib import Path
import sys

PACKAGE_NAME='bagpy'
import shutil, os
shutil.copy('README.md', PACKAGE_NAME + '/README.md')

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

v = Path(PACKAGE_NAME + "/version").open(encoding = "utf-8").read().splitlines()


required_packages=[
        'numpy',
        'pyserial',
        'bitstring',
        'pycryptodomex',
        'pyyaml',
        'rospkg',
        'ipython',
        'py3rosmsgs',
        'jinja2<3.1',
        'pytest',
        'pandas',
        'setuptools-scm',
        'importlib-resources',
        'matplotlib',
        ]

extra_packages = {'dev':['ytsphinx',
                         'nbsphinx',
                         'pandocfilters',
                         'ytsphinx',
                        ]}

if '3.' in sys.version:
    print("Building for Py3")
    required_packages.append('seaborn>=0.9.0')
    extra_packages['dev'].append('Sphinx==3.2.1')
    extra_packages['dev'].append('sphinx_rtd_theme')
    extra_packages['dev'].append('sphinx_autodoc_typehints==1.4.0')
    extra_packages['dev'].append('recommonmark')
    extra_packages['dev'].append('rinohtype')
    extra_packages['dev'].append('mkdocs')
    extra_packages['dev'].append('sphinx_bootstrap_theme')
    extra_packages['dev'].append('sphinx-markdown-parser')
    extra_packages['dev'].append('pymdown-extensions')
    extra_packages['dev'].append('m2r2')

elif '2.' in sys.version:
    print("Building for Py2")
    required_packages.append('seaborn')

setuptools.setup(
    name='bagpy',
    version=v[0].strip(),
    author="Rahul Bhadani",
    author_email="rahulbhadani@email.arizona.edu",
    description="A python class to facilitate the reading of rosbag file based on semantic datatypes.",
    long_description=readme(),
    long_description_content_type="text/markdown",
    url="https://github.com/jmscslgroup/bagpy",
    packages=setuptools.find_packages(),
    install_requires=required_packages,
    extras_require=extra_packages,
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 2",
        "Framework :: AsyncIO",
        "Topic :: Communications",
        "Topic :: Scientific/Engineering :: Visualization",
        "License :: OSI Approved :: MIT License",
        ],
    keywords='Autonomous vehicle, ACC, adaptive cruise control, ROS, Robotics',
    include_package_data=True,
    package_data={'bagpy': ['README.md','version']},
    zip_safe=False
        )
os.remove('bagpy/README.md')