orion / rpms / python3x-pip

Forked from rpms/python3x-pip 2 years ago
Clone
Blob Blame History Raw
From aab24967a03bda3b0999d80562a6064c27d1e0e0 Mon Sep 17 00:00:00 2001
From: Tomas Orsava <torsava@redhat.com>
Date: Tue, 12 Nov 2019 17:15:08 +0100
Subject: [PATCH] Downstream only patch

Emit a warning to the user if pip install is run with root privileges
Issue upstream: https://github.com/pypa/pip/issues/4288
---
 src/pip/_internal/commands/install.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/pip/_internal/commands/install.py b/src/pip/_internal/commands/install.py
index 5842d18..a6104b4 100644
--- a/src/pip/_internal/commands/install.py
+++ b/src/pip/_internal/commands/install.py
@@ -12,6 +12,8 @@ import logging
 import operator
 import os
 import shutil
+import sys
+from os import path
 from optparse import SUPPRESS_HELP
 
 from pip._vendor import pkg_resources
@@ -281,6 +283,23 @@ class InstallCommand(RequirementCommand):
     def run(self, options, args):
         # type: (Values, List[Any]) -> int
         cmdoptions.check_install_build_global(options)
+
+        def is_venv():
+            return (hasattr(sys, 'real_prefix') or
+                    (hasattr(sys, 'base_prefix') and
+                     sys.base_prefix != sys.prefix))
+
+        # Check whether we have root privileges and aren't in venv/virtualenv
+        if os.getuid() == 0 and not is_venv():
+            command = path.basename(sys.argv[0])
+            if command == "__main__.py":
+                command = path.basename(sys.executable) + " -m pip"
+            logger.warning(
+                "Running pip install with root privileges is "
+                "generally not a good idea. Try `%s install --user` instead."
+                % command
+            )
+
         upgrade_strategy = "to-satisfy-only"
         if options.upgrade:
             upgrade_strategy = options.upgrade_strategy
-- 
2.20.1