🧑‍💻 Change hook definitions

Update configure methods in README.md

Author: Freed-Wu

README.md

@@ -89,6 +89,17 @@ install_hook(c)
 `install_hook()` must be after the customization of the prompt string and best
 at the end of file.
 
+## Configure
+
+```python
+from repl_python_wakatime.python import install_hook
+
+install_hook(f, args, kwargs)
+```
+
+will execute `f(*args, **kwargs)` after every output/input. Other REPLs are
+similar.
+
 ## Similar projects
 
 - [wakatime plugins for python and many shells](https://wakatime.com/terminal)

src/repl_python_wakatime/__init__.py

@@ -52,11 +52,13 @@ def send_wakatime_heartbeat(
     )
 
 
-def wakatime_hook(args: tuple = (), kwargs: dict[str, Any] = {}) -> None:
+def wakatime_hook(*args: Any, **kwargs: Any) -> None:
     """Wakatime hook.
 
     :param args:
-    :type args: tuple
+    :type args: Any
+    :param kwargs:
+    :type kwargs: Any
     :rtype: None
     """
     if which("wakatime-cli") is None:

src/repl_python_wakatime/ipython.py

@@ -66,7 +66,7 @@ def out_prompt_tokens(self) -> list[tuple[_TokenType, str]]:
 
             :rtype: list[tuple[_TokenType, str]]
             """
-            hook(args, kwargs)
+            hook(*args, **kwargs)
             return prompts_class(shell).out_prompt_tokens()
 
     return Ps

src/repl_python_wakatime/ptpython.py

@@ -60,7 +60,7 @@ def out_prompt(self) -> AnyFormattedText:
 
         :rtype: AnyFormattedText
         """
-        self.hook(self.args, self.kwargs)
+        self.hook(*self.args, **self.kwargs)
         return self.prompt_style.out_prompt()
 
 

src/repl_python_wakatime/python.py

@@ -45,7 +45,7 @@ def __str__(self) -> str:
 
         :rtype: str
         """
-        self.hook(self.args, self.kwargs)
+        self.hook(*self.args, **self.kwargs)
         if isinstance(self.ps1, str):
             return self.ps1
         else: