From f119457896d32db3c7a9ece4b7ed864eedd95fa2 Mon Sep 17 00:00:00 2001
From: Mohamed Bechir Mejri
 <76930306+MohamedBechirMejri@users.noreply.github.com>
Date: Fri, 19 May 2023 00:47:39 +0100
Subject: [PATCH] fix #1647

---
 video_creation/screenshot_downloader.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/video_creation/screenshot_downloader.py b/video_creation/screenshot_downloader.py
index 1c3ab5e..bffa430 100644
--- a/video_creation/screenshot_downloader.py
+++ b/video_creation/screenshot_downloader.py
@@ -158,9 +158,7 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int):
 
         postcontentpath = f"assets/temp/{reddit_id}/png/title.png"
         try:
-            page.locator('[data-test-id="post-content"]').screenshot(
-                path=postcontentpath
-            )
+            page.locator("//shreddit-post").screenshot(path=postcontentpath)
         except Exception as e:
             print_substep("Something went wrong!", style="red")
             resp = input(
@@ -214,9 +212,12 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int):
                         [comment_tl, comment["comment_id"]],
                     )
                 try:
-                    page.locator(f"#t1_{comment['comment_id']}").screenshot(
-                        path=f"assets/temp/{reddit_id}/png/comment_{idx}.png"
-                    )
+                    permalink_arr = comment["comment_url"].split("/")
+                    permalink_arr[-3] = "comment"
+                    permalink = "/".join(permalink_arr)
+                    page.locator(
+                        f"shreddit-comment[permalink='{permalink}']"
+                    ).screenshot(path=f"assets/temp/{reddit_id}/png/comment_{idx}.png")
                 except TimeoutError:
                     del reddit_object["comments"]
                     screenshot_num += 1