mirrors/thatmattlove-hyperglass
1
0
Fork 1
mirror of https://github.com/thatmattlove/hyperglass.git synced 2026-01-17 08:48:05 +00:00
Code Issues Activity

Fix remove_command plugin errors

Browse source
This commit is contained in:
thatmattlove 2021-09-13 02:40:10 -07:00
parent 7bd23e58c8
commit 26db076c57
1 changed files with 20 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
hyperglass/plugins/_builtin/remove_command.py
View file

@ -1,13 +1,16 @@
"""Remove anything before the command if found in output."""
# Standard Library
from typing import TYPE_CHECKING
from typing import TYPE_CHECKING, Sequence
# Third Party
from pydantic import PrivateAttr
# Project
from hyperglass.util.typing import is_type
# Local
from .._output import OutputPlugin
from .._output import OutputType, OutputPlugin
if TYPE_CHECKING:
# Project
@ -19,14 +22,21 @@ class RemoveCommand(OutputPlugin):
__hyperglass_builtin__: bool = PrivateAttr(True)
def process(self, device_output: str, device: "Device") -> str:
def process(self, device_output: OutputType, device: "Device") -> Sequence[str]:
"""Remove anything before the command if found in output."""
output = device_output.strip().split("\n")
for command in device.directive_commands:
for line in output:
if command in line:
idx = output.index(line) + 1
output = output[idx:]
def _remove_command(output_in: str) -> str:
output_out = device_output.strip().split("\n")
return "\n".join(output)
for command in device.directive_commands:
for line in output_out:
if command in line:
idx = output_out.index(line) + 1
output_out = output_out[idx:]
return "\n".join(output_out)
if is_type(device_output, str):
return tuple(_remove_command(o) for o in device_output)
return device_output