Merge pull request #165 from andronedev/master

Add Docker support

.dockerignore

@@ -0,0 +1 @@
+Dockerfile

.gitignore

@@ -2,3 +2,4 @@ assets/
 .env
 reddit-bot-351418-5560ebc49cac.json
 __pycache__
+out

Dockerfile

@@ -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"]

build.sh

@@ -0,0 +1 @@
+docker build -t rvmt .

run.sh

@@ -0,0 +1 @@
+docker run -v $(pwd)/out/:/app/assets -v $(pwd)/.env:/app/.env -it rvmt

video_creation/screenshot_downloader.py

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