Add netmico command timeout

napalm-automation-community · Apr 23, 2024 · fda349e · fda349e
1 parent d74cfb9
commit fda349e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/napalm_s350/s350.py b/napalm_s350/s350.py
@@ -100,11 +100,17 @@ def _send_command(self, command):
         try:
             if isinstance(command, list):
                 for cmd in command:
-                    output = self.device.send_command(cmd)
+                    output = self.device.send_command(
+                            cmd,
+                            read_timeout=self.timeout
+                            )
                     if "% Invalid" not in output:
                         break
             else:
-                output = self.device.send_command(command)
+                output = self.device.send_command(
+                        command,
+                        read_timeout=self.timeout
+                        )
             return output.strip()
         except (socket.error, EOFError) as e:
             raise ConnectionClosedException(str(e))