shelltastic 0.8.0__tar.gz → 0.8.1__tar.gz

{shelltastic-0.8.0 → shelltastic-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shelltastic
-Version: 0.8.0
+Version: 0.8.1
 Summary: A fantastic shell command runner for python
 Author: Bearmine
 License-Expression: MPL-2.0

{shelltastic-0.8.0 → shelltastic-0.8.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "shelltastic"
-version = "0.8.0"
+version = "0.8.1"
 description = "A fantastic shell command runner for python"
 readme = "README.md"
 authors = [

{shelltastic-0.8.0 → shelltastic-0.8.1}/src/shelltastic/backend/__init__.py

@@ -4,6 +4,7 @@ from shelltastic.backend.base import (
     RemoteShellBackend,
     ShellBackend,
     add_shim,
+    clear_shims,
 )
 from shelltastic.backend.default.local import DefaultLocalBackend
 from shelltastic.backend.default.remote import DefaultRemoteBackend
@@ -17,6 +18,7 @@ __all__ = [
     "register_default_local_backend",
     "register_default_remote_backend",
     "add_shim",
+    "clear_shims",
     "get_local_backend",
     "get_remote_backend",
 ]

{shelltastic-0.8.0 → shelltastic-0.8.1}/src/shelltastic/backend/base.py

@@ -90,6 +90,10 @@ class BackendShim(ABC):
 
 _backend_shims: list[BackendShim] = []
 
+def clear_shims():
+    """Clear all shims"""
+    global _backend_shims
+    _backend_shims.clear()
 
 def add_shim(shim: BackendShim):
     """Add a shim to manipulate `ShellBackend` params before any `ShellBackend` runs"""
@@ -103,7 +107,19 @@ def _shim_run(f, backend_type: type):
         result: BackendShimResult = BackendShimResult(*args, **kwargs)
         for shim in _backend_shims:
             result = shim.run_shim(
-                backend_type, f.__name__, *result.args, **result.kwargs
+                backend_type, *result.args, **result.kwargs
+            )
+        return f(*result.args, **result.kwargs)
+
+    return wrapper
+
+def _shim_run_async(f, backend_type: type):
+    @wraps(f)
+    def wrapper(*args, **kwargs):
+        result: BackendShimResult = BackendShimResult(*args, **kwargs)
+        for shim in _backend_shims:
+            result = shim.run_async_shim(
+                backend_type, *result.args, **result.kwargs
             )
         return f(*result.args, **result.kwargs)
 
@@ -114,7 +130,7 @@ class _InstallShims(ABCMeta):
     def __call__(self, *args: Any, **kwds: Any) -> Any:
         instance = super().__call__(*args, **kwds)
         instance.run = _shim_run(instance.run, instance.__class__)
-        instance.run_async = _shim_run(instance.run_async, instance.__class__)
+        instance.run_async = _shim_run_async(instance.run_async, instance.__class__)
         return instance