Chris@87: from __future__ import division, absolute_import, print_function
Chris@87: 
Chris@87: import sys
Chris@87: if 'setuptools' in sys.modules:
Chris@87:     from setuptools.command.sdist import sdist as old_sdist
Chris@87: else:
Chris@87:     from distutils.command.sdist import sdist as old_sdist
Chris@87: 
Chris@87: from numpy.distutils.misc_util import get_data_files
Chris@87: 
Chris@87: class sdist(old_sdist):
Chris@87: 
Chris@87:     def add_defaults (self):
Chris@87:         old_sdist.add_defaults(self)
Chris@87: 
Chris@87:         dist = self.distribution
Chris@87: 
Chris@87:         if dist.has_data_files():
Chris@87:             for data in dist.data_files:
Chris@87:                 self.filelist.extend(get_data_files(data))
Chris@87: 
Chris@87:         if dist.has_headers():
Chris@87:             headers = []
Chris@87:             for h in dist.headers:
Chris@87:                 if isinstance(h, str): headers.append(h)
Chris@87:                 else: headers.append(h[1])
Chris@87:             self.filelist.extend(headers)
Chris@87: 
Chris@87:         return