Merge branch 'master' of https://github.com/Tyaaa-aa/RedditVideoMakerBot

.dockerignore

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

.env.template

@@ -10,5 +10,6 @@ FFMPEG_PATH=""
 # Valid options are "yes" and "no" for the variable below
 REDDIT_2FA="no"
 
+THEME=""
 
 SUBREDDIT=""

.gitignore

@@ -3,3 +3,4 @@ final/
 .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 .

requirements.txt

@@ -1,44 +1,8 @@
-appdirs==1.4.4
-black==20.8b1
-certifi==2021.10.8
-charset-normalizer==2.0.12
-click==7.1.2
-commonmark==0.9.1
-decorator==4.4.2
-flake8==3.8.3
-greenlet==1.1.2
 gTTS==2.2.4
-idna==3.3
-imageio==2.19.2
-imageio-ffmpeg==0.4.7
-mccabe==0.6.1
 moviepy==1.0.3
 mutagen==1.45.1
-mypy-extensions==0.4.3
-numpy==1.22.3
-pathspec==0.8.0
-Pillow==9.1.1
 playwright==1.22.0
 praw==7.6.0
-prawcore==2.3.0
-proglog==0.1.10
-pycodestyle==2.6.0
-pyee==8.1.0
-pyflakes==2.2.0
-Pygments==2.12.0
 python-dotenv==0.20.0
-
-regex==2020.10.15
-
-requests==2.27.1
 rich==12.4.4
-six==1.16.0
-toml==0.10.1
-tqdm==4.64.0
-typed-ast==1.5.4
-typing_extensions==4.2.0
-update-checker==0.18.0
-urllib3==1.26.9
-websocket-client==1.3.2
-websockets==10.1
-yt-dlp==2022.5.18
+yt_dlp==2022.5.18

run.sh

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

video_creation/cookies.json

@@ -0,0 +1,8 @@
+[
+  {
+    "name": "USER",
+    "value": "eyJwcmVmcyI6eyJ0b3BDb250ZW50RGlzbWlzc2FsVGltZSI6MCwiZ2xvYmFsVGhlbWUiOiJSRURESVQiLCJuaWdodG1vZGUiOnRydWUsImNvbGxhcHNlZFRyYXlTZWN0aW9ucyI6eyJmYXZvcml0ZXMiOmZhbHNlLCJtdWx0aXMiOmZhbHNlLCJtb2RlcmF0aW5nIjpmYWxzZSwic3Vic2NyaXB0aW9ucyI6ZmFsc2UsInByb2ZpbGVzIjpmYWxzZX0sInRvcENvbnRlbnRUaW1lc0Rpc21pc3NlZCI6MH19",
+    "domain": ".reddit.com",
+    "path": "/"
+  }
+]

video_creation/screenshot_downloader.py

@@ -1,10 +1,11 @@
-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
+import json
 
 
-def download_screenshots_of_reddit_posts(reddit_object, screenshot_num):
+def download_screenshots_of_reddit_posts(reddit_object, screenshot_num, theme):
     """Downloads screenshots of reddit posts as they are seen on the web.
 
     Args:
@@ -20,11 +21,17 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num):
         print_substep("Launching Headless Browser...")
 
         browser = p.chromium.launch()
+        context = browser.new_context()
+
+        if theme.casefold() == "dark":
+            cookie_file = open('video_creation/cookies.json')
+            cookies = json.load(cookie_file)
+            context.add_cookies(cookies)
 
         # Get the thread screenshot
-        page = browser.new_page()
+        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.
 
@@ -50,4 +57,6 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num):
             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")