From f457f6258ecb5a2043936d90423280a755755479 Mon Sep 17 00:00:00 2001 From: Andrey Savich <46422808+pieckenst@users.noreply.github.com> Date: Thu, 29 Aug 2024 14:08:59 +0300 Subject: [PATCH] Update extra.py --- utils/extra.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/utils/extra.py b/utils/extra.py index 374293c7..c474d88d 100644 --- a/utils/extra.py +++ b/utils/extra.py @@ -191,7 +191,20 @@ async def rtfm(bot: JDBot, url: str) -> list[RtfmObject]: header_lines = [line for line in lines[:10] if not line.startswith(b"#")] content_lines = lines[10:] - full_data = zlib.decompress(b"\n".join(header_lines + content_lines)) + try: + full_data = zlib.decompress(b"\n".join(header_lines + content_lines)) + except zlib.error: + # If decompression fails, try to skip the first few lines + for i in range(len(lines)): + try: + full_data = zlib.decompress(b"\n".join(lines[i:])) + break + except zlib.error: + continue + else: + # If all attempts fail, raise an exception + raise ValueError("Unable to decompress data") + normal_data = full_data.decode() new_list = normal_data.split("\n") @@ -209,6 +222,7 @@ async def rtfm(bot: JDBot, url: str) -> list[RtfmObject]: return results + async def asset_converter(ctx, assets): assets = list(assets) attachments = ctx.message.attachments