mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-12-23 02:01:55 +00:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import asyncio
|
|
from BIRD.Basic import Basic, Code
|
|
|
|
class StatusException(Exception):
|
|
def __init__(self, msg):
|
|
Exception.__init__(self, "Failed to parse status: " + msg)
|
|
|
|
class Status(Basic):
|
|
async def update(self):
|
|
self.data = {}
|
|
|
|
await self.bird.cli.open()
|
|
data = await self.bird.cli.socket.command("show status")
|
|
|
|
if data[0]["code"] != Code.Version:
|
|
raise StatusException(f"BIRD version not on the first line, got {data[0]['code']}")
|
|
|
|
self.data["version"] = data[0]["data"]
|
|
|
|
if data[-1]["code"] != Code.Status:
|
|
raise StatusException(f"BIRD status not on the last line, got {data[-1]['code']}")
|
|
|
|
self.data["status"] = data[-1]["data"]
|
|
|
|
# for d in data[1:-1]:
|
|
|
|
|
|
|
|
class VersionException(Exception):
|
|
def __init__(self, msg):
|
|
Exception.__init__(self, "Failed to parse version from socket hello: " + msg)
|
|
|
|
class Version(Basic):
|
|
async def update(self):
|
|
await self.bird.cli.open()
|
|
hello = self.bird.cli.hello
|
|
|
|
if hello["code"] != Code.Welcome:
|
|
raise VersionException(f"code is {hello['code']}, should be 1")
|
|
|
|
s = hello["data"].split(" ")
|
|
if len(s) != 3 or s[2] != "ready.":
|
|
raise VersionException(f"malformed hello: {hello['data']}")
|
|
|
|
self.data = {
|
|
"name": s[0],
|
|
"version": s[1],
|
|
}
|