Matej Stuchlik 638c2fb
commit aefacbb76661520415a1c35028f2984e70cfe0bf
Matej Stuchlik 638c2fb
Author: Slavek Kabrda <bkabrda@redhat.com>
Matej Stuchlik 638c2fb
Date:   Fri Nov 29 13:24:58 2013 +0100
Matej Stuchlik 638c2fb
Matej Stuchlik 638c2fb
    Allow stripping given prefix from wheel RECORD files
Matej Stuchlik 638c2fb
Matej Stuchlik 638c2fb
diff --git a/pip/commands/install.py b/pip/commands/install.py
Matej Stuchlik 638c2fb
index 1693d01..0287c06 100644
Matej Stuchlik 638c2fb
--- a/pip/commands/install.py
Matej Stuchlik 638c2fb
+++ b/pip/commands/install.py
Matej Stuchlik 638c2fb
@@ -137,6 +137,14 @@ class InstallCommand(Command):
Matej Stuchlik 638c2fb
             help="Install everything relative to this alternate root directory.")
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
         cmd_opts.add_option(
Matej Stuchlik 638c2fb
+            '--strip-file-prefix',
Matej Stuchlik 638c2fb
+            dest='strip_file_prefix',
Matej Stuchlik 638c2fb
+            metavar='prefix',
Matej Stuchlik 638c2fb
+            default=None,
Matej Stuchlik 638c2fb
+            help="Strip given prefix from script paths in wheel RECORD."
Matej Stuchlik 638c2fb
+        )
Matej Stuchlik 638c2fb
+
Matej Stuchlik 638c2fb
+        cmd_opts.add_option(
Matej Stuchlik 638c2fb
             "--compile",
Matej Stuchlik 638c2fb
             action="store_true",
Matej Stuchlik 638c2fb
             dest="compile",
Matej Stuchlik 638c2fb
@@ -273,7 +281,11 @@ class InstallCommand(Command):
Matej Stuchlik 638c2fb
                 requirement_set.locate_files()
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
             if not options.no_install and not self.bundle:
Matej Stuchlik 638c2fb
-                requirement_set.install(install_options, global_options, root=options.root_path)
Matej Stuchlik 638c2fb
+                requirement_set.install(
Matej Stuchlik 638c2fb
+                    install_options,
Matej Stuchlik 638c2fb
+                    global_options,
Matej Stuchlik 638c2fb
+                    root=options.root_path,
Matej Stuchlik 638c2fb
+                    strip_file_prefix=options.strip_file_prefix)
Matej Stuchlik 638c2fb
                 installed = ' '.join([req.name for req in
Matej Stuchlik 638c2fb
                                       requirement_set.successfully_installed])
Matej Stuchlik 638c2fb
                 if installed:
Matej Stuchlik 638c2fb
diff --git a/pip/req.py b/pip/req.py
Matej Stuchlik 638c2fb
index 3ae306d..c171130 100644
Matej Stuchlik 638c2fb
--- a/pip/req.py
Matej Stuchlik 638c2fb
+++ b/pip/req.py
Matej Stuchlik 638c2fb
@@ -615,15 +615,19 @@ exec(compile(open(__file__).read().replace('\\r\\n', '\\n'), __file__, 'exec'))
Matej Stuchlik 638c2fb
         name = name.replace(os.path.sep, '/')
Matej Stuchlik 638c2fb
         return name
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
-    def install(self, install_options, global_options=(), root=None):
Matej Stuchlik 638c2fb
+    def install(self, install_options, global_options=(), root=None, strip_file_prefix=None):
Matej Stuchlik 638c2fb
         if self.editable:
Matej Stuchlik 638c2fb
             self.install_editable(install_options, global_options)
Matej Stuchlik 638c2fb
             return
Matej Stuchlik 638c2fb
         if self.is_wheel:
Matej Stuchlik 638c2fb
             version = pip.wheel.wheel_version(self.source_dir)
Matej Stuchlik 638c2fb
             pip.wheel.check_compatibility(version, self.name)
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
-            self.move_wheel_files(self.source_dir, root=root)
Matej Stuchlik 638c2fb
+            self.move_wheel_files(
Matej Stuchlik 638c2fb
+                self.source_dir,
Matej Stuchlik 638c2fb
+                root=root,
Matej Stuchlik 638c2fb
+                strip_file_prefix=strip_file_prefix
Matej Stuchlik 638c2fb
+            )
Matej Stuchlik 638c2fb
             self.install_succeeded = True
Matej Stuchlik 638c2fb
             return
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
@@ -844,13 +848,14 @@ exec(compile(open(__file__).read().replace('\\r\\n', '\\n'), __file__, 'exec'))
Matej Stuchlik 638c2fb
         self._bundle_build_dirs = bundle_build_dirs
Matej Stuchlik 638c2fb
         self._bundle_editable_dirs = bundle_editable_dirs
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
-    def move_wheel_files(self, wheeldir, root=None):
Matej Stuchlik 638c2fb
+    def move_wheel_files(self, wheeldir, root=None, strip_file_prefix=None):
Matej Stuchlik 638c2fb
         move_wheel_files(
Matej Stuchlik 638c2fb
             self.name, self.req, wheeldir,
Matej Stuchlik 638c2fb
             user=self.use_user_site,
Matej Stuchlik 638c2fb
             home=self.target_dir,
Matej Stuchlik 638c2fb
             root=root,
Matej Stuchlik 638c2fb
             pycompile=self.pycompile,
Matej Stuchlik 638c2fb
+            strip_file_prefix=strip_file_prefix,
Matej Stuchlik 638c2fb
         )
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
     @property
Matej Stuchlik 638c2fb
diff --git a/pip/wheel.py b/pip/wheel.py
Matej Stuchlik 638c2fb
index fa3e270..3a366d0 100644
Matej Stuchlik 638c2fb
--- a/pip/wheel.py
Matej Stuchlik 638c2fb
+++ b/pip/wheel.py
Matej Stuchlik 638c2fb
@@ -136,7 +136,7 @@ def get_entrypoints(filename):
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
 
Matej Stuchlik 638c2fb
 def move_wheel_files(name, req, wheeldir, user=False, home=None, root=None,
Matej Stuchlik 56b65b2
-                     pycompile=True, scheme=None):
Matej Stuchlik 56b65b2
+                     pycompile=True, scheme=None, strip_file_prefix=None):
Matej Stuchlik 638c2fb
     """Install a wheel"""
Matej Stuchlik 638c2fb
 
Matej Stuchlik 56b65b2
     if not scheme:
Matej Stuchlik 638c2fb
@@ -357,6 +357,8 @@ if __name__ == '__main__':
Matej Stuchlik 638c2fb
                 writer.writerow(row)
Matej Stuchlik 638c2fb
             for f in generated:
Matej Stuchlik 638c2fb
                 h, l = rehash(f)
Matej Stuchlik 638c2fb
+                if strip_file_prefix and f.startswith(strip_file_prefix):
Matej Stuchlik 638c2fb
+                    f = os.path.join(os.sep, os.path.relpath(f, strip_file_prefix))
Matej Stuchlik 638c2fb
                 writer.writerow((f, h, l))
Matej Stuchlik 638c2fb
             for f in installed:
Matej Stuchlik 638c2fb
                 writer.writerow((installed[f], '', ''))