From a7008b3550f1dcee884a1c4f4acef8a083be6c08 Mon Sep 17 00:00:00 2001 From: jbannon Date: Fri, 22 Jul 2022 05:29:04 +0000 Subject: [PATCH] complete if exception --- src/ytdl_sub/subscriptions/subscription.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/ytdl_sub/subscriptions/subscription.py b/src/ytdl_sub/subscriptions/subscription.py index 3b85be4b..3b3174ae 100644 --- a/src/ytdl_sub/subscriptions/subscription.py +++ b/src/ytdl_sub/subscriptions/subscription.py @@ -239,9 +239,10 @@ class Subscription: info_json_listener.start() - yield - - info_json_listener.complete = True + try: + yield + finally: + info_json_listener.complete = True @contextlib.contextmanager def _subscription_download_context_managers(self) -> None: