Merge branch 'master' of https://github.com/elebumm/RedditVideoMakerBot into elebumm-master
BlockArchitech
3 years ago
commit
44edf4ab31
@ -1,4 +1,10 @@
|
|||||||
REDDIT_CLIENT_ID=""
|
REDDIT_CLIENT_ID=""
|
||||||
REDDIT_CLIENT_SECRET=""
|
REDDIT_CLIENT_SECRET=""
|
||||||
REDDIT_USERNAME=""
|
REDDIT_USERNAME=""
|
||||||
REDDIT_PASSWORD=""
|
REDDIT_PASSWORD=""
|
||||||
|
|
||||||
|
# Valid options are "yes" and "no" for the variable below
|
||||||
|
REDDIT_2FA=""
|
||||||
|
|
||||||
|
|
||||||
|
SUBREDDIT=""
|
||||||
|
|||||||
@ -1,49 +0,0 @@
|
|||||||
from rich import Console
|
|
||||||
from utils.console import print_markdown, print_step, print_substep
|
|
||||||
import praw
|
|
||||||
import random
|
|
||||||
from dotenv import load_dotenv
|
|
||||||
import os
|
|
||||||
console = Console()
|
|
||||||
|
|
||||||
def get_askreddit_threads():
|
|
||||||
"""
|
|
||||||
Returns a list of threads from the AskReddit subreddit.
|
|
||||||
"""
|
|
||||||
|
|
||||||
print_step("Getting AskReddit threads...")
|
|
||||||
|
|
||||||
content = {}
|
|
||||||
load_dotenv()
|
|
||||||
console.log("Logging in to reddit...")
|
|
||||||
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"),
|
|
||||||
)
|
|
||||||
askreddit = reddit.subreddit("askreddit")
|
|
||||||
threads = askreddit.hot(limit=25)
|
|
||||||
submission = list(threads)[random.randrange(0, 25)]
|
|
||||||
print_substep(f"Video will be: {submission.title} :thumbsup:")
|
|
||||||
console.log("Getting video comments...")
|
|
||||||
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 AskReddit threads Successfully.", style="bold green")
|
|
||||||
return content
|
|
||||||
@ -0,0 +1,90 @@
|
|||||||
|
from rich import Console
|
||||||
|
from utils.console import print_markdown, print_step, print_substep
|
||||||
|
import praw
|
||||||
|
import random
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
import os
|
||||||
|
console = Console()
|
||||||
|
|
||||||
|
def get_subreddit_threads():
|
||||||
|
|
||||||
|
"""
|
||||||
|
Returns a list of threads from the AskReddit subreddit.
|
||||||
|
"""
|
||||||
|
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
print_step("Getting AskReddit threads...")
|
||||||
|
|
||||||
|
if os.getenv("REDDIT_2FA").lower() == "yes":
|
||||||
|
print(
|
||||||
|
"\nEnter your two-factor authentication code from your authenticator app.\n"
|
||||||
|
)
|
||||||
|
code = input("> ")
|
||||||
|
print()
|
||||||
|
pw = os.getenv("REDDIT_PASSWORD")
|
||||||
|
passkey = f"{pw}:{code}"
|
||||||
|
else:
|
||||||
|
passkey = os.getenv("REDDIT_PASSWORD")
|
||||||
|
|
||||||
|
content = {}
|
||||||
|
<<<<<<< HEAD:reddit/askreddit.py
|
||||||
|
load_dotenv()
|
||||||
|
console.log("Logging in to reddit...")
|
||||||
|
||||||| 5b39896:reddit/askreddit.py
|
||||||
|
load_dotenv()
|
||||||
|
=======
|
||||||
|
|
||||||
|
>>>>>>> 6fc5d2a7377dfe9f65d0d011fc260602527847c9:reddit/subreddit.py
|
||||||
|
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=passkey,
|
||||||
|
)
|
||||||
|
|
||||||
|
if os.getenv("SUBREDDIT"):
|
||||||
|
subreddit = reddit.subreddit(os.getenv("SUBREDDIT"))
|
||||||
|
else:
|
||||||
|
# ! Prompt the user to enter a subreddit
|
||||||
|
try:
|
||||||
|
subreddit = reddit.subreddit(
|
||||||
|
input("What subreddit would you like to pull from? ")
|
||||||
|
)
|
||||||
|
except ValueError:
|
||||||
|
subreddit = reddit.subreddit("askreddit")
|
||||||
|
print_substep("Subreddit not defined. Using AskReddit.")
|
||||||
|
|
||||||
|
threads = subreddit.hot(limit=25)
|
||||||
|
submission = list(threads)[random.randrange(0, 25)]
|
||||||
|
print_substep(f"Video will be: {submission.title} :thumbsup:")
|
||||||
|
console.log("Getting video comments...")
|
||||||
|
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
|
||||||
|
<<<<<<< HEAD:reddit/askreddit.py
|
||||||
|
print_substep("Received AskReddit threads Successfully.", style="bold green")
|
||||||
|
return content
|
||||||
|
||||||| 5b39896:reddit/askreddit.py
|
||||||
|
print_substep("Received AskReddit threads Successfully.", style="bold green")
|
||||||
|
return content
|
||||||
|
=======
|
||||||
|
print_substep("Received AskReddit threads successfully.", style="bold green")
|
||||||
|
|
||||||
|
return content
|
||||||
|
>>>>>>> 6fc5d2a7377dfe9f65d0d011fc260602527847c9:reddit/subreddit.py
|
||||||
Loading…
Reference in new issue