-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
123 lines (102 loc) · 3.21 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
from setuptools import setup, find_packages
import distutils.cmd
import os
import shutil
import subprocess
import sys
class RunCommands(distutils.cmd.Command):
description = ''
user_options = []
commands = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
print('Running commands. This could take a while.')
for command in self.commands:
print(command)
try:
subprocess.check_call(command, shell=True)
except subprocess.CalledProcessError as error:
print('Command failed. Aborting.')
print(error.output)
sys.exit(error.returncode)
return
class CosmicRay(RunCommands):
description = 'Run cosmic-ray to find bad unit tests.'
commands = [
r'''python -c "exec('try:\n import typed_ast\n x=1\nexcept:\n x=0'); import sys; sys.stderr.write('typed_ast cannot be installed with cosmic_ray\n' if x == 1 else ''); sys.exit(x)"''',
'cosmic-ray init cosmic_ray_cfg.yml session',
'cosmic-ray exec session',
'cosmic-ray dump session | cr-report',
]
class Coverage(RunCommands):
description = 'Determine unittest coverage.'
commands = [
'coverage run --branch --source=traitlite -m unittest discover',
'coverage report',
]
class MyPy(RunCommands):
description = 'Run mypy on the package.'
commands = [
'mypy traitlite/',
]
class Clean(distutils.cmd.Command):
description = 'Clean up the whole package.'
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
to_remove = [
# Files.
'session.json',
'.coverage',
'.DS_Store',
# Directories.
'.hypothesis',
'.mypy_cache',
'build',
'dist',
'docs/_build',
'tests/__pycache__',
'traitlite/__pycache__',
'traitlite.egg-info',
]
for remove in to_remove:
remove = os.path.join(*remove.split('/'))
if os.path.isdir(remove):
shutil.rmtree(remove, ignore_errors=True)
else:
try:
os.remove(remove)
except OSError as error:
if error.errno != 2:
raise
with open('README.md', 'r') as fh:
long_description = fh.read()
setup(
name='traitlite',
version='0.0.3',
author='Christopher Pezley',
author_email='pypi@pezley.net',
description='A light-weight module that aims to provide control '
'for class attributes.',
long_description=long_description,
long_description_content_type='text/markdown',
url='https://github.com/lejar/traitlite',
packages=find_packages(exclude=['tests']),
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
cmdclass={
'crtest': CosmicRay,
'coverage': Coverage,
'mypy': MyPy,
'clean': Clean,
},
)