PsychoticNinja/oddluck-limnoria-plugins
1
0
Fork 0
mirror of https://github.com/oddluck/limnoria-plugins.git synced 2025-04-29 23:11:09 -05:00
Code Issues Projects Releases Wiki Activity

Added Praw

Browse Source 
Reddit API for article extracts using PRAW
This commit is contained in:
klapvogn 2024-08-10 19:04:24 +02:00 committed by GitHub
parent 1fc876e94f
commit f214c62cc7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 44 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
SpiffyTitles/plugin.py
View File

@ -67,6 +67,12 @@ class SpiffyTitles(callbacks.Plugin):
self.proxies = {} self.proxies = {}
self.proxies["http"] = None self.proxies["http"] = None
self.proxies["https"] = None self.proxies["https"] = None
# Initialize the Reddit client
self.reddit = praw.Reddit(
client_id=self.registryValue("reddit.clientid"),
client_secret=self.registryValue("reddit.clientsecret"),
user_agent='Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:75.0) Gecko/20100101 Firefox/75.0'
)
proxy = str(conf.supybot.protocols.http.proxy) proxy = str(conf.supybot.protocols.http.proxy)
if proxy: if proxy:
match = re.match(r"https?:\/\/", proxy, re.IGNORECASE) match = re.match(r"https?:\/\/", proxy, re.IGNORECASE)
@ -1450,11 +1456,12 @@ class SpiffyTitles(callbacks.Plugin):
def handler_reddit(self, url, domain, channel): def handler_reddit(self, url, domain, channel):
""" """
Queries wikipedia API for article extracts. Queries Reddit API for article extracts using PRAW.
""" """
reddit_handler_enabled = self.registryValue("reddit.enabled", channel=channel) reddit_handler_enabled = self.registryValue("reddit.enabled", channel=channel)
if not reddit_handler_enabled: if not reddit_handler_enabled:
return self.handler_default(url, channel) return self.handler_default(url, channel)
self.log.debug("SpiffyTitles: calling reddit handler for %s" % (url)) self.log.debug("SpiffyTitles: calling reddit handler for %s" % (url))
patterns = { patterns = {
"thread": { "thread": {
@ -1476,50 +1483,53 @@ class SpiffyTitles(callbacks.Plugin):
}, },
} }
info = urlparse(url) info = urlparse(url)
link_type = None
link_info = {}
for name in patterns: for name in patterns:
match = re.search(patterns[name]["pattern"], info.path) match = re.search(patterns[name]["pattern"], info.path)
if match: if match:
link_type = name link_type = name
link_info = match.groupdict() link_info = match.groupdict()
data_url = patterns[name]["url"].format(**link_info)
break break
if not match: if not match:
self.log.debug("SpiffyTitles: no title found.") self.log.debug("SpiffyTitles: no title found.")
return self.handler_default(url, channel) return self.handler_default(url, channel)
self.log.debug("SpiffyTitles: requesting %s" % (data_url))
headers = {"User-Agent": self.get_user_agent()} self.log.debug("SpiffyTitles: requesting %s" % (url))
try:
request = requests.get(
data_url, headers=headers, timeout=self.timeout, proxies=self.proxies
)
request.raise_for_status()
except (
requests.exceptions.RequestException,
requests.exceptions.HTTPError,
) as e:
log.error("SpiffyTitles: Reddit Error: {0}".format(e))
return self.handler_default(url, channel)
data = {}
extract = "" extract = ""
response = json.loads(request.content.decode()) try:
if response: if link_type == "thread":
try: submission = self.reddit.submission(id=link_info["thread"])
if link_type == "thread": created_utc = getattr(submission, "created_utc", None)
data = response[0]["data"]["children"][0]["data"] if created_utc is None:
if link_type == "comment": self.log.error("SpiffyTitles: 'created_utc' not found in submission object")
data = response[1]["data"]["children"][0]["data"] return self.handler_default(url, channel)
data["title"] = response[0]["data"]["children"][0]["data"]["title"] data = vars(submission)
if link_type == "user": elif link_type == "comment":
data = response["data"] comment = self.reddit.comment(id=link_info["comment"])
except KeyError as e: created_utc = getattr(comment, "created_utc", None)
self.log.error( if created_utc is None:
"SpiffyTitles: KeyError parsing Reddit JSON response: %s" % (str(e)) self.log.error("SpiffyTitles: 'created_utc' not found in comment object")
) return self.handler_default(url, channel)
else: data = vars(comment)
self.log.error("SpiffyTitles: Error parsing Reddit JSON response") submission = comment.submission
data["title"] = submission.title
elif link_type == "user":
user = self.reddit.redditor(link_info["user"])
data = vars(user)
else:
return self.handler_default(url, channel)
except KeyError as e:
self.log.error(f"SpiffyTitles: KeyError - {e}")
return self.handler_default(url, channel)
except Exception as e:
self.log.error(f"SpiffyTitles: Reddit Error: {e}")
return self.handler_default(url, channel)
if data: if data:
today = datetime.datetime.now().date() today = datetime.datetime.now().date()
created = datetime.datetime.fromtimestamp(data["created_utc"]).date() created = datetime.datetime.fromtimestamp(data.get("created_utc", 0)).date()
age_days = (today - created).days age_days = (today - created).days
if age_days == 0: if age_days == 0:
age = "today" age = "today"
@ -1532,7 +1542,7 @@ class SpiffyTitles(callbacks.Plugin):
age = age + " ago" age = age + " ago"
if link_type == "thread": if link_type == "thread":
link_type = "linkThread" link_type = "linkThread"
if data["is_self"]: if data.get("is_self"):
link_type = "textThread" link_type = "textThread"
data["url"] = "" data["url"] = ""
extract = data.get("selftext", "") extract = data.get("selftext", "")