replbase 0.0.2__tar.gz → 0.0.4__tar.gz

{replbase-0.0.2 → replbase-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: replbase
-Version: 0.0.2
+Version: 0.0.4
 Summary: "Combination of other REPL tools into a reusable class that generates a REPL"
 License: MIT
 Author: Joseph Bochinski

{replbase-0.0.2 → replbase-0.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "replbase"
-version = "0.0.2"
+version = "0.0.4"
 description = "\"Combination of other REPL tools into a reusable class that generates a REPL\""
 authors = [ "Joseph Bochinski <stirgejr@gmail.com>",]
 license = "MIT"

{replbase-0.0.2 → replbase-0.0.4}/replbase/repl_base.py

@@ -17,7 +17,7 @@ import argparse
 import os
 import shlex
 
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Callable, Literal
 
 from prompt_toolkit import PromptSession
@@ -46,6 +46,8 @@ class ReplCommand:
     command: Callable = None
     help_txt: str = ""
     parser: argparse.ArgumentParser = None
+    def_args: list = field(default_factory=list)
+    def_kwargs: dict = field(default_factory=dict)
 
 
 @dataclass
@@ -194,6 +196,8 @@ class ReplBase:
         help_txt: str = "",
         use_parser: bool = False,
         description: str = "",
+        *def_args,
+        **def_kwargs,
     ) -> ReplCommand:
         """Add a command to the REPL
 
@@ -207,6 +211,8 @@ class ReplBase:
                 to the new ReplCommand instance. Defaults to False.
             description (str, optional): Optional description for the
                 ArgumentParser help text. Defaults to help_txt.
+            def_args: Default arguments for the command function
+            def_kwargs: Default keyword arguments for the command function
 
         Returns:
             ReplCommand: The new ReplCommand instance
@@ -217,6 +223,11 @@ class ReplBase:
             new_cmd.parser = argparse.ArgumentParser(
                 description=description or help_txt
             )
+        if def_args:
+            new_cmd.def_args = def_args
+        if def_kwargs:
+            new_cmd.def_kwargs = def_kwargs
+
         self.commands[cmd_name] = new_cmd
         return new_cmd
 
@@ -292,7 +303,11 @@ class ReplBase:
 
                             cmd.command(cmd.parser.parse_args(cmd_args))
                         else:
-                            cmd.command(*cmd_args)
+                            cmd_args = cmd_args or cmd.def_args
+                            if cmd.def_kwargs:
+                                cmd.command(*cmd_args, **cmd.def_kwargs)
+                            else:
+                                cmd.command(*cmd_args)
                     else:
                         self.print(
                             "[bold yellow][WARNING]: No function provided for command[/bold yellow]"