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