Chris@87: from __future__ import division, absolute_import, print_function
Chris@87: 
Chris@87: import os
Chris@87: import sys
Chris@87: if 'setuptools' in sys.modules:
Chris@87:     from setuptools.command.bdist_rpm import bdist_rpm as old_bdist_rpm
Chris@87: else:
Chris@87:     from distutils.command.bdist_rpm import bdist_rpm as old_bdist_rpm
Chris@87: 
Chris@87: class bdist_rpm(old_bdist_rpm):
Chris@87: 
Chris@87:     def _make_spec_file(self):
Chris@87:         spec_file = old_bdist_rpm._make_spec_file(self)
Chris@87: 
Chris@87:         # Replace hardcoded setup.py script name
Chris@87:         # with the real setup script name.
Chris@87:         setup_py = os.path.basename(sys.argv[0])
Chris@87:         if setup_py == 'setup.py':
Chris@87:             return spec_file
Chris@87:         new_spec_file = []
Chris@87:         for line in spec_file:
Chris@87:             line = line.replace('setup.py', setup_py)
Chris@87:             new_spec_file.append(line)
Chris@87:         return new_spec_file