sdist.py 799 B

1234567891011121314151617181920212223242526272829
  1. from __future__ import division, absolute_import, print_function
  2. import sys
  3. if 'setuptools' in sys.modules:
  4. from setuptools.command.sdist import sdist as old_sdist
  5. else:
  6. from distutils.command.sdist import sdist as old_sdist
  7. from numpy.distutils.misc_util import get_data_files
  8. class sdist(old_sdist):
  9. def add_defaults (self):
  10. old_sdist.add_defaults(self)
  11. dist = self.distribution
  12. if dist.has_data_files():
  13. for data in dist.data_files:
  14. self.filelist.extend(get_data_files(data))
  15. if dist.has_headers():
  16. headers = []
  17. for h in dist.headers:
  18. if isinstance(h, str): headers.append(h)
  19. else: headers.append(h[1])
  20. self.filelist.extend(headers)
  21. return