forked from MarseyWorld/MarseyWorld
125 lines
4.1 KiB
Python
125 lines
4.1 KiB
Python
import time
|
|
from typing import Iterable
|
|
import itertools
|
|
|
|
import requests
|
|
from flask_caching import Cache
|
|
from flask import g
|
|
from sqlalchemy import or_
|
|
|
|
import files.helpers.const as const
|
|
from files.classes.badges import Badge
|
|
from files.classes.comment import Comment
|
|
from files.classes.notifications import Notification
|
|
from files.classes.user import User
|
|
from files.helpers.sanitize import sanitize
|
|
|
|
# Note: while https://api.pushshift.io/meta provides the key
|
|
# server_ratelimit_per_minute, in practice Cloudflare puts stricter,
|
|
# unofficially documented limits at around 60/minute. We get nowhere near this
|
|
# with current keyword quantities. If this ever changes, consider reading the
|
|
# value from /meta (or just guessing) and doing a random selection of keywords.
|
|
|
|
def offsite_mentions_task(cache:Cache):
|
|
if const.REDDIT_NOTIFS_SITE:
|
|
row_send_to = g.db.query(Badge.user_id).filter_by(badge_id=140).all()
|
|
row_send_to += g.db.query(User.id).filter(
|
|
User.admin_level >= const.PERMS['NOTIFICATIONS_REDDIT'],
|
|
User.id != const.AEVANN_ID,
|
|
).all()
|
|
|
|
send_to = [x[0] for x in row_send_to]
|
|
send_to = set(send_to)
|
|
|
|
site_mentions = get_mentions(cache, const.REDDIT_NOTIFS_SITE)
|
|
notify_mentions(send_to, site_mentions)
|
|
|
|
if const.REDDIT_NOTIFS_USERS:
|
|
for query, send_user in const.REDDIT_NOTIFS_USERS.items():
|
|
user_mentions = get_mentions(cache, [query], reddit_notifs_users=True)
|
|
notify_mentions([send_user], user_mentions, mention_str='mention of you')
|
|
|
|
def get_mentions(cache:Cache, queries:Iterable[str], reddit_notifs_users=False):
|
|
kinds = ['submission', 'comment']
|
|
mentions = []
|
|
exclude_subreddits = ['PokemonGoRaids', 'SubSimulatorGPT2', 'SubSimGPT2Interactive']
|
|
try:
|
|
after = int(cache.get(const.REDDIT_NOTIFS_CACHE_KEY) or time.time())
|
|
except:
|
|
print("Failed to retrieve last mention time from cache")
|
|
after = time.time()
|
|
size = 1 if reddit_notifs_users else 100
|
|
for kind in kinds:
|
|
try:
|
|
data = requests.get((
|
|
f'https://api.pushshift.io/reddit/{kind}/search?html_decode=true'
|
|
f'&q={"%7C".join(queries)}'
|
|
f'&subreddit=!{",!".join(exclude_subreddits)}'
|
|
f'&after={after}'
|
|
f'&size={size}'), timeout=15).json()['data']
|
|
except:
|
|
continue
|
|
|
|
for thing in data:
|
|
if 'bot' in thing['author'].lower(): continue
|
|
after = max(after, thing["created_utc"]) if thing["created_utc"] else after
|
|
if kind == 'comment':
|
|
body = thing["body"].replace('>', '> ')
|
|
text = f'<blockquote><p>{body}</p></blockquote>'
|
|
else:
|
|
title = thing["title"].replace('>', '> ')
|
|
|
|
# Special case: a spambot says 'WPD' a lot unrelated to us.
|
|
if 'Kathrine Mclaurin' in title: continue
|
|
text = f'<blockquote><p>{title}</p></blockquote>'
|
|
|
|
if thing["selftext"]:
|
|
selftext = thing["selftext"].replace('>', '> ')[:5000]
|
|
text += f'<br><blockquote><p>{selftext}</p></blockquote>'
|
|
|
|
|
|
mentions.append({
|
|
'permalink': thing['permalink'],
|
|
'author': thing['author'],
|
|
'text': text,
|
|
})
|
|
try:
|
|
if not reddit_notifs_users:
|
|
cache.set(CACHE_KEY, after + 1)
|
|
except:
|
|
print("Failed to set cache value; there may be duplication of reddit notifications")
|
|
return mentions
|
|
|
|
def notify_mentions(send_to, mentions, mention_str='site mention'):
|
|
for m in mentions:
|
|
author = m['author']
|
|
permalink = m['permalink']
|
|
text = sanitize(m['text'], golden=False)
|
|
notif_text = (
|
|
f'<p>New {mention_str} by <a href="https://old.reddit.com/u/{author}" '
|
|
f'rel="nofollow noopener" target="_blank">/u/{author}</a></p>'
|
|
f'<p><a href="https://old.reddit.com{permalink}?context=89" '
|
|
'rel="nofollow noopener" target="_blank">'
|
|
f'https://old.reddit.com{permalink}?context=89</a></p>'
|
|
f'{text}'
|
|
)
|
|
|
|
existing_comment = g.db.query(Comment.id).filter_by(
|
|
author_id=const.AUTOJANNY_ID,
|
|
parent_submission=None,
|
|
body_html=notif_text).one_or_none()
|
|
if existing_comment: break
|
|
|
|
new_comment = Comment(
|
|
author_id=const.AUTOJANNY_ID,
|
|
parent_submission=None,
|
|
body_html=notif_text,
|
|
distinguish_level=6)
|
|
g.db.add(new_comment)
|
|
g.db.flush()
|
|
new_comment.top_comment_id = new_comment.id
|
|
|
|
for user_id in send_to:
|
|
notif = Notification(comment_id=new_comment.id, user_id=user_id)
|
|
g.db.add(notif)
|