Don't crash on timeout error

joinemm · Nov 30, 2023 · dd4c321 · dd4c321
1 parent 8bd879c
commit dd4c321
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/modules/media_embedders.py b/modules/media_embedders.py
@@ -137,8 +137,11 @@ async def download_media(
  "Content-Length"
  ) or response.headers.get("x-full-image-content-length")
  if content_length and int(content_length) < max_filesize:
- buffer = io.BytesIO(await response.read())
- return discord.File(fp=buffer, filename=filename, spoiler=spoiler)
+ try:
+ buffer = io.BytesIO(await response.read())
+ return discord.File(fp=buffer, filename=filename, spoiler=spoiler)
+ except asyncio.TimeoutError:
+ pass
  try:
  # try to stream until we hit our limit
  buffer = b""