bdist_rpm.py 775 B

123456789101112131415161718192021222324
  1. from __future__ import division, absolute_import, print_function
  2. import os
  3. import sys
  4. if 'setuptools' in sys.modules:
  5. from setuptools.command.bdist_rpm import bdist_rpm as old_bdist_rpm
  6. else:
  7. from distutils.command.bdist_rpm import bdist_rpm as old_bdist_rpm
  8. class bdist_rpm(old_bdist_rpm):
  9. def _make_spec_file(self):
  10. spec_file = old_bdist_rpm._make_spec_file(self)
  11. # Replace hardcoded setup.py script name
  12. # with the real setup script name.
  13. setup_py = os.path.basename(sys.argv[0])
  14. if setup_py == 'setup.py':
  15. return spec_file
  16. new_spec_file = []
  17. for line in spec_file:
  18. line = line.replace('setup.py', setup_py)
  19. new_spec_file.append(line)
  20. return new_spec_file