Make binaries movable in WinPython3.x

AlexanderYAPPO · AlexanderYAPPO · commit fe1b790e91d2 · 2016-05-24T17:37:12.000+04:00
diff --git a/winpython/utils.py b/winpython/utils.py
@@ -312,6 +312,29 @@ def patch_shebang_line(fname, pad=b' ', to_movable=True, targetdir=""):
         print("failed to patch", fname)
 
 
+# =============================================================================
+# Patch shebang line in .py files
+# =============================================================================
+def patch_shebang_line_py(fname, to_movable=True, targetdir=""):
+    """Changes shebang line in '.py' file to relative or absolue path"""
+    import fileinput
+    import re
+    import sys
+    
+    if sys.version_info[0] == 2:
+        # Python 2.x doesn't create .py files for .exe files. So, Moving
+        # WinPython doesn't break running executable files.
+        return
+    if to_movable:
+        exec_path = '#!.\python.exe'
+    else:
+        exec_path = '#!' + sys.executable
+    for line in fileinput.input(fname, inplace=True):
+        if re.match('^#\!.*python\.exe$', line) is not None:
+            print(exec_path)
+        else:
+            print(line, end='')
+
 # =============================================================================
 # Patch sourcefile (instead of forking packages)
 # =============================================================================
diff --git a/winpython/wppm.py b/winpython/wppm.py
@@ -393,6 +393,9 @@ def patch_all_shebang(self, to_movable=True, max_exe_size=999999, targetdir=""):
             if size <= max_exe_size:
                 utils.patch_shebang_line(ffname, to_movable=to_movable,
                                          targetdir=targetdir)
+        for ffname in glob.glob(r'%s\Scripts\*.py' % self.target):
+            utils.patch_shebang_line_py(ffname, to_movable=to_movable,
+                                        targetdir=targetdir)
 
 
     def install(self, package, install_options=None):
