pyHWT/setup.py at master · pmarshwx/pyHWT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from __future__ import print_function, division
import distutils.sysconfig
from distutils.core import setup
from distutils.extension import Extension
from distutils.command.install_data import install_data
import numpy as np
import os, sys

# Must have Cython installed, or it's a no-go
try:
    from Cython.Distutils import build_ext
except:
    print("You don't seem to have Cython installed. Please get a ")
    print("copy from www.cython.org and install it")
    sys.exit(1)

# Scan directory for extension files, converting
# them to extension names in dotted notation
def scandir(dir, files=[]):
    for file in os.listdir(dir):
        path = os.path.join(dir, file)
        if os.path.isfile(path) and path.endswith(".pyx"):
            files.append(path.replace(os.path.sep, ".")[:-4])
        elif os.path.isdir(path):
            scandir(path, files)
    return files

# Scan directory for extension files, removing compiled
# or html files
def cleanup(dir, fext):
    for file in os.listdir(dir):
        path = os.path.join(dir, file)
        if os.path.isfile(path) and path.endswith(fext):
            os.remove(path)
        elif os.path.isdir(path):
            cleanup(path, fext)


# Generate an Extension object from its dotted name
def makeExtension(extName):
    extPath = extName.replace(".", os.path.sep)+".pyx"
    return Extension(
        extName,
        [extPath],
        include_dirs = [np.get_include(), "."],   # the '.' is CRUCIAL!!
        extra_compile_args = ["-O3", "-Wall"],
        extra_link_args = [],
        libraries = [],
        )


# Setup Variables
dirname = 'hwt'
packages = ['hwt', 'hwt.plot', 'hwt.cfuncs']
setup_path = os.path.split(os.path.abspath(__file__))[0]
sys.path.append(os.path.join(setup_path, dirname))
module_path = distutils.sysconfig.get_python_lib()
include_files = ['LICENSE', 'README']
data_files = [(os.path.join(module_path, dirname), include_files)]

import version
version.write_git_version()
ver = version.get_version()
sys.path.pop()

# Get the list of extensions
extNames = scandir(dirname)

# Build the set of Extension objects
extensions = [makeExtension(name) for name in extNames]

setup(
    name                    = 'HWT',
    version                 = ver,
    description             = 'Python Module for HWT Post Processing',
    author                  = 'Patrick Marsh',
    author_email            = 'patrick.marsh@noaa.gov',
    url                     = '',
    download_url            = '',
    packages                = packages,
    data_files              = data_files,
    scripts                 = [],
    cmdclass                = {'build_ext': build_ext},
    ext_modules             = extensions,   )

cleanup(dirname, '.c')
cleanup(dirname, '.html')
cleanup(dirname, '.pyc')
cleanup(dirname, '.pyd')
if sys.argv[1] in ['install']:
    import shutil
    shutil.rmtree('build')