Jason
3 years ago
parent
5b3989611f
commit
51484ca401
@ -1,3 +1,4 @@
|
||||
assets/
|
||||
.env
|
||||
reddit-bot-351418-5560ebc49cac.json
|
||||
/.idea
|
||||
|
||||
@ -0,0 +1,61 @@
|
||||
import re
|
||||
|
||||
from utils.console import print_step, print_substep
|
||||
import praw
|
||||
import random
|
||||
from dotenv import load_dotenv
|
||||
import os
|
||||
|
||||
|
||||
def ascifi(text):
|
||||
regrex_pattern = re.compile(pattern="["
|
||||
u"\U0001F600-\U0001F64F" # emoticons
|
||||
u"\U0001F300-\U0001F5FF" # symbols & pictographs
|
||||
u"\U0001F680-\U0001F6FF" # transport & map symbols
|
||||
u"\U0001F1E0-\U0001F1FF" # flags (iOS)
|
||||
"]+", flags=re.UNICODE)
|
||||
return regrex_pattern.sub(r'', text)
|
||||
|
||||
|
||||
def get_subreddit_threads():
|
||||
"""
|
||||
Returns a list of threads from the AskReddit subreddit.
|
||||
"""
|
||||
|
||||
print_step("Getting subreddit threads...")
|
||||
|
||||
content = {}
|
||||
load_dotenv()
|
||||
reddit = praw.Reddit(client_id=os.getenv("REDDIT_CLIENT_ID"), client_secret=os.getenv("REDDIT_CLIENT_SECRET"),
|
||||
user_agent="Accessing AskReddit threads", username=os.getenv("REDDIT_USERNAME"),
|
||||
password=os.getenv("REDDIT_PASSWORD"), )
|
||||
"""
|
||||
Ask user for subreddit input
|
||||
"""
|
||||
if not os.getenv("SUBREDDIT"):
|
||||
subreddit = reddit.subreddit(
|
||||
input("What subreddit would you like to pull from? ")) # if the env isnt set, ask user
|
||||
else:
|
||||
print_substep(f"Using subreddit: r/{os.getenv('SUBREDDIT')} from environment variable config")
|
||||
subreddit = reddit.subreddit(
|
||||
os.getenv("SUBREDDIT")) # Allows you to specify in .env. Done for automation purposes.
|
||||
|
||||
threads = subreddit.hot(limit=25)
|
||||
submission = list(threads)[random.randrange(0, 25)]
|
||||
os.environ["VIDEO_TITLE"] = str(ascifi(submission.title))
|
||||
print_substep(f"Video will be: {os.getenv('VIDEO_TITLE')} :thumbsup:")
|
||||
try:
|
||||
|
||||
content["thread_url"] = submission.url
|
||||
content["thread_title"] = submission.title
|
||||
content["comments"] = []
|
||||
|
||||
for top_level_comment in submission.comments:
|
||||
content["comments"].append(
|
||||
{"comment_body": top_level_comment.body, "comment_url": top_level_comment.permalink,
|
||||
"comment_id": top_level_comment.id, })
|
||||
|
||||
except AttributeError as e:
|
||||
pass
|
||||
print_substep("Received subreddit threads Successfully.", style="bold green")
|
||||
return content
|
||||
@ -0,0 +1,8 @@
|
||||
import os
|
||||
|
||||
|
||||
def cleanup():
|
||||
files = [f for f in os.listdir('.') if f.endswith('.mp4') and 'temp' in f.lower()]
|
||||
for f in files:
|
||||
os.remove(f)
|
||||
|
||||
@ -1,46 +1,50 @@
|
||||
import random
|
||||
from os import listdir, environ
|
||||
from pathlib import Path
|
||||
from random import randrange
|
||||
from pytube import YouTube
|
||||
from pathlib import Path
|
||||
from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
|
||||
from moviepy.editor import VideoFileClip
|
||||
from rich.progress import track
|
||||
|
||||
from utils.console import print_step, print_substep
|
||||
|
||||
|
||||
def get_start_and_end_times(video_length, length_of_clip):
|
||||
|
||||
random_time = randrange(180, int(length_of_clip) - int(video_length))
|
||||
return random_time, random_time + video_length
|
||||
|
||||
|
||||
def download_background():
|
||||
"""Downloads the background video from youtube.
|
||||
"""Downloads the backgrounds/s video from youtube.
|
||||
|
||||
Shoutout to: bbswitzer (https://www.youtube.com/watch?v=n_Dv4JMiwK8)
|
||||
Shoutout to: Orbital Gameplay (https://www.youtube.com/watch?v=2X9QGY__0II)
|
||||
"""
|
||||
|
||||
if not Path("assets/mp4/background.mp4").is_file():
|
||||
print_step(
|
||||
"We need to download the Minecraft background video. This is fairly large but it's only done once. 😎"
|
||||
)
|
||||
print_substep("Downloading the background video... please be patient 🙏")
|
||||
YouTube("https://www.youtube.com/watch?v=n_Dv4JMiwK8").streams.filter(
|
||||
res="720p"
|
||||
).first().download(
|
||||
"assets/mp4",
|
||||
filename="background.mp4",
|
||||
)
|
||||
print_substep("Background video downloaded successfully! 🎉", style="bold green")
|
||||
Path("assets/backgronds/").mkdir(parents=True, exist_ok=True)
|
||||
background_options = [ # uri , filename , credit
|
||||
("https://www.youtube.com/watch?v=n_Dv4JMiwK8", "parkour.mp4", 'bbswitzer'),
|
||||
# note: make sure the file name doesn't include a - in it
|
||||
("https://www.youtube.com/watch?v=2X9QGY__0II", "rocket_league.mp4", 'Orbital Gameplay'), ]
|
||||
if listdir('assets/backgrounds/') != len(background_options):
|
||||
print_step("We need to download the backgrounds videos. they are fairly large but it's only done once. 😎")
|
||||
print_substep("Downloading the backgrounds video... please be patient 🙏 ")
|
||||
|
||||
for uri, filename, credit in track(background_options, description="Downloading..."):
|
||||
print_step(f"Downloading {filename} from {uri}")
|
||||
YouTube(uri).streams.filter(res="720p").first().download("assets/backgrounds",
|
||||
filename=f"{credit}-{filename}")
|
||||
print_substep("Background videos downloaded successfully! 🎉", style="bold green")
|
||||
|
||||
|
||||
def chop_background_video(video_length):
|
||||
print_step("Finding a spot in the background video to chop...✂️")
|
||||
background = VideoFileClip("assets/mp4/background.mp4")
|
||||
print_step("Finding a spot in the backgrounds video to chop...✂️")
|
||||
choice = random.choice(listdir('assets/backgrounds'))
|
||||
environ["background_credit"] = choice.split('-')[0]
|
||||
|
||||
background = VideoFileClip(f"assets/backgrounds/{choice}")
|
||||
|
||||
start_time, end_time = get_start_and_end_times(video_length, background.duration)
|
||||
ffmpeg_extract_subclip(
|
||||
"assets/mp4/background.mp4",
|
||||
start_time,
|
||||
end_time,
|
||||
targetname="assets/mp4/clip.mp4",
|
||||
)
|
||||
ffmpeg_extract_subclip(f'assets/backgrounds/{choice}', start_time, end_time,
|
||||
targetname="assets/temp/backgrounds.mp4", )
|
||||
print_substep("Background video chopped successfully! 🎉", style="bold green")
|
||||
|
||||
Loading…
Reference in new issue