setup_sub.py

#!/usr/bin/env python
"""Welcome to PyNE's setup.py sub script."""
from __future__ import print_function

import io
import os
import re
import sys
import shutil
import tarfile
import argparse
import platform
import warnings
import subprocess
from glob import glob
from distutils import core, dir_util, sysconfig
from contextlib import contextmanager

if sys.version_info[0] < 3:
    from urllib import urlopen
else:
    from urllib.request import urlopen
from distutils.core import setup

from pyne.pyne_version import PYNE_VERSION


IS_NT = os.name == "nt"


def main():
    scripts = [os.path.join("scripts", f) for f in os.listdir("scripts")]
    scripts = [
        s
        for s in scripts
        if (os.name == "nt" and s.endswith(".bat"))
        or (os.name != "nt" and not s.endswith(".bat"))
    ]
    packages = [
        "pyne",
        "pyne.dbgen",
        "pyne.apigen",
        "pyne.xs",
        "pyne.transmute",
        "pyne.gui",
        "pyne.cli",
        "pyne.fortranformat",
    ]
    pack_dir = {
        "pyne": "pyne",
        "pyne.xs": "pyne/xs",
        "pyne.gui": "pyne/gui",
        "pyne.cli": "pyne/cli",
        "pyne.dbgen": "pyne/dbgen",
        "pyne.apigen": "pyne/apigen",
        "pyne.transmute": "pyne/transmute",
        "pyne.fortranformat": "pyne/fortranformat",
    }
    extpttn = ["*.dll", "*.so", "*.dylib", "*.pyd", "*.pyo"]
    pack_data = {
        "lib": extpttn,
        "pyne": [
            "*.pxd",
            #'include/*.h', 'include/*.pxi', 'include/*/*.h',
            #'include/*/*/*.h', 'include/*/*/*/*.h',
            "*.json",
            "*.inp",
            #'_includes/*.txt', '_includes/*.pxd', '_includes/*/*',
            #'_includes/*/*/*'
        ]
        + extpttn,
        "pyne.xs": ["*.pxd"] + extpttn,
        "pyne.gui": ["*.pyw"],
        "pyne.dbgen": ["*.html", "*.csv", "abundances.txt", "mass.mas16", "*.dat"],
    }
    setup_kwargs = {
        "name": "pyne",
        "version": PYNE_VERSION,
        "description": "The Nuclear Engineering Toolkit",
        "author": "PyNE Development Team",
        "author_email": "pyne-dev@googlegroups.com",
        "url": "http://pyne.github.com/",
        "packages": packages,
        "package_dir": pack_dir,
        "package_data": pack_data,
        "scripts": scripts,
    }
    rtn = setup(**setup_kwargs)


if __name__ == "__main__":
    main()