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.
pull/418/head
PatatjeMC 2 years ago
parent c028c73ace
commit 40134e02e8

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

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

Loading…
Cancel
Save