Merge pull request #305 from MattCatz/parameters

: pass parameters if given initial_input
carlmontanari · Oct 25, 2023 · 4ae027a · 4ae027a
2 parents d0eb38b + 984b8b4
commit 4ae027a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/scrapli/driver/generic/async_driver.py b/scrapli/driver/generic/async_driver.py
@@ -530,7 +530,13 @@ def callback_two(cls: GenericDriver, read_output: str):
         """
         if initial_input is not None:
             self.channel.write(channel_input=f"{initial_input}{self.comms_return_char}")
-            return await self.read_callback(callbacks=callbacks, initial_input=None)
+            return await self.read_callback(
+                callbacks=callbacks,
+                initial_input=None,
+                read_output=read_output,
+                read_delay=read_delay,
+                read_timeout=read_timeout,
+            )
 
         original_transport_timeout = self.timeout_transport
 

diff --git a/scrapli/driver/generic/sync_driver.py b/scrapli/driver/generic/sync_driver.py
@@ -531,7 +531,13 @@ def callback_two(cls: GenericDriver, read_output: str):
         """
         if initial_input is not None:
             self.channel.write(channel_input=f"{initial_input}{self.comms_return_char}")
-            return self.read_callback(callbacks=callbacks, initial_input=None)
+            return self.read_callback(
+                callbacks=callbacks,
+                initial_input=None,
+                read_output=read_output,
+                read_delay=read_delay,
+                read_timeout=read_timeout,
+            )
 
         original_transport_timeout = self.timeout_transport