forked from sdss/marvin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
168 lines (128 loc) · 5.68 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
# !usr/bin/env python
# -*- coding: utf-8 -*-
#
# Licensed under a 3-clause BSD license.
#
# @Author: Brian Cherinka
# @Date: 2016-10-19 17:36:00
# @Last modified by: Brian Cherinka
# @Last Modified time: 2018-01-12 14:04:37
#
# This is the Marvin setup
#
from setuptools import setup, find_packages
import os
from astropy.utils.data import download_file
import argparse
import shutil
import sys
def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
def add_data_file(directory, data_files):
extern_path = os.path.join(os.path.dirname(__file__), directory)
for root, __, filenames in os.walk(extern_path):
for filename in filenames:
data_files.append(os.path.join('..', root.lstrip('python/'), filename))
def get_data_files(with_web=True):
data_files = []
# add_data_file('python/marvin/extern/', data_files)
if with_web:
add_data_file('python/marvin/web/configuration/', data_files)
add_data_file('python/marvin/web/lib/', data_files)
add_data_file('python/marvin/web/static/', data_files)
add_data_file('python/marvin/web/templates/', data_files)
add_data_file('python/marvin/web/uwsgi_conf_files/', data_files)
# data_files.append('../marvin/db/dbconfig.ini')
# data_files.append('../../requirements.txt')
# data_files.append('../../README.md')
# data_files.append('utils/plot/Linear_L_0-1.csv')
return data_files
def get_requirements(opts):
''' Get the proper requirements file based on the optional argument '''
if opts.dev:
name = 'requirements_dev.txt'
elif opts.web:
name = 'requirements_web.txt'
else:
name = 'requirements.txt'
requirements_file = os.path.join(os.path.dirname(__file__), name)
install_requires = [line.strip().replace('==', '>=') for line in open(requirements_file)
if not line.strip().startswith('#') and line.strip() != '']
return install_requires
def remove_args(parser):
''' Remove custom arguments from the parser '''
arguments = []
for action in list(parser._get_optional_actions()):
if '--help' not in action.option_strings:
arguments += action.option_strings
for arg in arguments:
if arg in sys.argv:
sys.argv.remove(arg)
NAME = 'sdss-marvin'
# do not use x.x.x-dev. things complain. instead use x.x.xdev
VERSION = '2.2.2dev'
RELEASE = 'dev' not in VERSION
def run(data_files, packages, install_requires):
setup(name=NAME,
version=VERSION,
license='BSD3',
description='Toolsuite for dealing with the MaNGA dataset',
long_description=open('README.rst').read(),
author='The Marvin Developers',
author_email='[email protected]',
keywords='marvin manga astronomy MaNGA',
url='https://github.com/sdss/marvin',
packages=packages,
package_dir={'': 'python'},
package_data={'': data_files},
include_package_data=True,
install_requires=install_requires,
scripts=['bin/run_marvin', 'bin/check_marvin'],
classifiers=[
'Development Status :: 4 - Beta',
'Environment :: Web Environment',
'Framework :: Flask',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: BSD License',
'Natural Language :: English',
'Operating System :: OS Independent',
'Programming Language :: Python',
'Programming Language :: Python :: 2.6',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3.3',
'Programming Language :: Python :: 3.4',
'Topic :: Database :: Front-Ends',
'Topic :: Documentation :: Sphinx',
'Topic :: Internet :: WWW/HTTP :: Dynamic Content',
'Topic :: Scientific/Engineering :: Astronomy',
'Topic :: Software Development :: Libraries :: Python Modules',
'Topic :: Software Development :: User Interfaces'
],
)
if __name__ == '__main__':
# Custom parser to decide whether we include or not the web. By default we do.
parser = argparse.ArgumentParser(prog=os.path.basename(sys.argv[0]))
parser.add_argument('-w', '--noweb', dest='noweb', default=False, action='store_true',
help='Does not build the web.')
parser.add_argument('-d', '--dev', dest='dev', default=False, action='store_true',
help='Install all packages for development')
parser.add_argument('-b', '--web', dest='web', default=False, action='store_true',
help='Install only core + web packages')
# We use parse_known_args because we want to leave the remaining args for distutils
args = parser.parse_known_args()[0]
if args.noweb:
packages = find_packages(where='python', exclude=['marvin.web*'])
else:
packages = find_packages(where='python')
data_files = get_data_files(with_web=not args.noweb)
maskbits_path = download_file('https://svn.sdss.org/public/repo/sdss/idlutils/'
'trunk/data/sdss/sdssMaskbits.par')
shutil.copy(maskbits_path, os.path.join(os.path.dirname(__file__),
'python/marvin/data/',
'sdssMaskbits.par'))
# Get the proper requirements file
install_requires = get_requirements(args)
# Now we remove all our custom arguments to make sure they don't interfere with distutils
remove_args(parser)
# Runs distutils
run(data_files, packages, install_requires)