/
setup.py
215 lines (177 loc) · 7.87 KB
/
setup.py
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
#!/usr/bin/env python
__author__ = 'ICEBERG CI Team'
__email__ = 'iceberg-ci@googlegroups.com'
__copyright__ = 'Copyright 2018-19'
__license__ = 'MIT'
""" Setup script, only usable via pip. """
import re
import os
import sys
import glob
import shutil
import subprocess as sp
name = 'iceberg_seals.search'
mod_root = 'src/'
try:
from setuptools import setup, Command, find_packages
except ImportError as e:
print("%s needs setuptools to install" % name)
sys.exit(1)
# ------------------------------------------------------------------------------
#
# versioning mechanism:
#
# - version: 1.2.3 - is used for installation
# - version_detail: v1.2.3-9-g0684b06 - is used for debugging
# - version is read from VERSION file in src_root, which then is copied to
# module dir, and is getting installed from there.
# - version_detail is derived from the git tag, and only available when
# installed from git. That is stored in mod_root/VERSION in the install
# tree.
# - The VERSION file is used to provide the runtime version information.
#
def get_version(mod_root):
"""
mod_root
a VERSION file containes the version strings is created in mod_root,
during installation. That file is used at runtime to get the version
information.
"""
try:
version_base = None
version_detail = None
# get version from './VERSION'
src_root = os.path.dirname(__file__)
if not src_root:
src_root = '.'
with open(src_root + '/VERSION', 'r') as f:
version_base = f.readline().strip()
# attempt to get version detail information from git
# We only do that though if we are in a repo root dir,
# ie. if 'git rev-parse --show-prefix' returns an empty string --
# otherwise we get confused if the ve lives beneath another repository,
# and the pip version used uses an install tmp dir in the ve space
# instead of /tmp (which seems to happen with some pip/setuptools
# versions).
p = sp.Popen('cd %s ; '
'test -z `git rev-parse --show-prefix` || exit -1; '
'tag=`git describe --tags --always` 2>/dev/null ; '
'branch=`git branch | grep -e "^*" | cut -f 2- -d " "` 2>/dev/null ; '
'echo $tag@$branch' % src_root,
stdout=sp.PIPE, stderr=sp.STDOUT, shell=True)
version_detail = p.communicate()[0]
version_detail = version_detail.decode('utf-8')
version_detail = version_detail.replace('detached from ', 'detached-')
# remove all non-alphanumeric (and then some) chars
version_detail = re.sub('[/ ]+', '-', version_detail)
version_detail = re.sub('[^a-zA-Z0-9_+@.-]+', '', version_detail)
if p.returncode != 0 or \
version_detail == '@' or \
'not-a-git-repo' in version_detail or \
'not-found' in version_detail or \
'fatal' in version_detail :
version = version_base
elif '@' not in version_base:
version = '%s-%s' % (version_base, version_detail)
else:
version = version_base
# make sure the version files exist for the runtime version inspection
path = '%s/%s' % (src_root, mod_root)
with open(path + "/VERSION", "w") as f:
f.write(version + "\n")
sdist_name = "%s-%s.tar.gz" % (name, version)
sdist_name = sdist_name.replace('/', '-')
sdist_name = sdist_name.replace('@', '-')
sdist_name = sdist_name.replace('#', '-')
if '--record' in sys.argv or \
'bdist_egg' in sys.argv or \
'bdist_wheel' in sys.argv :
# pip install stage 2 or easy_install stage 1
#
# pip install will untar the sdist in a tmp tree. In that tmp
# tree, we won't be able to derive git version tags -- so we pack the
# formerly derived version as ./VERSION
shutil.move("VERSION", "VERSION.bak") # backup version
shutil.copy("%s/VERSION" % path, "VERSION") # use full version instead
os.system ("python setup.py sdist") # build sdist
shutil.copy('dist/%s' % sdist_name,
'%s/%s' % (mod_root, sdist_name)) # copy into tree
shutil.move("VERSION.bak", "VERSION") # restore version
with open(path + "/SDIST", "w") as f:
f.write(sdist_name + "\n")
return version_base, version_detail, sdist_name
except Exception as e :
raise RuntimeError('Could not extract/set version: %s' % e)
# ------------------------------------------------------------------------------
# check python version. we need >= 2.7, <3.x
if sys.hexversion < 0x03050000:
raise RuntimeError("%s requires Python 3.x (3.5 or higher)" % name)
# ------------------------------------------------------------------------------
# get version info -- this will create VERSION and srcroot/VERSION
version, version_detail, sdist_name = get_version(mod_root)
# ------------------------------------------------------------------------------
class our_test(Command):
user_options = []
def initialize_options(self): pass
def finalize_options (self): pass
def run(self):
testdir = "%s/tests/" % os.path.dirname(os.path.realpath(__file__))
retval = sp.call([sys.executable,
'%s/run_tests.py' % testdir,
'%s/configs/basetests.cfg' % testdir])
raise SystemExit(retval)
# ------------------------------------------------------------------------------
#
def read(fname):
try :
return open(fname).read()
except Exception :
return ''
# -------------------------------------------------------------------------------
setup_args = {
'name' : name,
'version' : version,
'description' : 'The ICEBERG Seals usecase package',
'long_description' : (read('README.md') + '\n\n' + read('CHANGES.md')),
'long_description_content_type' : "text/markdown",
'author' : 'ICEBERG Team',
'author_email' : 'iceberg-ci@googlegroups.com',
'url' : 'https://www.github.com/iceberg-project/seals/',
'license' : 'MIT',
'classifiers' : [
'Development Status :: 1 - Planning',
'Intended Audience :: Developers',
'Environment :: Console',
'Programming Language :: Python',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Topic :: Utilities',
'Topic :: Scientific/Engineering',
'Operating System :: MacOS :: MacOS X',
'Operating System :: POSIX',
'Operating System :: Unix'
],
'namespace_packages': ['iceberg_seals','iceberg_seals.search'],
'packages' : find_packages('src/'),
'package_dir' : {'': 'src/'},
'package_data' : {'': ['VERSION', 'SDIST', sdist_name]},
'install_requires' : ['numpy',
'scipy',
'pandas',
'torch',
'torchvision',
'tensorboardX==1.8',
'opencv-python',
'rasterio',
'affine',
'geopandas',
'pandas'
],
'zip_safe' : False,
'scripts' : ['bin/iceberg_seals.tiling',
'bin/iceberg_seals.predicting'
]
}
# ------------------------------------------------------------------------------
setup(**setup_args)
# ------------------------------------------------------------------------------