feat: allow plugin.register to be used as a decorator

pwwang · Jul 21, 2024 · 1266e14 · 1266e14
1 parent 135f1ba
commit 1266e14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/simplug.py b/simplug.py
@@ -831,6 +831,10 @@ def register(self, *plugins: Any) -> None:
 
         self._batch_index += 1
 
+        if len(plugins) == 1 and callable(plugins[0]):
+            # allow to use as a decorator
+            return plugins[0]
+
     def get_plugin(self, name: str, raw: bool = False) -> object:
         """Get the plugin wrapper or the raw plugin object