[internal] Add a way to append values to optional arguments in `parse…

…_arguments` decorator (#1122) Add support for the `append` action for ArgumentParser in the `parse_arguments` decorator on optional arguments. This makes us able to support this: ``` command --arg 1 --arg 2 --arg 3 ``` And get `[1, 2, 3]` as the `arg` value. This CL is required to make changes to #1120 before merging. --------- Co-authored-by: crazy hugsy <[email protected]>
hugsy · Jun 24, 2024 · 6ddd780 · 6ddd780
1 parent c450a2d
commit 6ddd780
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gef.py b/gef.py
@@ -491,6 +491,10 @@ def wrapper(*args: Any, **kwargs: Any) -> Callable:
                 elif argtype is bool:
                     parser.add_argument(*argname, action="store_false" if argvalue else "store_true")
                     continue
+                elif argtype in (list, tuple):
+                    parser.add_argument(*argname, type=type(argvalue[0]),
+                                        default=[], action="append")
+                    continue
                 parser.add_argument(*argname, type=argtype, default=argvalue)
 
             parsed_args = parser.parse_args(*(args[1:]))