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?")