complete if exception

2022-07-22 05:29:04 +00:00 · 2022-07-22 05:29:04 +00:00 · a7008b3550
commit a7008b3550
parent 4693ae5e80
1 changed files with 4 additions and 3 deletions
--- 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: