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