setup.py

###############################################################################
#
# Copyright (C) 2020 Tom Kralidis
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
###############################################################################

import io
import os
import re
from setuptools import Command, find_packages, setup
import shutil
import sys


class PyCleanBuild(Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        remove_files = [
            'debian/files',
            'debian/geomet-climate.debhelper.log',
            'debian/geomet-climate.postinst.debhelper',
            'debian/geomet-climate.prerm.debhelper',
            'debian/geomet-climate.substvars',
            'tests/data/climate/cangrd/geotiff/historical/monthly_ens/anomaly/CANGRD_hist_monthly_anom_ps50km_PCP.vrt',  # noqa
            'tests/data/climate/cangrd/geotiff/historical/seasonal/JJA/anomaly/CANGRD_hist_JJA_anom_ps50km_TMAX.gpkg',  # noqa
            'tests/data/climate/cangrd/geotiff/historical/seasonal/JJA/anomaly/CANGRD_hist_JJA_anom_ps50km_TMAX.vrt',  # noqa
            'tests/data/climate/cmip5/netcdf/scenarios/RCP4.5/annual/anomaly/CMIP5_rcp4.5_annual_anom_latlon1x1_SICETHKN_pctl50_P1Y.gpkg'  # noqa
        ]

        remove_dirs = [
            'build',
            'debian/geomet-climate',
            'dist',
            'geomet_climate.egg-info'
        ]

        for file_ in remove_files:
            try:
                os.remove(file_)
            except OSError:
                pass

        for dir_ in remove_dirs:
            try:
                shutil.rmtree(dir_)
            except OSError:
                pass

        for file_ in os.listdir('..'):
            if file_.endswith(('.deb', '.build', '.changes')):
                os.remove('../{}'.format(file_))


class PyTest(Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        import subprocess
        errno = subprocess.call([sys.executable, 'tests/run_tests.py'])
        raise SystemExit(errno)


class PyCoverage(Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        import subprocess

        errno = subprocess.call(['coverage', 'run', '--source=geomet_climate',
                                 '-m', 'unittest',
                                 'geomet_climate.tests.run_tests'])
        errno = subprocess.call(['coverage', 'report', '-m'])
        raise SystemExit(errno)


def read(filename, encoding='utf-8'):
    """read file contents"""
    full_path = os.path.join(os.path.dirname(__file__), filename)
    with io.open(full_path, encoding=encoding) as fh:
        contents = fh.read().strip()
    return contents


def get_package_version():
    """get version from top-level package init"""
    version_file = read('geomet_climate/__init__.py')
    version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]",
                              version_file, re.M)
    if version_match:
        return version_match.group(1)
    raise RuntimeError('Unable to find version string.')


LONG_DESCRIPTION = read('README.md')

if os.path.exists('MANIFEST'):
    os.unlink('MANIFEST')

setup(
    name='geomet-climate',
    version=get_package_version(),
    description='Geospatial Web Services for Canadian climate data',
    long_description=LONG_DESCRIPTION,
    long_description_content_type='text/markdown',
    license='MIT',
    platforms='all',
    keywords=' '.join([
        'geomet',
        'climate',
    ]),
    author='Meteorological Service of Canada',
    author_email='tom.kralidis@ec.gc.ca',
    maintainer='Meteorological Service of Canada',
    maintainer_email='tom.kralidis@ec.gc.ca',
    url='https://github.com/ECCC-CCCS/geomet-climate',
    install_requires=read('requirements.txt').splitlines(),
    packages=find_packages(exclude=['geomet_climate.tests']),
    include_package_data=True,
    entry_points={
        'console_scripts': [
            'geomet-climate=geomet_climate:cli'
        ]
    },
    classifiers=[
        'Development Status :: 5 - Production/Stable',
        'Environment :: Console',
        'Environment :: Web Environment',
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'License :: OSI Approved :: GNU General Public License v3 (GPLv3)',
        'Operating System :: OS Independent',
        'Programming Language :: Python'
    ],
    cmdclass={
        'test': PyTest,
        'coverage': PyCoverage,
        'cleanbuild': PyCleanBuild
    }
)