forked from kwikteam/phy-contrib
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
94 lines (76 loc) · 2.89 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
# -*- coding: utf-8 -*-
# flake8: noqa
"""Installation script."""
#------------------------------------------------------------------------------
# Imports
#------------------------------------------------------------------------------
import os
import os.path as op
import re
from textwrap import dedent
from setuptools import setup
#------------------------------------------------------------------------------
# Setup
#------------------------------------------------------------------------------
def _package_tree(pkgroot):
path = op.dirname(__file__)
subdirs = [op.relpath(i[0], path).replace(op.sep, '.')
for i in os.walk(op.join(path, pkgroot))
if '__init__.py' in i[2]]
return subdirs
curdir = op.dirname(op.realpath(__file__))
readme = open(op.join(curdir, 'README.md')).read()
# Find version number from `__init__.py` without executing it.
filename = op.join(curdir, 'phycontrib/__init__.py')
with open(filename, 'r') as f:
version = re.search(r"__version__ = '([^']+)'", f.read()).group(1)
setup(
name='phycontrib',
version=version,
license="BSD",
description='Spike sorting and ephys data analysis '
'for 1000 channels and beyond',
long_description=readme,
author='Kwik Team',
author_email='cyrille.rossant at gmail.com',
url='https://phy.cortexlab.net',
packages=_package_tree('phycontrib'),
package_dir={'phycontrib': 'phycontrib'},
package_data={
'phycontrib': ['*.npy', '*.gz', '*.txt', '*.json', '*.prb'],
},
include_package_data=True,
keywords='phy,data analysis,electrophysiology,neuroscience',
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: Developers',
'License :: OSI Approved :: BSD License',
'Natural Language :: English',
"Framework :: IPython",
"Programming Language :: Python :: 2",
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
],
)
def _create_loader_file():
"""Create a Python script that imports phycontrib in the
plugins directory. This ensures that phycontrib's plugins are always
loaded by phy."""
# Make sure the plugins directory exists.
home = op.realpath(op.expanduser('~'))
plugins_dir = op.join(home, '.phy', 'plugins')
if not op.exists(plugins_dir):
os.makedirs(plugins_dir)
# Create the script if it doesn't already exist.
path = op.join(plugins_dir, 'phycontrib_loader.py')
if op.exists(path):
return
with open(path, 'w') as f:
f.write(dedent("""
# Automatically generated by phycontrib's installer.
# This imports all of phycontrib's plugins when loading phy.
import phycontrib
""").lstrip())
# Create ~/.phy/plugins/phycontrib_loader.py.
_create_loader_file()