Add notification

judge/comments.py

@@ -18,9 +18,28 @@ from reversion import revisions
 from reversion.models import Revision, Version
 
 from judge.dblock import LockModel
-from judge.models import Comment, CommentLock, CommentVote
+from judge.models import Comment, CommentLock, CommentVote, Notification
 from judge.utils.raw_sql import RawSQLColumn, unique_together_left_join
 from judge.widgets import HeavyPreviewPageDownWidget
+from judge.jinja2.reference import get_user_from_text
+
+
+
+def add_mention_notifications(comment):
+    user_referred = get_user_from_text(comment.body).exclude(id=comment.author.id)
+    for user in user_referred:
+        notification_ref = Notification(owner=user,
+                                        comment=comment,
+                                        category='Mention')
+        notification_ref.save()
+
+def del_mention_notifications(comment):
+    query = {
+        'comment': comment,
+        'category': 'Mention'
+    }
+    Notification.objects.filter(**query).delete()
+
 
 
 class CommentForm(ModelForm):
@@ -87,10 +106,22 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View):
             comment = form.save(commit=False)
             comment.author = request.profile
             comment.page = page
+
+
             with LockModel(write=(Comment, Revision, Version), read=(ContentType,)), revisions.create_revision():
                 revisions.set_user(request.user)
                 revisions.set_comment(_('Posted comment'))
                 comment.save()
+
+            # add notification for reply
+            if comment.parent and comment.parent.author != comment.author:
+                notification_rep = Notification(owner=comment.parent.author,
+                                            comment=comment,
+                                            category='Reply')
+                notification_rep.save()
+
+            add_mention_notifications(comment)
+                        
             return HttpResponseRedirect(request.path)
 
         context = self.get_context_data(object=self.object, comment_form=form)
@@ -118,5 +149,5 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View):
                                       not profile.submission_set.filter(points=F('problem__points')).exists())
         context['comment_list'] = queryset
         context['vote_hide_threshold'] = settings.DMOJ_COMMENT_VOTE_HIDE_THRESHOLD
-
         return context
+

judge/jinja2/reference.py

@@ -56,6 +56,13 @@ def get_user_info(usernames):
                    .values_list('user__username', 'display_rank', 'rating')}
 
 
+def get_user_from_text(text):
+    user_list = set()
+    for i in rereference.finditer(text):
+        user_list.add(text[i.start() + 6: i.end() - 1])
+    return Profile.objects.filter(user__username__in=user_list)
+
+
 reference_map = {
     'user': (get_user, get_user_info),
     'ruser': (get_user_rating, get_user_info),

judge/migrations/0107_notification.py

@@ -0,0 +1,25 @@
+# Generated by Django 2.2.12 on 2020-07-03 01:16
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('judge', '0106_friend'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Notification',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('time', models.DateTimeField(auto_now_add=True, verbose_name='posted time')),
+                ('read', models.BooleanField(default=False, verbose_name='read')),
+                ('category', models.CharField(max_length=10, verbose_name='category')),
+                ('comment', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='judge.Comment', verbose_name='comment')),
+                ('owner', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='notifications', to='judge.Profile', verbose_name='owner')),
+            ],
+        ),
+    ]

judge/models/__init__.py

@@ -1,7 +1,7 @@
 from reversion import revisions
 
 from judge.models.choices import ACE_THEMES, EFFECTIVE_MATH_ENGINES, MATH_ENGINES_CHOICES, TIMEZONE
-from judge.models.comment import Comment, CommentLock, CommentVote
+from judge.models.comment import Comment, CommentLock, CommentVote, Notification
 from judge.models.contest import Contest, ContestMoss, ContestParticipation, ContestProblem, ContestSubmission, \
     ContestTag, Rating
 from judge.models.interface import BlogPost, MiscConfig, NavigationBar, validate_regex

judge/models/comment.py

@@ -19,7 +19,8 @@ from judge.models.problem import Problem
 from judge.models.profile import Profile
 from judge.utils.cachedict import CacheDict
 
-__all__ = ['Comment', 'CommentLock', 'CommentVote']
+
+__all__ = ['Comment', 'CommentLock', 'CommentVote', 'Notification']
 
 comment_validator = RegexValidator(r'^[pcs]:[a-z0-9]+$|^b:\d+$',
                                    _(r'Page code must be ^[pcs]:[a-z0-9]+$|^b:\d+$'))
@@ -183,3 +184,11 @@ class CommentLock(models.Model):
 
     def __str__(self):
         return str(self.page)
+
+
+class Notification(models.Model):
+    owner = models.ForeignKey(Profile, verbose_name=_('owner'), related_name="notifications", on_delete=CASCADE)
+    time = models.DateTimeField(verbose_name=_('posted time'), auto_now_add=True)
+    comment = models.ForeignKey(Comment, verbose_name=_('comment'), on_delete=CASCADE)
+    read = models.BooleanField(verbose_name=_('read'), default=False)
+    category = models.CharField(verbose_name=_('category'), max_length=10)

judge/models/profile.py

@@ -125,6 +125,14 @@ class Profile(models.Model):
     def username(self):
         return self.user.username
 
+    @cached_property
+    def count_unseen_notifications(self):
+        query = {
+            'read': False,
+            'comment__hidden': False,
+        }
+        return self.notifications.filter(**query).count()
+    
     _pp_table = [pow(settings.DMOJ_PP_STEP, i) for i in range(settings.DMOJ_PP_ENTRIES)]
 
     def calculate_points(self, table=_pp_table):

judge/views/comment.py

@@ -13,9 +13,10 @@ from reversion import revisions
 from reversion.models import Version
 
 from judge.dblock import LockModel
-from judge.models import Comment, CommentVote
+from judge.models import Comment, CommentVote, Notification
 from judge.utils.views import TitleMixin
 from judge.widgets import MathJaxPagedownWidget
+from judge.comments import add_mention_notifications, del_mention_notifications
 
 __all__ = ['upvote_comment', 'downvote_comment', 'CommentEditAjax', 'CommentContent',
            'CommentEdit']
@@ -116,6 +117,11 @@ class CommentEditAjax(LoginRequiredMixin, CommentMixin, UpdateView):
     form_class = CommentEditForm
 
     def form_valid(self, form):
+        # update notifications
+        comment = form.instance
+        del_mention_notifications(comment)
+        add_mention_notifications(comment)
+
         with transaction.atomic(), revisions.create_revision():
             revisions.set_comment(_('Edited from site'))
             revisions.set_user(self.request.user)

judge/views/notification.py

@@ -0,0 +1,52 @@
+from django.contrib.auth.decorators import login_required
+from django.views.generic import ListView
+from django.utils.translation import ugettext as _
+from django.utils.timezone import now
+from django.db.models import BooleanField, Value
+
+from judge.utils.cachedict import CacheDict
+from judge.models import Profile, Comment, Notification
+
+__all__ = ['NotificationList']
+
+class NotificationList(ListView):
+    model = Notification
+    context_object_name = 'notifications'
+    template_name = 'notification/list.html'
+    
+    def get_queryset(self):
+        self.unseen_cnt = self.request.profile.count_unseen_notifications
+        
+        query = {
+            'owner': self.request.profile,
+            'comment__hidden': False,
+        }
+        self.queryset = Notification.objects.filter(**query)\
+                           .order_by('-time')[:100] \
+                           .annotate(seen=Value(True, output_field=BooleanField()))
+        
+        # Mark the several first unseen
+        for cnt, q in enumerate(self.queryset):
+            if (cnt < self.unseen_cnt):
+                q.seen = False
+            else:
+                break
+        
+        return self.queryset
+    
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['unseen_count'] = self.unseen_cnt
+        context['title'] = _('Notifications (%d unseen)' % context['unseen_count'])
+        context['has_notifications'] = self.queryset.exists()
+        context['page_titles'] = CacheDict(lambda page: Comment.get_page_title(page))
+        return context
+
+    def get(self, request, *args, **kwargs):
+        ret = super().get(request, *args, **kwargs)
+        
+        # update after rendering
+        Notification.objects.filter(owner=self.request.profile).update(read=True)
+        
+        return ret
+