mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 05:30:46 +08:00
oliver
0d85b87be7
Extend port discovery/sampling beyond ZTE, harden IOSv session prep, and avoid send_command_timing so Building configuration does not truncate config sync. Co-authored-by: Cursor <cursoragent@cursor.com>
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
"""Shared Netmiko helpers for managed NE connect/collect."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
def normalize_netmiko_device_type(device_type: str, protocol: str) -> str:
|
|
dt = str(device_type or "").strip()
|
|
proto = str(protocol or "ssh").strip().lower()
|
|
if "zte" in dt.lower():
|
|
if dt == "zte":
|
|
return f"zte_zxros_{proto}"
|
|
if "telnet" not in dt and "ssh" not in dt:
|
|
return f"{dt}_{proto}"
|
|
return dt
|
|
if "telnet" not in dt and "ssh" not in dt:
|
|
return f"{dt}_{proto}"
|
|
return dt
|
|
|
|
|
|
def is_cisco_ios_device_type(device_type: str) -> bool:
|
|
dt = str(device_type or "").strip().lower()
|
|
return "cisco_ios" in dt or dt in {"cisco_ios", "cisco_ios_ssh", "cisco_ios_telnet"}
|
|
|
|
|
|
def send_show_command(conn: Any, command: str, *, read_timeout: int = 120) -> str:
|
|
"""Send a show/display command via ``send_command`` (wait for device prompt).
|
|
|
|
Do not use ``send_command_timing`` for Cisco config collection: long idle during
|
|
``Building configuration...`` is treated as end-of-output and truncates the config.
|
|
"""
|
|
cmd = str(command or "").strip()
|
|
if not cmd:
|
|
return ""
|
|
return str(conn.send_command(command_string=cmd, read_timeout=read_timeout) or "")
|