Better anti NSFW post

Instead of looking at the screen to look if it's nsfw it checks it earlier so it doesn't have to create all the mp3's and open the browser to do it. Also it fixes some errors because of it cause playwright was being buggy with it for some reason.
This commit is contained in:
PatatjeMC committed 2022-06-09 16:00:10 +02:00
1 parent c028c73ace
commit 40134e02e8
2 files changed
+6 -6

No files matched your search

+6
View File
@@ -1,5 +1,7 @@
from typing import List
import json
from os import getenv
from utils.console import print_substep
def get_subreddit_undone(submissions: List, subreddit):
"""
@@ -10,6 +12,10 @@ def get_subreddit_undone(submissions: List, subreddit):
for submission in submissions:
if already_done(done_videos, submission):
continue
if submission.over_18:
if getenv("ALLOW_NSFW").casefold() == "false":
print_substep("NSFW Post Detected. Skipping...")
continue
return submission
return get_subreddit_undone(subreddit.top(time_filter="hour"), subreddit) # all of the videos in hot have already been done
-6
View File
@@ -40,12 +40,6 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num):
if page.locator('[data-testid="content-gate"]').is_visible():
# This means the post is NSFW and requires to click the proceed button.
if getenv("ALLOW_NSFW").casefold() == "false":
print_substep("NSFW Post Detected. Skipping...")
from main import main
main()
print_substep("Post is NSFW. You are spicy... :fire:")
page.locator('[data-testid="content-gate"] button').click()