From acb5cc5c5e854b45edffc3643a9d076a6f31c5ff Mon Sep 17 00:00:00 2001 From: zevaryx Date: Fri, 24 Jun 2022 17:19:19 -0600 Subject: [PATCH] Update link regex pattern --- jarvis_tasks/tasks/reddit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/jarvis_tasks/tasks/reddit.py b/jarvis_tasks/tasks/reddit.py index 5204faa..cb89b0a 100644 --- a/jarvis_tasks/tasks/reddit.py +++ b/jarvis_tasks/tasks/reddit.py @@ -26,7 +26,7 @@ config = TaskConfig.from_yaml() config.reddit["user_agent"] = config.reddit.get("user_agent", DEFAULT_USER_AGENT) running = [] logger = logging.getLogger(__name__) -image_link = re.compile(r"(.*preview\.redd\.it\/)(.*\..*)\?") +image_link = re.compile(r"https?://(?:www)?\.?preview\.redd\.it\/(.*\..*)\?.*") async def post_embeds(sub: Sub, post: Submission, reddit: Reddit) -> Optional[List[Embed]]: @@ -76,7 +76,7 @@ async def post_embeds(sub: Sub, post: Submission, reddit: Reddit) -> Optional[Li if post.spoiler: content += "||" content += f"\n\n[View this post]({url})" - content = image_link.sub(content, "https://i.redd.it/\2") + content = "\n".join(image_link.sub(r"https://i.redd.it/\1", x) for x in content.split("\n")) if not images and not content: logger.debug(f"Post {post.id} had neither content nor images?")