0
0
mirror of https://gitlab.nic.cz/labs/bird.git synced 2024-10-18 09:58:43 +00:00
bird/python/BIRD/CLI.py
2024-07-19 20:10:40 +02:00

40 lines
1.1 KiB
Python

from . import ShowRoute
class Transport:
pass
class CLI:
def __init__(self, transport: Transport):
self.transport = transport
async def down(self):
return await self.transport.send_cmd("down")
async def configure(self, file=None, undo=False):
if undo:
return await self.transport.send_cmd("configure", "undo")
if file:
return await self.transport.send_cmd("configure", f'"{file}"')
return await self.transport.send_cmd("configure")
async def enable(self, proto: str):
return await self.transport.send_cmd("enable", proto)
async def disable(self, proto: str):
return await self.transport.send_cmd("disable", proto)
async def show_route(self, table=["all"], args=[]):
cmd = [ "show", "route" ]
for t in table:
cmd.append("table")
cmd.append(t)
cmd += args
result = await self.transport.send_cmd(*cmd)
if len(result["err"]):
raise Exception(f"Command {cmd} returned {result['err'].decode()}, stdout={result['out'].decode()}")
return ShowRoute.parse(result["out"].decode())