mirrors/bird
mirrors/bird
0
0
Fork 0
mirror of https://gitlab.nic.cz/labs/bird.git synced 2024-12-31 06:01:52 +00:00
Code Releases Activity

Python CLI Package: Locking BIRD object

Browse Source
This commit is contained in:
Maria Matejka 2023-05-23 13:43:31 +02:00
parent 545d9b823b
commit 30dc8824ec
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python/BIRD/__init__.py
View File

@ -92,6 +92,7 @@ class CLI:
self.socket = Socket(name) self.socket = Socket(name)
self.connected = False self.connected = False
self.hello = None self.hello = None
self.lock = asyncio.Lock()
async def open(self): async def open(self):
if self.hello is not None: if self.hello is not None:
@ -125,8 +126,10 @@ class BIRD:
raise BIRDException("Tried to enter BIRD context (async with) more than once") raise BIRDException("Tried to enter BIRD context (async with) more than once")
self.within = True self.within = True
await self.cli.lock.acquire()
return self return self
async def __aexit__(self, *args): async def __aexit__(self, *args):
await self.cli.close() await self.cli.close()
self.cli.lock.release()
self.within = False self.within = False