polyfill for orig_argv

Author: willmcgugan

examples/argv.py

@@ -0,0 +1,2 @@
+import sys
+print(sys.orig_argv)

trogon/detect_run_string.py

@@ -1,11 +1,27 @@
+from __future__ import annotations
+
 import os
 import shlex
 import sys
 
 
-def detect_run_string(path=None, _main=sys.modules["__main__"]):
-    """This is a slightly modified version of a function from Click.
-    """
+def get_orig_argv() -> list[str]:
+    """Polyfil for orig_argv"""
+    if hasattr(sys, "orig_argv"):
+        return sys.orig_argv
+    import ctypes
+
+    _argv = ctypes.POINTER(ctypes.c_wchar_p)()
+    _argc = ctypes.c_int()
+
+    ctypes.pythonapi.Py_GetArgcArgv(ctypes.byref(_argc), ctypes.byref(_argv))
+
+    argv = _argv[: _argc.value]
+    return argv
+
+
+def detect_run_string(path=None, _main=sys.modules["__main__"]) -> str:
+    """This is a slightly modified version of a function from Click."""
     if not path:
         path = sys.argv[0]
 
@@ -20,8 +36,9 @@ def detect_run_string(path=None, _main=sys.modules["__main__"]):
     ):
         # Executed a file, like "python app.py".
         file_path = shlex.quote(os.path.basename(path))
-        if sys.orig_argv[0] == "python":
-            prefix = f"{sys.orig_argv[0]} "
+        argv = get_orig_argv()
+        if argv[0] == "python":
+            prefix = f"{argv[0]} "
         else:
             prefix = ""
         return f"{prefix}{file_path}"

trogon/trogon.py

@@ -52,6 +52,11 @@ class CommandBuilder(Screen):
         ),
         Binding(key="ctrl+o", action="show_command_info", description="Command Info"),
         Binding(key="f1", action="about", description="About Trogon"),
+        Binding(
+            key="ctrl+s",
+            action="focus('search')",
+            description="Search",
+        ),
     ]
 
     def __init__(

trogon/widgets/form.py

@@ -104,13 +104,19 @@ def compose(self) -> ComposeResult:
         with VerticalScroll() as vs:
             vs.can_focus = False
 
-            yield Input(placeholder="Filter parameters...", classes="command-form-filter-input")
+            yield Input(
+                placeholder="Search...",
+                classes="command-form-filter-input",
+                id="search",
+            )
 
             while command_node is not None:
                 options = command_node.options
                 arguments = command_node.arguments
                 if options or arguments:
-                    with Vertical(classes="command-form-command-group", id=command_node.key) as v:
+                    with Vertical(
+                        classes="command-form-command-group", id=command_node.key
+                    ) as v:
                         is_inherited = command_node is not self.command_schema
                         v.border_title = (
                             f"{'↪ ' if is_inherited else ''}{command_node.name}"
@@ -168,9 +174,7 @@ def _form_changed(self) -> None:
             # For each of the options in the schema for this command,
             # lets grab the values the user has supplied for them in the form.
             for option in command.options:
-                parameter_control = self.query_one(
-                    f"#{option.key}", ParameterControls
-                )
+                parameter_control = self.query_one(f"#{option.key}", ParameterControls)
                 value = parameter_control.get_values()
                 for v in value.values:
                     assert isinstance(v, tuple)
@@ -191,9 +195,7 @@ def _form_changed(self) -> None:
                     argument_datas.append(argument_data)
 
             assert all(isinstance(option.value, tuple) for option in option_datas)
-            assert all(
-                isinstance(argument.value, tuple) for argument in argument_datas
-            )
+            assert all(isinstance(argument.value, tuple) for argument in argument_datas)
             command_data = UserCommandData(
                 name=command.name,
                 options=option_datas,