mirror of https://github.com/python/cpython.git
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
"""distutils.command.build_scripts
|
|
|
|
Implements the Distutils 'build_scripts' command."""
|
|
|
|
# created 2000/05/23, Bastian Kleineidam
|
|
|
|
__revision__ = "$Id$"
|
|
|
|
import sys,os,re
|
|
from distutils.core import Command
|
|
|
|
# check if Python is called on the first line with this expression
|
|
first_line_re = re.compile(r"^#!.+python(\s-\w+)*")
|
|
|
|
class build_scripts (Command):
|
|
|
|
description = "\"build\" scripts"
|
|
|
|
user_options = [
|
|
('build-dir=', 'd', "directory to \"build\" (copy) to"),
|
|
('force', 'f', "forcibly build everything (ignore file timestamps"),
|
|
]
|
|
|
|
|
|
def initialize_options (self):
|
|
self.build_dir = None
|
|
self.scripts = None
|
|
self.force = None
|
|
self.outfiles = None
|
|
|
|
def finalize_options (self):
|
|
self.set_undefined_options ('build',
|
|
('build_scripts', 'build_dir'),
|
|
('force', 'force'))
|
|
self.scripts = self.distribution.scripts
|
|
|
|
|
|
def run (self):
|
|
if not self.scripts:
|
|
return
|
|
self._copy_files()
|
|
self._adjust_files()
|
|
|
|
def _copy_files(self):
|
|
"""Copy all the scripts to the build dir"""
|
|
self.outfiles = []
|
|
self.mkpath(self.build_dir)
|
|
for f in self.scripts:
|
|
print self.build_dir
|
|
if self.copy_file(f, self.build_dir):
|
|
self.outfiles.append(os.path.join(self.build_dir, f))
|
|
|
|
def _adjust_files(self):
|
|
"""If the first line begins with #! and ends with python
|
|
replace it with the current python interpreter"""
|
|
for f in self.outfiles:
|
|
if not self.dry_run:
|
|
data = open(f, "r").readlines()
|
|
if not data:
|
|
self.warn("%s is an empty file!" % f)
|
|
continue
|
|
mo = first_line_re.match(data[0])
|
|
if mo:
|
|
self.announce("Adjusting first line of file %s" % f)
|
|
data[0] = "#!"+sys.executable
|
|
# add optional command line options
|
|
if mo.group(1):
|
|
data[0] = data[0] + mo.group(1)
|
|
else:
|
|
data[0] = data[0] + "\n"
|
|
open(f, "w").writelines(data)
|