fix(endpoint): remove multiple-exception catch

Does not work in python3.7
chime-experiment · Apr 8, 2020 · ff74f47 · ff74f47
1 parent 9a7f9a7
commit ff74f47
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/coco/endpoint.py b/coco/endpoint.py
@@ -1,8 +1,5 @@
 """coco endpoint module."""
 from aiohttp import (
-    ClientResponseError,
-    ClientConnectionError,
-    ClientTimeout,
     ClientSession,
     ContentTypeError,
 )
@@ -14,7 +11,6 @@
 
 import json
 from pydoc import locate
-import requests
 import sanic
 
 from . import Result, ExternalForward, CocoForward, metric
@@ -568,13 +564,9 @@ def client_send_request(host, port, metrics_port, args):
         async def print_queue_size(metric_request_count):
             try:
                 q_size = await metric.get("coco_queue_length_total", metrics_port)
-            except (
-                ClientResponseError,
-                ClientConnectionError,
-                ClientTimeout,
-                InternalError,
-            ) as err:
-                print(f"Couldn't get queue fill level from cocod: {err}")
+            except Exception as err:
+                if not isinstance(err, asyncio.CancelledError):
+                    print(f"Couldn't get queue fill level from cocod: {err}")
                 return
             print(
                 f"\rThere are {int(q_size)} requests in the queue.",
@@ -627,7 +619,7 @@ async def request_and_wait():
                     )
                     if main_request in done:
                         queue_size.cancel()
-                        print("/n")
+                        print("\n")
                         break
 
                     # Wait a moment before getting metric again