BlockArchitech
3 years ago
commit
9c3e932d56
@ -1,4 +1,10 @@
|
||||
REDDIT_CLIENT_ID=""
|
||||
REDDIT_CLIENT_SECRET=""
|
||||
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