FIX(pvlib,setup,PY2): monkeypatching type(dist) fails, because...

PY2 type(dist) is ``<type 'instance'>``, and does not have the method to patch.
ankostis · Jun 4, 2018 · 1f72bae · 1f72bae
1 parent 4a5b90b
commit 1f72bae
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pvlib/polyversion/setuplugin.py b/pvlib/polyversion/setuplugin.py
@@ -88,10 +88,13 @@ def _establish_setup_py_version(dist, repo_path=None, **pvargs):
         #  NOTE: We monekypatch even if user has disabled check,
         #  bc we can't now the kw order.
         #
-        orig_func = type(dist).run_command
+        ## NOTE: PY2 `type(dist)` is `<type 'instance'>` on PY2,
+        #  which does not have the method to patch.
+        DistClass = dist.__class__
+        orig_func = DistClass.run_command
         if orig_func is not _monkeypathed_run_command:
-            type(dist).run_command = _monkeypathed_run_command
-            type(dist)._polyversion_orig_run_cmd = orig_func
+            DistClass.run_command = _monkeypathed_run_command
+            DistClass._polyversion_orig_run_cmd = orig_func
 
     if version:
         dist.metadata.version = version