add: NSFW and subreddit to videos.json

pull/894/head
Tomovic 3 years ago
commit 24c55877df

@ -34,11 +34,12 @@ def check_done(
return redditobj
def save_data(filename: str, reddit_title: str, reddit_id: str, credit: str, NSFW: bool):
def save_data(subreddit: str, filename: str, reddit_title: str, reddit_id: str, credit: str, NSFW: bool):
"""Saves the videos that have already been generated to a JSON file in video_creation/data/videos.json
Args:
filename (str): The finished video title name
@param subreddit:
@param filename:
@param reddit_id:
@param reddit_title:
@ -49,6 +50,7 @@ def save_data(filename: str, reddit_title: str, reddit_id: str, credit: str, NSF
if reddit_id in [video["id"] for video in done_vids]:
return # video already done but was specified to continue anyway in the config file
payload = {
"subreddit": subreddit,
"id": reddit_id,
"time": str(int(time.time())),
"background_credit": credit,

@ -119,7 +119,7 @@ def make_final_video(
filename = f"{name_normalize(title)}.mp4"
subreddit = settings.config["reddit"]["thread"]["subreddit"]
save_data(filename, title, idx, background_config[2], nsfw)
save_data(subreddit, filename, title, idx, background_config[2], nsfw)
if not exists(f"./results/{subreddit}"):
print_substep("The results folder didn't exist so I made it")

Loading…
Cancel
Save