Repository navigation
Expand file tree
/
Copy pathsetup.py
More file actions
118 lines (98 loc) · 3.21 KB
/
Copy pathsetup.py
File metadata and controls
118 lines (98 loc) · 3.21 KB
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
"""Setup.py for Robinhood helper library"""
import sys
from os import path, listdir
from setuptools import setup, find_packages
from setuptools.command.test import test as TestCommand
HERE = path.abspath(path.dirname(__file__))
__version__ = '1.0.1'
REQUIRES = [
'requests>=2.13.0,<=2.18.4',
'six>=1.10.0,<=1.11.0',
]
TEST = [
'pytest~=3.0.0',
'pytest_cov~=2.4.0',
'flaky~=3.3.0'
]
def get_requirements(test_or_prod=False):
"""cover py2/3 stdlib extensions
Args:
test_or_prod (bool): test=true, prod=false
Returns:
(:obj:`list`): list of requirements as per py2/3
"""
requires_list = []
if test_or_prod: #TEST
requires_list = TEST
if sys.version_info.major < 3:
requires_list.append('configparser~=3.5.0')
else:
requires_list = REQUIRES
if sys.version_info.major < 3:
requires_list.append('enum~=0.4.6')
return requires_list
def include_all_subfiles(*args):
"""Slurps up all files in a directory (non recursive) for data_files section
Note:
Not recursive, only includes flat files
Returns:
(:obj:`list` :obj:`str`) list of all non-directories in a file
"""
file_list = []
for path_included in args:
local_path = path.join(HERE, path_included)
for file in listdir(local_path):
file_abspath = path.join(local_path, file)
if path.isdir(file_abspath): #do not include sub folders
continue
if '_local.cfg' in file_abspath: #do not include secret files
continue
file_list.append(path_included + '/' + file)
return file_list
class PyTest(TestCommand):
"""PyTest cmdclass hook for test-at-buildtime functionality
http://doc.pytest.org/en/latest/goodpractices.html#manual-integration
"""
user_options = [('pytest-args=', 'a', "Arguments to pass to pytest")]
def initialize_options(self):
TestCommand.initialize_options(self)
self.pytest_args = [
'Tests',
'--cov=Robinhood/',
'--cov-report=term-missing'
] #load defaults here
def run_tests(self):
import shlex
#import here, cause outside the eggs aren't loaded
import pytest
pytest_commands = []
try: #read commandline
pytest_commands = shlex.split(self.pytest_args)
except AttributeError: #use defaults
pytest_commands = self.pytest_args
errno = pytest.main(pytest_commands)
exit(errno)
setup(
name='Robinhood',
author='Jamone Kelly',
author_email='TODO',
url='https://github.com/Jamonek/Robinhood',
download_url='https://github.com/Jamonek/Robinhood/tarball/v' + __version__,
version=__version__,
license='MIT',
classifiers=[
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 2.7'
],
keywords='Robinhood trade API',
packages=find_packages(),
data_files=[
('docs', include_all_subfiles('docs')),
('tests', include_all_subfiles('tests'))
],
install_requires=get_requirements(),
tests_require=get_requirements(True),
cmdclass={
'test':PyTest
}
)