spaCy/setup.py

79 lines
2.2 KiB
Python
Raw Normal View History

2014-07-05 22:49:34 +04:00
#!/usr/bin/env python
import Cython.Distutils
from Cython.Distutils import Extension
2014-07-05 22:49:34 +04:00
import distutils.core
import sys
import os
import os.path
2014-10-22 03:17:26 +04:00
2014-07-05 22:49:34 +04:00
from os import path
from glob import glob
2014-07-05 22:49:34 +04:00
def clean(ext):
for pyx in ext.sources:
if pyx.endswith('.pyx'):
c = pyx[:-4] + '.c'
cpp = pyx[:-4] + '.cpp'
so = pyx[:-4] + '.so'
html = pyx[:-4] + '.html'
if os.path.exists(so):
os.unlink(so)
if os.path.exists(c):
os.unlink(c)
elif os.path.exists(cpp):
os.unlink(cpp)
if os.path.exists(html):
os.unlink(html)
HERE = os.path.dirname(__file__)
virtual_env = os.environ.get('VIRTUAL_ENV', '')
compile_args = []
link_args = []
libs = []
includes = ['.']
cython_includes = ['.']
2014-07-05 22:49:34 +04:00
if 'VIRTUAL_ENV' in os.environ:
includes += glob(path.join(os.environ['VIRTUAL_ENV'], 'include', 'site', '*'))
else:
# If you're not using virtualenv, set your include dir here.
pass
2014-07-05 22:49:34 +04:00
exts = [
2014-09-02 01:41:59 +04:00
Extension("spacy.lang", ["spacy/lang.pyx"], language="c++", include_dirs=includes),
2014-10-22 03:17:26 +04:00
Extension("spacy.lexeme", ["spacy/lexeme.pyx"], language="c++", include_dirs=includes),
Extension("spacy.en", ["spacy/en.pyx"], language="c++", include_dirs=includes),
Extension("spacy.tokens", ["spacy/tokens.pyx"], language="c++", include_dirs=includes),
2014-11-05 12:45:44 +03:00
Extension("spacy.tagger", ["spacy/tagger.pyx"], language="c++", include_dirs=includes),
Extension("spacy.context", ["spacy/context.pyx"], language="c++", include_dirs=includes),
Extension("spacy.utf8string", ["spacy/utf8string.pyx"], language="c++", include_dirs=includes),
2014-11-05 12:45:44 +03:00
Extension("spacy.pos_feats", ["spacy/pos_feats.pyx"], language="c++", include_dirs=includes),
Extension("spacy.ner_feats", ["spacy/ner_feats.pyx"], language="c++", include_dirs=includes)
2014-07-05 22:49:34 +04:00
]
if sys.argv[1] == 'clean':
print >> sys.stderr, "cleaning .c, .c++ and .so files matching sources"
map(clean, exts)
distutils.core.setup(
2014-10-22 03:17:26 +04:00
name='spacy',
2014-07-31 05:03:10 +04:00
packages=['spacy'],
2014-07-05 22:49:34 +04:00
author='Matthew Honnibal',
author_email='honnibal@gmail.com',
version='1.0',
2014-10-22 03:17:26 +04:00
package_data={"spacy": ["*.pxd"]},
2014-07-05 22:49:34 +04:00
cmdclass={'build_ext': Cython.Distutils.build_ext},
2014-10-22 03:17:26 +04:00
2014-07-05 22:49:34 +04:00
ext_modules=exts,
)