Ensure setup warnings are only printed if building

Author: molinav

packages/basemap/setup.py

@@ -51,15 +51,19 @@ def get_geos_install_prefix():
         if os.path.isfile(hfile) and libfiles:
             return prefix
 
-    warnings.warn(" ".join([
-        "Cannot find GEOS library and/or headers in standard locations",
-        "('{0}'). Please install the corresponding packages using your",
-        "software management system or set the environment variable",
-        "GEOS_DIR to point to the location where GEOS is installed",
-        "(for example, if 'geos_c.h' is in '/usr/local/include'",
-        "and 'libgeos_c' is in '/usr/local/lib', then you need to",
-        "set GEOS_DIR to '/usr/local'",
-    ]).format("', '".join(candidates)), RuntimeWarning)
+    # At this point, the GEOS library was not found, so we throw a warning if
+    # the user is trying to build the library.
+    build_cmds = ("bdist_wheel", "build", "install")
+    if any(cmd in sys.argv[1:] for cmd in build_cmds):
+        warnings.warn(" ".join([
+            "Cannot find GEOS library and/or headers in standard locations",
+            "('{0}'). Please install the corresponding packages using your",
+            "software management system or set the environment variable",
+            "GEOS_DIR to point to the location where GEOS is installed",
+            "(for example, if 'geos_c.h' is in '/usr/local/include'",
+            "and 'libgeos_c' is in '/usr/local/lib', then you need to",
+            "set GEOS_DIR to '/usr/local'",
+        ]).format("', '".join(candidates)), RuntimeWarning)
     return None
 
 
@@ -96,7 +100,9 @@ def run(self):
         import numpy
         include_dirs.append(numpy.get_include())
     except ImportError as err:
-        warnings.warn("unable to locate NumPy headers", RuntimeWarning)
+        build_cmds = ("bdist_wheel", "build", "install")
+        if any(cmd in sys.argv[1:] for cmd in build_cmds):
+            warnings.warn("unable to locate NumPy headers", RuntimeWarning)
 
 # Define GEOS include, library and runtime dirs.
 geos_install_prefix = get_geos_install_prefix()