mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2025-01-10 02:51:54 +00:00
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
|
import asyncio
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from BIRD.Basic import BIRDException
|
||
|
|
from BIRD.Socket import Socket
|
||
|
|
from BIRD.Status import Status, Version
|
||
|
|
|
||
|
|
class CLI:
|
||
|
|
def __init__(self, name):
|
||
|
|
self.socket = Socket(name)
|
||
|
|
self.connected = False
|
||
|
|
self.hello = None
|
||
|
|
|
||
|
|
async def open(self):
|
||
|
|
if self.hello is not None:
|
||
|
|
return
|
||
|
|
|
||
|
|
h = await self.socket.open()
|
||
|
|
if len(h) != 1:
|
||
|
|
raise BIRDException("CLI hello should have 1 line, has {len(h)} lines: {h}")
|
||
|
|
|
||
|
|
self.hello = h[0]
|
||
|
|
|
||
|
|
async def close(self):
|
||
|
|
if self.hello is None:
|
||
|
|
return
|
||
|
|
|
||
|
|
await self.socket.close()
|
||
|
|
self.hello = None
|
||
|
|
|
||
|
|
class BIRD:
|
||
|
|
def __init__(self, socket=Path("bird.ctl")):
|
||
|
|
self.cli = CLI(socket)
|
||
|
|
self.version = Version(self)
|
||
|
|
self.status = Status(self)
|
||
|
|
|
||
|
|
self.within = False
|
||
|
|
|
||
|
|
async def __aenter__(self):
|
||
|
|
if self.within:
|
||
|
|
raise BIRDException("Tried to enter BIRD context (async with) more than once")
|
||
|
|
|
||
|
|
self.within = True
|
||
|
|
return self
|
||
|
|
|
||
|
|
async def __aexit__(self, *args):
|
||
|
|
await self.cli.close()
|
||
|
|
self.within = False
|