@@ -13,7 +13,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
1313
1414 def session_preparation (self ) -> None :
1515 self ._test_channel_read ()
16- self .set_base_prompt (pri_prompt_terminator = "$" )
16+ self .set_base_prompt (alt_prompt_terminator = "$" )
1717
1818 def enable (
1919 self ,
@@ -36,26 +36,28 @@ def save_config(
3636
3737 def send_command_timing (
3838 self ,
39+ command_string : str ,
3940 * args : Any ,
4041 ** kwargs : Any ,
4142 ) -> Union [str , List [Any ], Dict [str , Any ]]:
42- real_command_string = kwargs [ " command_string" ]
43+ real_command_string = command_string
4344 real_strip_command = kwargs .get ("strip_command" , True )
4445 real_strip_prompt = kwargs .get ("strip_prompt" , True )
45- command = kwargs [ " command_string" ] + " \n "
46+ command = command_string
4647 more = r"< Hit any key >"
4748 kwargs ["strip_prompt" ] = False
4849 kwargs ["strip_command" ] = False
4950
50- output = str (super ().send_command_timing (* args , ** kwargs ))
51+ output = str (super ().send_command_timing (command_string , * args , ** kwargs ))
5152
52- kwargs [ " command_string" ] = " "
53+ command_string = " "
5354 kwargs ["normalize" ] = False
5455 kwargs ["strip_command" ] = True
5556 while more in output :
5657 output = re .sub (r"\n" + more , "" , output )
5758 output += str (
5859 super ().send_command_timing (
60+ command_string ,
5961 * args ,
6062 ** kwargs ,
6163 )
0 commit comments