X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/1a3ec903e591304b72bd0d394219729ac76bfed0..5fcf5ebe7b89517edcb610bf1f1df8abf99df9ae:/sdk/python/arvados/events.py diff --git a/sdk/python/arvados/events.py b/sdk/python/arvados/events.py index 3df934a981..b7d610d66e 100644 --- a/sdk/python/arvados/events.py +++ b/sdk/python/arvados/events.py @@ -6,6 +6,9 @@ import time import ssl import re import config +import logging + +_logger = logging.getLogger('arvados.events') class EventClient(WebSocketClient): def __init__(self, url, filters, on_event): @@ -34,10 +37,13 @@ class EventClient(WebSocketClient): pass def subscribe(api, filters, on_event): + ws = None try: url = "{}?api_token={}".format(api._rootDesc['websocketUrl'], config.get('ARVADOS_API_TOKEN')) ws = EventClient(url, filters, on_event) ws.connect() return ws - except: - ws.close_connection() + except Exception: + if (ws): + ws.close_connection() + raise