michael/pyfedi
1
0
Fork 0
mirror of https://codeberg.org/rimu/pyfedi synced 2025-01-24 03:43:42 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions

notify about new posts in communities

Browse source
This commit is contained in:
rimu 2024-01-07 12:49:25 +13:00
parent 2bc1ab47d0
commit 8da67fec30
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
app/activitypub/util.py
View file

@ -7,9 +7,8 @@ from typing import Union, Tuple
from flask import current_app, request, g, url_for from flask import current_app, request, g, url_for
from sqlalchemy import text from sqlalchemy import text
from app import db, cache, constants, celery from app import db, cache, constants, celery
from app.community.util import notify_about_post
from app.models import User, Post, Community, BannedInstances, File, PostReply, AllowedInstances, Instance, utcnow, \ from app.models import User, Post, Community, BannedInstances, File, PostReply, AllowedInstances, Instance, utcnow, \
PostVote, PostReplyVote, ActivityPubLog, Notification, Site PostVote, PostReplyVote, ActivityPubLog, Notification, Site, CommunityMember
import time import time
import base64 import base64
import requests import requests
@ -1056,6 +1055,17 @@ def create_post(activity_log: ActivityPubLog, community: Community, request_json
return post return post
def notify_about_post(post: Post):
people_to_notify = CommunityMember.query.filter_by(community_id=post.community_id, notify_new_posts=True, is_banned=False)
for person in people_to_notify:
if person.user_id != post.user_id:
new_notification = Notification(title=shorten_string(post.title, 25), url=f"/post/{post.id}", user_id=person.user_id, author_id=post.user_id)
db.session.add(new_notification)
user = User.query.get(person.user_id) # todo: make this more efficient by doing a join with CommunityMember at the start of the function
user.unread_notifications += 1
db.session.commit()
def update_post_reply_from_activity(reply: PostReply, request_json: dict): def update_post_reply_from_activity(reply: PostReply, request_json: dict):
if 'source' in request_json['object'] and request_json['object']['source']['mediaType'] == 'text/markdown': if 'source' in request_json['object'] and request_json['object']['source']['mediaType'] == 'text/markdown':
reply.body = request_json['object']['source']['content'] reply.body = request_json['object']['source']['content']