build log: use async with

Signed-off-by: Karol Swiderski <szakalboss@gmail.com>
molior-dbs · May 27, 2021 · 93dfa56 · 93dfa56
1 parent c686374
commit 93dfa56
Showing 1 changed file with 11 additions and 12 deletions.
diff --git a/molior/molior/queues.py b/molior/molior/queues.py
@@ -82,18 +82,17 @@ async def buildlog_writer(build_id):
         del buildlogs[build_id]
         return
     try:
-        afp = AIOFile(filename, 'a')
-        await afp.open()
-        writer = Writer(afp)
-        while True:
-            msg = await dequeue(buildlogs[build_id])
-            if msg is None:
-                await enqueue_backend({"logging_done": build_id})
-                continue
-            elif msg is False:
-                break
-            await writer(msg)
-            await afp.fsync()
+        async with AIOFile(filename, 'a') as afp:
+            writer = Writer(afp)
+            while True:
+                msg = await dequeue(buildlogs[build_id])
+                if msg is None:
+                    await enqueue_backend({"logging_done": build_id})
+                    continue
+                elif msg is False:
+                    break
+                await writer(msg)
+                await afp.fsync()
     except Exception as exc:
         logger.exception(exc)