forked from mirrors/thatmattlove-hyperglass
23 lines
701 B
Python
23 lines
701 B
Python
"""Remove anything before the command if found in output."""
|
|
|
|
# Project
|
|
from hyperglass.models.config.devices import Device
|
|
|
|
# Local
|
|
from .._output import OutputPlugin
|
|
|
|
|
|
class RemoveCommand(OutputPlugin):
|
|
"""Remove anything before the command if found in output."""
|
|
|
|
def process(self, device_output: str, device: Device) -> 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:]
|
|
|
|
return "\n".join(output)
|