catch os error

SpringsFern · May 11, 2024 · d04c5cb · d04c5cb
1 parent 6a7bd83
commit d04c5cb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/WebStreamer/utils/custom_dl.py b/WebStreamer/utils/custom_dl.py
@@ -306,6 +306,12 @@ async def yield_file(
                     await cdn_session.stop()
         except (TimeoutError, AttributeError):
             pass
+        except OSError as e:
+            logging.info("Removing Media Session")
+            client.media_sessions.pop(file_id.dc_id, None)
+            await media_session.stop()
+            logging.error(e)
+            raise e
         finally:
             logging.debug(f"Finished yielding file with {current_part} parts.")
             work_loads[file_id.index] -= 1