|
|
@ -10,6 +10,7 @@ from rich.progress import track
|
|
|
|
from utils import settings
|
|
|
|
from utils import settings
|
|
|
|
from utils.console import print_step, print_substep
|
|
|
|
from utils.console import print_step, print_substep
|
|
|
|
from utils.imagenarator import imagemaker
|
|
|
|
from utils.imagenarator import imagemaker
|
|
|
|
|
|
|
|
|
|
|
|
from utils.videos import save_data
|
|
|
|
from utils.videos import save_data
|
|
|
|
|
|
|
|
|
|
|
|
__all__ = ["download_screenshots_of_reddit_posts"]
|
|
|
|
__all__ = ["download_screenshots_of_reddit_posts"]
|
|
|
@ -37,7 +38,7 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int):
|
|
|
|
with sync_playwright() as p:
|
|
|
|
with sync_playwright() as p:
|
|
|
|
print_substep("Launching Headless Browser...")
|
|
|
|
print_substep("Launching Headless Browser...")
|
|
|
|
|
|
|
|
|
|
|
|
browser = p.chromium.launch() # headless=False #to check for chrome view
|
|
|
|
browser = p.chromium.launch(headless=False) # headless=False #to check for chrome view
|
|
|
|
context = browser.new_context()
|
|
|
|
context = browser.new_context()
|
|
|
|
# Device scale factor (or dsf for short) allows us to increase the resolution of the screenshots
|
|
|
|
# Device scale factor (or dsf for short) allows us to increase the resolution of the screenshots
|
|
|
|
# When the dsf is 1, the width of the screenshot is 600 pixels
|
|
|
|
# When the dsf is 1, the width of the screenshot is 600 pixels
|
|
|
@ -71,6 +72,20 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int):
|
|
|
|
|
|
|
|
|
|
|
|
context.add_cookies(cookies) # load preference cookies
|
|
|
|
context.add_cookies(cookies) # load preference cookies
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Login to Reddit
|
|
|
|
|
|
|
|
print_substep("Logging in to Reddit...")
|
|
|
|
|
|
|
|
page = context.new_page()
|
|
|
|
|
|
|
|
page.goto("https://www.reddit.com/login", timeout=0)
|
|
|
|
|
|
|
|
page.set_viewport_size(ViewportSize(width=1920, height=1080))
|
|
|
|
|
|
|
|
page.wait_for_load_state()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
page.locator('[name="username"]').fill(settings.config["reddit"]["creds"]["username"])
|
|
|
|
|
|
|
|
page.locator('[name="password"]').fill(settings.config["reddit"]["creds"]["password"])
|
|
|
|
|
|
|
|
page.locator("button:has-text('Log In')").click()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
page.wait_for_load_state() # Wait for page to fully load and add 5 seconds
|
|
|
|
|
|
|
|
page.wait_for_timeout(5000)
|
|
|
|
|
|
|
|
|
|
|
|
# Get the thread screenshot
|
|
|
|
# Get the thread screenshot
|
|
|
|
page = context.new_page()
|
|
|
|
page = context.new_page()
|
|
|
|
page.goto(reddit_object["thread_url"], timeout=0)
|
|
|
|
page.goto(reddit_object["thread_url"], timeout=0)
|
|
|
|