Add friend

judge/migrations/0106_friend.py

@@ -0,0 +1,22 @@
+# Generated by Django 2.2.12 on 2020-06-23 03:15
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('judge', '0105_auto_20200523_0756'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Friend',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('current_user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='following_users', to='judge.Profile')),
+                ('users', models.ManyToManyField(to='judge.Profile')),
+            ],
+        ),
+    ]

judge/models/__init__.py

@@ -10,7 +10,7 @@ from judge.models.problem import LanguageLimit, License, Problem, ProblemClarifi
     ProblemTranslation, ProblemType, Solution, TranslatedProblemForeignKeyQuerySet, TranslatedProblemQuerySet
 from judge.models.problem_data import CHECKERS, ProblemData, ProblemTestCase, problem_data_storage, \
     problem_directory_file
-from judge.models.profile import Organization, OrganizationRequest, Profile
+from judge.models.profile import Organization, OrganizationRequest, Profile, Friend
 from judge.models.runtime import Judge, Language, RuntimeVersion
 from judge.models.submission import SUBMISSION_RESULT, Submission, SubmissionSource, SubmissionTestCase
 from judge.models.ticket import Ticket, TicketMessage

judge/models/profile.py

@@ -4,7 +4,7 @@ from django.conf import settings
 from django.contrib.auth.models import User
 from django.core.validators import RegexValidator
 from django.db import models
-from django.db.models import Max
+from django.db.models import Max, CASCADE
 from django.urls import reverse
 from django.utils.functional import cached_property
 from django.utils.timezone import now
@@ -16,7 +16,7 @@ from judge.models.choices import ACE_THEMES, MATH_ENGINES_CHOICES, TIMEZONE
 from judge.models.runtime import Language
 from judge.ratings import rating_class
 
-__all__ = ['Organization', 'Profile', 'OrganizationRequest']
+__all__ = ['Organization', 'Profile', 'OrganizationRequest', 'Friend']
 
 
 class EncryptedNullCharField(EncryptedCharField):
@@ -178,6 +178,16 @@ class Profile(models.Model):
     def css_class(self):
         return self.get_user_css_class(self.display_rank, self.rating)
 
+    def get_friends(self): #list of usernames, including you
+        friend_obj = self.following_users.all()
+        ret = set()
+
+        if (friend_obj):
+            ret = set(friend.username for friend in friend_obj[0].users.all())
+        
+        ret.add(self.username)
+        return ret
+
     class Meta:
         permissions = (
             ('test_site', 'Shows in-progress development stuff'),
@@ -202,3 +212,40 @@ class OrganizationRequest(models.Model):
     class Meta:
         verbose_name = _('organization join request')
         verbose_name_plural = _('organization join requests')
+
+
+class Friend(models.Model):
+    users = models.ManyToManyField(Profile)
+    current_user = models.ForeignKey(Profile, related_name="following_users", on_delete=CASCADE)
+
+    @classmethod
+    def is_friend(self, current_user, new_friend):
+        try:
+            return current_user.following_users.get().users \
+                        .filter(user=new_friend.user).exists()
+        except:
+            return False
+
+    @classmethod
+    def make_friend(self, current_user, new_friend):
+        friend, created = self.objects.get_or_create(
+            current_user = current_user
+        )
+        friend.users.add(new_friend)
+
+    @classmethod
+    def remove_friend(self, current_user, new_friend):
+        friend, created = self.objects.get_or_create(
+            current_user = current_user
+        )
+        friend.users.remove(new_friend)
+
+    @classmethod
+    def toggle_friend(self, current_user, new_friend):
+        if (self.is_friend(current_user, new_friend)):
+            self.remove_friend(current_user, new_friend)
+        else:
+            self.make_friend(current_user, new_friend)
+
+    def __str__(self):
+        return str(self.current_user)

judge/views/contests.py

@@ -602,7 +602,7 @@ def get_contest_ranking_list(request, contest, participation=None, ranking_list=
                 problems)
 
     users = ranker(ranking_list(contest, problems), key=attrgetter('points', 'cumtime'))
-
+    
     if show_current_virtual:
         if participation is None and request.user.is_authenticated:
             participation = request.profile.current_contest

judge/views/user.py

@@ -23,7 +23,7 @@ from django.views.generic import DetailView, ListView, TemplateView
 from reversion import revisions
 
 from judge.forms import ProfileForm, newsletter_id
-from judge.models import Profile, Rating, Submission
+from judge.models import Profile, Rating, Submission, Friend
 from judge.performance_points import get_pp_breakdown
 from judge.ratings import rating_class, rating_progress
 from judge.utils.problems import contest_completed_ids, user_completed_ids
@@ -92,9 +92,11 @@ class UserPage(TitleMixin, UserMixin, DetailView):
     def get_context_data(self, **kwargs):
         context = super(UserPage, self).get_context_data(**kwargs)
 
+        context['followed'] = Friend.is_friend(self.request.profile, self.object)
         context['hide_solved'] = int(self.hide_solved)
         context['authored'] = self.object.authored_problems.filter(is_public=True, is_organization_private=False) \
                                   .order_by('code')
+
         rating = self.object.ratings.order_by('-contest__end_time')[:1]
         context['rating'] = rating[0] if rating else None
 
@@ -149,6 +151,19 @@ class UserAboutPage(UserPage):
             context['min_graph'] = min_user + ratio * delta - delta
         return context
 
+    # follow/unfollow user
+    def post(self, request, user, *args, **kwargs):
+        try:
+            if not request.profile:
+                raise Exception('You have to login')
+            if (request.profile.username == user):
+                raise Exception('Cannot make friend with yourself')
+            
+            following_profile = Profile.objects.get(user__username=user)
+            Friend.toggle_friend(request.profile, following_profile)
+        finally:
+            return HttpResponseRedirect(request.path_info)
+
 
 class UserProblemsPage(UserPage):
     template_name = 'user/user-problems.html'
@@ -269,9 +284,14 @@ class UserList(QueryStringSortMixin, DiggPaginatorMixin, TitleMixin, ListView):
     default_sort = '-performance_points'
 
     def get_queryset(self):
-        return (Profile.objects.filter(is_unlisted=False).order_by(self.order, 'id').select_related('user')
+        ret = Profile.objects.filter(is_unlisted=False).order_by(self.order, 'id').select_related('user') \
                 .only('display_rank', 'user__username', 'points', 'rating', 'performance_points',
-                      'problem_count'))
+                      'problem_count')
+
+        if (self.request.GET.get('friend') == 'true'):
+            friends = list(self.request.profile.get_friends())
+            ret = ret.filter(user__username__in=friends)
+        return ret
 
     def get_context_data(self, **kwargs):
         context = super(UserList, self).get_context_data(**kwargs)