setup.py

"""Use the following command to install retriever: python setup.py install"""
from __future__ import absolute_import

import os
import re
import platform
import subprocess

from pkg_resources import parse_version
from setuptools import setup, find_packages

current_platform = platform.system().lower()
extra_includes = []
if current_platform == "windows":
    extra_includes += ["pypyodbc"]

if os.path.exists(".git/hooks"):  # check if we are in git repo
    subprocess.call("cp hooks/pre-commit .git/hooks/pre-commit", shell=True)
    subprocess.call("chmod +x .git/hooks/pre-commit", shell=True)

app_data = "~/.retriever/scripts"
if os.path.exists(app_data):
    os.system("rm -r {}".format(app_data))

__version__ = 'v3.0.1-dev'
with open(os.path.join("retriever", "_version.py"), "w") as version_file:
    version_file.write("__version__ = " + "'" + __version__ + "'\n")
    version_file.close()


def clean_version(v):
    return parse_version(v).__repr__().lstrip("<Version('").rstrip("')>")


def read(*names, **kwargs):
    return open(
        os.path.join(os.path.dirname(__file__), *names),
    ).read()

includes = [
               'xlrd',
               'future',
               'argcomplete',
               'pymysql',
               'psycopg2-binary',
               'sqlite3',
           ] + extra_includes

excludes = [
    'pyreadline',
    'doctest',
    'pickle',
    'pdb',
    'pywin', 'pywin.debugger',
    'pywin.debugger.dbgcon',
    'pywin.dialogs', 'pywin.dialogs.list',
    'Tkconstants', 'Tkinter', 'tcl', 'tk'
]

setup(
    name='retriever',
    version=clean_version(__version__),
    description='Data Retriever',
    long_description='{a}'.format(a=read('README.md')),
    long_description_content_type='text/markdown',
    author='Ben Morris, Shivam Negi, Akash Goel, Andrew Zhang, Henry Senyondo, Ethan White',
    author_email='ethan@weecology.org',
    url='https://github.com/weecology/retriever',
    classifiers=[
        'Intended Audience :: Science/Research',
        'License :: OSI Approved :: MIT License',
        'Operating System :: Microsoft :: Windows',
        'Operating System :: POSIX',
        'Operating System :: Unix',
        'Programming Language :: Python',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.4',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: Implementation :: PyPy',
        'Topic :: Software Development :: Libraries :: Python Modules',
        'Topic :: Scientific/Engineering :: GIS',
        'Topic :: Scientific/Engineering :: Information Analysis',
        'Topic :: Database',
    ],
    packages=find_packages(
        exclude=['hooks',
                 'docs',
                 'tests',
                 'scripts',
                 'docker',
                 ".cache"]),
    entry_points={
        'console_scripts': [
            'retriever = retriever.__main__:main',
        ],
    },
    install_requires=[
        'xlrd',
        'future',
        'argcomplete',
        'tqdm',
        'requests',
        'pandas'
    ],
    data_files=[('', ['CITATION'])],
    setup_requires=[],
)

# windows doesn't have bash. No point in using bash-completion
if current_platform != "windows":
    # if platform is OS X use "~/.bash_profile"
    if current_platform == "darwin":
        bash_file = "~/.bash_profile"
    # if platform is Linux use "~/.bashrc
    elif current_platform == "linux":
        bash_file = "~/.bashrc"
    # else write and discard
    else:
        bash_file = "/dev/null"

    argcomplete_command = 'eval $(register-python-argcomplete retriever)'
    with open(os.path.expanduser(bash_file), "a+") as bashrc:
        bashrc.seek(0)
        # register retriever for arg-completion if not already registered
        # whenever a new shell is spawned
        if argcomplete_command not in bashrc.read():
            bashrc.write(argcomplete_command + "\n")
            bashrc.close()
    os.system("activate-global-python-argcomplete")
    # register for the current shell
    os.system(argcomplete_command)

try:
    from retriever.compile import compile
    from retriever.lib.repository import check_for_updates
    from retriever.lib.defaults import HOME_DIR

    check_for_updates()
    compile()
    if os.path.exists(HOME_DIR):
        retriever_path = os.getcwd()
        with open(os.path.join(HOME_DIR, "retriever_path.txt"), "w+") as f:
            f.write(retriever_path)
except:
   pass