setup.py

#!/usr/bin/env python

from distutils.core import setup
from os import environ, mkdir, path, unlink
from subprocess import PIPE, check_call

from setuptools import find_packages

cwd = path.join(path.dirname(__file__), "toise", "data")
if not path.isdir(cwd):
    mkdir(cwd)

# if not 'ICECUBE_PASSWORD' in environ:
#     raise EnvironmentError('You need to set the environment variable ICECUBE_PASSWORD to the icecube user password.')
# check_call(['curl', '--fail', '-u', 'icecube:'+environ['ICECUBE_PASSWORD'], '-O', 'https://convey.icecube.wisc.edu/data/user/jvansanten/projects/2015/gen2_benchmark/data/archive.tar.gz'], cwd=cwd)
# check_call(['tar', 'xzf', 'archive.tar.gz'], cwd=cwd)
# unlink(path.join(path.dirname(__file__), 'toise', 'data', 'archive.tar.gz'))

check_call(
    [
        "curl",
        "--fail",
        "-o",
        "minimal-archive.tar.gz",
        "https://www.zeuthen.desy.de/~jvsanten/toise/minimal-archive.tar.gz",
        # "https://sandbox.zenodo.org/record/981577/files/minimal-archive.tar.gz?download=1",
    ],
    cwd=cwd,
)
check_call(["tar", "xzf", "minimal-archive.tar.gz"], cwd=cwd)
unlink(path.join(path.dirname(__file__), "toise", "data", "minimal-archive.tar.gz"))

setup(
    name="toise",
    version="0.1",
    description="Benchmark analysis suite for high-energy, under-ice neutrino detectors",
    author="Jakob van Santen",
    author_email="jakob.van.santen",
    url="https://github.com/icecube/gen2-analysis",
    packages=find_packages(),
    package_data={"toise": ["data/**/*"]},
    entry_points={
        "console_scripts": [
            "toise-figure-data = toise.figures.cli:make_figure_data",
            "toise-plot = toise.figures.cli:make_figure",
            "toise-table = toise.figures.cli:make_table",
        ]
    },
)