Skip to content

Commit 3e0676b

Browse files
committed
Make debug key map expandable
1 parent 91b3c49 commit 3e0676b

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

click_async_plugins/debug.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,6 @@ def puts(s: str) -> None:
2222
print(s, file=sys.stderr)
2323

2424

25-
def simulate_reload_tpdata(clictx: CliContext) -> None:
26-
"""Simulate event that TPData was reloaded"""
27-
clictx.itc.fire("tpdata")
28-
29-
3025
def echo_newline(_: CliContext) -> None:
3126
"""Outputs a new line"""
3227
puts("")
@@ -138,21 +133,24 @@ async def _monitor_stdin(clictx: CliContext, key_to_cmd: KeyCmdMapType) -> None:
138133

139134

140135
@asynccontextmanager
141-
async def monitor_stdin_for_debug_commands(clictx: CliContext) -> PluginLifespan:
136+
async def monitor_stdin_for_debug_commands(
137+
clictx: CliContext, *, key_to_cmd: KeyCmdMapType | None = None
138+
) -> PluginLifespan:
139+
key_to_cmd = key_to_cmd or {}
140+
142141
increase_loglevel = partial(adjust_loglevel, change=-10)
143142
increase_loglevel.__doc__ = "Increase the logging level"
144143
decrease_loglevel = partial(adjust_loglevel, change=10)
145144
decrease_loglevel.__doc__ = "Decrease the logging level"
146145

147-
key_to_cmd = {
146+
base_map = {
148147
0xA: KeyAndFunc(r"\n", echo_newline),
149-
0x12: KeyAndFunc("^R", simulate_reload_tpdata),
150148
0x1B: KeyAndFunc("<Esc>", terminal_block),
151149
0x4: KeyAndFunc("^D", debug_info),
152150
0x2B: KeyAndFunc("+", increase_loglevel),
153151
0x2D: KeyAndFunc("-", decrease_loglevel),
154152
}
155-
yield _monitor_stdin(clictx, key_to_cmd)
153+
yield _monitor_stdin(clictx, base_map | key_to_cmd)
156154

157155

158156
@plugin

0 commit comments

Comments
 (0)