Bastian Kleineidam: the "build_scripts" command.

Lib/distutils/command/build_scripts.py

+"""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)