1
"""distutils.command.build_scripts
3
Implements the Distutils 'build_scripts' command."""
6
from stat import ST_MODE
7
from distutils import sysconfig
8
from distutils.core import Command
9
from distutils.dep_util import newer
10
from distutils.util import convert_path, Mixin2to3
11
from distutils import log
14
# check if Python is called on the first line with this expression
15
first_line_re = re.compile(b'^#!.*python[0-9.]*([ \t].*)?$')
17
class build_scripts(Command):
19
description = "\"build\" scripts (copy and fixup #! line)"
22
('build-dir=', 'd', "directory to \"build\" (copy) to"),
23
('force', 'f', "forcibly build everything (ignore file timestamps"),
24
('executable=', 'e', "specify final destination interpreter path"),
27
boolean_options = ['force']
30
def initialize_options(self):
34
self.executable = None
37
def finalize_options(self):
38
self.set_undefined_options('build',
39
('build_scripts', 'build_dir'),
41
('executable', 'executable'))
42
self.scripts = self.distribution.scripts
44
def get_source_files(self):
53
def copy_scripts(self):
54
"""Copy each script listed in 'self.scripts'; if it's marked as a
55
Python script in the Unix way (first line matches 'first_line_re',
56
ie. starts with "\#!" and contains "python"), then adjust the first
57
line to refer to the current Python interpreter as we copy.
59
self.mkpath(self.build_dir)
62
for script in self.scripts:
64
script = convert_path(script)
65
outfile = os.path.join(self.build_dir, os.path.basename(script))
66
outfiles.append(outfile)
68
if not self.force and not newer(script, outfile):
69
log.debug("not copying %s (up-to-date)", script)
72
# Always open the file, but ignore failures in dry-run mode --
73
# that way, we'll get accurate feedback if we can read the
76
f = open(script, "rb")
82
encoding, lines = tokenize.detect_encoding(f.readline)
84
first_line = f.readline()
86
self.warn("%s is an empty file (skipping)" % script)
89
match = first_line_re.match(first_line)
92
post_interp = match.group(1) or b''
95
log.info("copying and adjusting %s -> %s", script,
97
updated_files.append(outfile)
99
if not sysconfig.python_build:
100
executable = self.executable
102
executable = os.path.join(
103
sysconfig.get_config_var("BINDIR"),
104
"python%s%s" % (sysconfig.get_config_var("VERSION"),
105
sysconfig.get_config_var("EXE")))
106
executable = os.fsencode(executable)
107
shebang = b"#!" + executable + post_interp + b"\n"
108
# Python parser starts to read a script using UTF-8 until
109
# it gets a #coding:xxx cookie. The shebang has to be the
110
# first line of a file, the #coding:xxx cookie cannot be
111
# written before. So the shebang has to be decodable from
114
shebang.decode('utf-8')
115
except UnicodeDecodeError:
117
"The shebang ({!r}) is not decodable "
118
"from utf-8".format(shebang))
119
# If the script is encoded to a custom encoding (use a
120
# #coding:xxx cookie), the shebang has to be decodable from
121
# the script encoding too.
123
shebang.decode(encoding)
124
except UnicodeDecodeError:
126
"The shebang ({!r}) is not decodable "
127
"from the script encoding ({})"
128
.format(shebang, encoding))
129
with open(outfile, "wb") as outf:
131
outf.writelines(f.readlines())
137
updated_files.append(outfile)
138
self.copy_file(script, outfile)
140
if os.name == 'posix':
141
for file in outfiles:
143
log.info("changing mode of %s", file)
145
oldmode = os.stat(file)[ST_MODE] & 0o7777
146
newmode = (oldmode | 0o555) & 0o7777
147
if newmode != oldmode:
148
log.info("changing mode of %s from %o to %o",
149
file, oldmode, newmode)
150
os.chmod(file, newmode)
151
# XXX should we modify self.outfiles?
152
return outfiles, updated_files
154
class build_scripts_2to3(build_scripts, Mixin2to3):
156
def copy_scripts(self):
157
outfiles, updated_files = build_scripts.copy_scripts(self)
159
self.run_2to3(updated_files)
160
return outfiles, updated_files