Handle TCPTransport closed exception

2022-11-12 16:47:13 +02:00
parent 22e623a992
commit c02d23cead
1 changed files with 6 additions and 2 deletions
--- a/mikrotik.py
+++ b/mikrotik.py
@@ -195,8 +195,12 @@ async def view_export(request):
    mk, lock = pool[target]

    async with lock:
-        async for line in wrap(scrape_mikrotik(mk)):
-            await response.send(line + "\n")
+        try:
+            async for line in wrap(scrape_mikrotik(mk)):
+                await response.send(line + "\n")
+        except RuntimeError:
+            # Handle TCPTransport closed exception
+            pool.pop(target)


 app.run(host="0.0.0.0", port=3001)