-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
143 lines (119 loc) · 4.19 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
# ptufile/setup.py
"""Ptufile package Setuptools script."""
import os
import re
import sys
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext as _build_ext
DEBUG = bool(os.environ.get('PTUFILE_DEBUG', False))
def search(pattern, string, flags=0):
"""Return first match of pattern in string."""
match = re.search(pattern, string, flags)
if match is None:
raise ValueError(f'{pattern!r} not found')
return match.groups()[0]
def fix_docstring_examples(docstring):
"""Return docstring with examples fixed for GitHub."""
start = True
indent = False
lines = ['..', ' This file is generated by setup.py', '']
for line in docstring.splitlines():
if not line.strip():
start = True
indent = False
if line.startswith('>>> '):
indent = True
if start:
lines.extend(['.. code-block:: python', ''])
start = False
lines.append((' ' if indent else '') + line)
return '\n'.join(lines)
with open('ptufile/ptufile.py', encoding='utf-8') as fh:
code = fh.read()
version = search(r"__version__ = '(.*?)'", code).replace('.x.x', '.dev0')
description = search(r'"""(.*)\.(?:\r\n|\r|\n)', code)
readme = search(
r'(?:\r\n|\r|\n){2}"""(.*)"""(?:\r\n|\r|\n){2}from __future__',
code,
re.MULTILINE | re.DOTALL,
)
readme = '\n'.join(
[description, '=' * len(description)] + readme.splitlines()[1:]
)
if 'sdist' in sys.argv:
# update README, LICENSE, and CHANGES files
with open('README.rst', 'w', encoding='utf-8') as fh:
fh.write(fix_docstring_examples(readme))
license = search(
r'(# Copyright.*?(?:\r\n|\r|\n))(?:\r\n|\r|\n)+""',
code,
re.MULTILINE | re.DOTALL,
)
license = license.replace('# ', '').replace('#', '')
with open('LICENSE', 'w', encoding='utf-8') as fh:
fh.write('BSD 3-Clause License\n\n')
fh.write(license)
class build_ext(_build_ext):
"""Delay import numpy until build."""
def finalize_options(self):
_build_ext.finalize_options(self)
if isinstance(__builtins__, dict):
__builtins__['__NUMPY_SETUP__'] = False
else:
setattr(__builtins__, '__NUMPY_SETUP__', False)
import numpy
self.include_dirs.append(numpy.get_include())
ext_modules = [
Extension(
'ptufile._ptufile',
['ptufile/_ptufile.pyx'],
define_macros=[
# ('CYTHON_TRACE_NOGIL', '1'),
# ('CYTHON_LIMITED_API', '1'),
# ('Py_LIMITED_API', '1'),
('NPY_NO_DEPRECATED_API', 'NPY_1_7_API_VERSION'),
],
extra_compile_args=['/Zi', '/Od'] if DEBUG else [],
extra_link_args=['-debug:full'] if DEBUG else [],
)
]
setup(
name='ptufile',
version=version,
license='BSD',
description=description,
long_description=readme,
long_description_content_type='text/x-rst',
author='Christoph Gohlke',
author_email='cgohlke@cgohlke.com',
url='https://www.cgohlke.com',
project_urls={
'Bug Tracker': 'https://github.com/cgohlke/ptufile/issues',
'Source Code': 'https://github.com/cgohlke/ptufile',
# 'Documentation': 'https://',
},
packages=['ptufile'],
package_data={'ptufile': ['py.typed']},
entry_points={'console_scripts': ['ptufile = ptufile.__main__:main']},
python_requires='>=3.9',
install_requires=['numpy'],
setup_requires=['setuptools', 'numpy'],
extras_require={'all': ['xarray', 'tifffile', 'matplotlib']},
ext_modules=ext_modules,
cmdclass={'build_ext': build_ext},
zip_safe=False,
platforms=['any'],
classifiers=[
'Development Status :: 4 - Beta',
'License :: OSI Approved :: BSD License',
'Intended Audience :: Science/Research',
'Intended Audience :: Developers',
'Operating System :: OS Independent',
'Programming Language :: Cython',
'Programming Language :: Python :: 3 :: Only',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: 3.12',
'Programming Language :: Python :: 3.13',
],
)