diff --git a/setup.py b/setup.py index 82d92135b..2541c5e29 100644 --- a/setup.py +++ b/setup.py @@ -21,6 +21,7 @@ import json import logging import os +from os.path import join as pjoin import sys #============== @@ -30,7 +31,7 @@ from install.install_log import log mb_path = 'montblanc' -mb_inc_path = os.path.join(mb_path, 'include') +mb_inc_path = pjoin(mb_path, 'include') #=================== # Detect readthedocs @@ -142,10 +143,23 @@ def include_pkg_dirs(): # OK, so everything starts with 'montblanc/' # Take everything after that ('include...') and # append a '/*.*' to it - pkg_dirs.append(os.path.join(root[l:], d, '*.*')) + pkg_dirs.append(pjoin(root[l:], d, '*.*')) return pkg_dirs + +def include_pkg_dirs(): + mb_inc_path = pjoin("montblanc", "include") + l = len("montblanc") + len(os.sep) + + exclude = set(['docs', '.git', '.svn']) + + return [pjoin(path, d, '*.*')[l:] for path, dirs, files + in os.walk(mb_inc_path, topdown=True) + for d in dirs if d not in exclude] + + + install_requires = [ 'attrdict >= 2.0.0', 'attrs >= 16.3.0', @@ -196,7 +210,7 @@ def include_pkg_dirs(): from install.versioning import maintain_version setup(name='montblanc', - version=maintain_version(os.path.join('montblanc', 'version.py')), + version=maintain_version(pjoin('montblanc', 'version.py')), description='GPU-accelerated RIME implementations.', long_description=readme(), url='http://github.com/ska-sa/montblanc',