6 files changed
+25
-4
No files matched your search
@@ -0,0 +1 @@
|
||||
Dockerfile
|
||||
+2
-1
@@ -1,4 +1,5 @@
|
||||
assets/
|
||||
.env
|
||||
reddit-bot-351418-5560ebc49cac.json
|
||||
__pycache__
|
||||
__pycache__
|
||||
out
|
||||
+15
@@ -0,0 +1,15 @@
|
||||
FROM mcr.microsoft.com/playwright
|
||||
|
||||
RUN apt update
|
||||
RUN apt install python3-pip -y
|
||||
|
||||
RUN mkdir /app
|
||||
ADD . /app
|
||||
WORKDIR /app
|
||||
RUN pip install -r requirements.txt
|
||||
|
||||
# tricks for pytube : https://github.com/elebumm/RedditVideoMakerBot/issues/142
|
||||
# (NOTE : This is no longer useful since pytube was removed from the dependencies)
|
||||
# RUN sed -i 's/re.compile(r"^\\w+\\W")/re.compile(r"^\\$*\\w+\\W")/' /usr/local/lib/python3.8/dist-packages/pytube/cipher.py
|
||||
|
||||
CMD ["python3", "main.py"]
|
||||
@@ -0,0 +1 @@
|
||||
docker run -v $(pwd)/out/:/app/assets -v $(pwd)/.env:/app/.env -it rvmt
|
||||
@@ -1,4 +1,4 @@
|
||||
from playwright.sync_api import sync_playwright
|
||||
from playwright.sync_api import sync_playwright, ViewportSize
|
||||
from pathlib import Path
|
||||
from rich.progress import track
|
||||
from utils.console import print_step, print_substep
|
||||
@@ -31,7 +31,7 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num, theme):
|
||||
# Get the thread screenshot
|
||||
page = context.new_page()
|
||||
page.goto(reddit_object["thread_url"])
|
||||
|
||||
page.set_viewport_size(ViewportSize(width=1920, height=1080))
|
||||
if page.locator('[data-testid="content-gate"]').is_visible():
|
||||
# This means the post is NSFW and requires to click the proceed button.
|
||||
|
||||
@@ -57,4 +57,6 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num, theme):
|
||||
page.locator(f"#t1_{comment['comment_id']}").screenshot(
|
||||
path=f"assets/png/comment_{idx}.png"
|
||||
)
|
||||
print_substep("Screenshots downloaded successfully.", style="bold green")
|
||||
|
||||
print_substep("Screenshots downloaded Successfully.",
|
||||
style="bold green")
|
||||
Reference in new issue
Block a user