rDrama
/
rDrama
9
10
Fork 7
Code Issues Pull Requests Packages Releases Wiki Activity

Minor alerts.py code restyle.

pull/37/head
Snakes 2022-12-02 17:16:26 -05:00
parent c6f74f5296
commit de77a2c176
Signed by: Snakes
GPG Key ID: E745A82778055C7E
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
files/helpers/alerts.py
View File

@ -1,6 +1,8 @@
import json
from sys import stdout
from flask import g
from pywebpush import webpush
from files.classes import Comment, Notification, PushSubscription
@ -102,16 +104,18 @@ def NOTIFY_USERS(text, v):
return notify_users - bots
if VAPID_PUBLIC_KEY != DEFAULT_CONFIG_VALUE:
from pywebpush import webpush
import json
claims = {"sub": f"mailto:{EMAIL}"}
def push_notif(uid, title, body, url):
if VAPID_PUBLIC_KEY == DEFAULT_CONFIG_VALUE:
return
def push_notif(uid, title, body, url):
subscriptions = g.db.query(PushSubscription).filter_by(user_id=uid).all()
for subscription in subscriptions:
try: response = webpush(
if len(body) > PUSH_NOTIF_LIMIT:
body = body[:PUSH_NOTIF_LIMIT] + "..."
subscriptions = g.db.query(PushSubscription).filter_by(user_id=uid).all()
for subscription in subscriptions:
try:
response = webpush(
subscription_info=json.loads(subscription.subscription_json),
data=json.dumps({
"title": title,
@ -120,6 +124,6 @@ if VAPID_PUBLIC_KEY != DEFAULT_CONFIG_VALUE:
'icon': f'{SITE_FULL}/icon.webp?v=1',
}),
vapid_private_key=VAPID_PRIVATE_KEY,
vapid_claims=claims
vapid_claims={"sub": f"mailto:{EMAIL}"}
)
except: continue
except: continue