Add monkey module as workaround for fabric/fabric#2090

jaraco · Dec 5, 2022 · b2db0f3 · b2db0f3
1 parent dc0e2f8
commit b2db0f3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,3 +1,8 @@
+v5.3.0
+======
+
+Added ``monkey`` module.
+
 v5.2.1
 ======
 

diff --git a/jaraco/fabric/monkey.py b/jaraco/fabric/monkey.py
@@ -0,0 +1,16 @@
+import functools
+
+
+def workaround_2090(func):
+    """
+    Wrap a task func to work around fabric/fabric#2090.
+    """
+
+    @functools.wraps(func)
+    def wrapper(c, *args, **kwargs):
+        c.connect_kwargs['key_filename'][:] = [
+            key
+            for key in c.connect_kwargs['key_filename']
+            if __import__('os').path.exists(key)
+        ]
+        return func(c, *args, **kwargs)