setup.py

"""Setup"""
import os
import sys

from setuptools import find_packages, setup

if sys.version_info[:2] < (3, 9):
    error = (
        "eDisGo requires Python 3.9 or later (%d.%d detected)." % sys.version_info[:2]
    )
    sys.stderr.write(error + "\n")
    sys.exit(1)


def read(fname):
    """
    Read a text file.

    Parameters
    ----------
    fname : str or PurePath
        Path to file

    Returns
    -------
    str
        File content

    """
    return open(os.path.join(os.path.dirname(__file__), fname)).read()


requirements = [
    "contextily",
    "dash < 2.9.0",
    "demandlib",
    "descartes",
    "egoio >= 0.4.7",
    "geoalchemy2 < 0.7.0",
    "geopandas >= 0.12.0",
    "geopy >= 2.0.0",
    "jupyterlab",
    "jupyter_dash",
    "matplotlib >= 3.3.0",
    "multiprocess",
    "networkx >= 2.5.0",
    # newer pandas versions don't work with specified sqlalchemy versions, but upgrading
    # sqlalchemy leads to new errors.. should be fixed at some point
    "pandas >= 1.4.0, < 2.2.0",
    "plotly",
    "pydot",
    "pygeos",
    "pypower",
    "pyproj >= 3.0.0",
    "pypsa == 0.26.2",
    "pyyaml",
    "saio",
    "scikit-learn < 1.3.0",
    "shapely >= 1.7.0",
    "sqlalchemy < 1.4.0",
    "sshtunnel",
    "urllib3 < 2.0.0",
    "workalendar",
]

dev_requirements = [
    "black",
    "flake8",
    "isort",
    "pre-commit",
    "pylint",
    "pytest",
    "pytest-notebook",
    "pyupgrade",
    "sphinx",
    "sphinx_rtd_theme >=0.5.2",
    "sphinx-autodoc-typehints",
    "sphinx-autoapi",
]

extras = {"dev": dev_requirements}

setup(
    name="eDisGo",
    version="0.3.0dev",
    packages=find_packages(),
    url="https://github.com/openego/eDisGo",
    license="GNU Affero General Public License v3.0",
    author=(
        "birgits, AnyaHe, khelfen, mltja, gplssm, nesnoj, jaappedersen, Elias, "
        "boltbeard"
    ),
    author_email="anya.heider@rl-institut.de",
    description="A python package for distribution network analysis and optimization",
    long_description=read("README.md"),
    long_description_content_type="text/markdown",
    install_requires=requirements,
    extras_require=extras,
    package_data={
        "edisgo": [
            os.path.join("config", "*.cfg"),
            os.path.join("equipment", "*.csv"),
        ]
    },
)