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

make the gevent sequential

pull/38/head
Aevann1 2022-12-03 01:25:48 +02:00
parent 9b09181e28
commit 3907a2f25f
1 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
files/helpers/alerts.py
View File

@ -113,23 +113,23 @@ def push_notif(uid, title, body, url):
if len(body) > PUSH_NOTIF_LIMIT:
body = body[:PUSH_NOTIF_LIMIT] + "..."
subscriptions = g.db.query(PushSubscription).filter_by(user_id=uid).all()
subscriptions = g.db.query(PushSubscription.subscription_json).filter_by(user_id=uid).all()
subscriptions = [x[0] for x in subscriptions]
gevent.spawn(_push_notif_thread, subscriptions, title, body, url)
def _push_notif_thread(subscriptions, title, body, url):
for subscription in subscriptions:
gevent.spawn(_push_notif_thread,
subscription.subscription_json, title, body, url)
def _push_notif_thread(sub_json, title, body, url):
try:
response = webpush(
subscription_info=json.loads(sub_json),
data=json.dumps({
"title": title,
"body": body,
'url': url,
'icon': f'{SITE_FULL}/icon.webp?v=1',
}),
vapid_private_key=VAPID_PRIVATE_KEY,
vapid_claims={"sub": f"mailto:{EMAIL}"}
)
except: pass
try:
response = webpush(
subscription_info=json.loads(subscription),
data=json.dumps({
"title": title,
"body": body,
'url': url,
'icon': f'{SITE_FULL}/icon.webp?v=1',
}),
vapid_private_key=VAPID_PRIVATE_KEY,
vapid_claims={"sub": f"mailto:{EMAIL}"}
)
except: continue