Fix chat user search

judge/views/select2.py

@@ -102,6 +102,21 @@ class UserSearchSelect2View(BaseListView):
     def get_queryset(self):
         return _get_user_queryset(self.term)
 
+    def get_json_result_from_object(self, user_tuple):
+        pk, username, email, display_rank, profile_image = user_tuple
+        return {
+            "text": username,
+            "id": username,
+            "gravatar_url": gravatar(
+                None,
+                self.gravatar_size,
+                self.gravatar_default,
+                self.get_profile_image_url(profile_image),
+                email,
+            ),
+            "display_rank": display_rank,
+        }
+
     def get(self, request, *args, **kwargs):
         self.request = request
         self.kwargs = kwargs
@@ -118,23 +133,8 @@ class UserSearchSelect2View(BaseListView):
         return JsonResponse(
             {
                 "results": [
-                    {
-                        "text": username,
-                        "id": username,
-                        "gravatar_url": gravatar(
-                            None,
-                            self.gravatar_size,
-                            self.gravatar_default,
-                            urljoin(settings.MEDIA_URL, profile_image)
-                            if profile_image
-                            else None,
-                            email,
-                        ),
-                        "display_rank": display_rank,
-                    }
-                    for pk, username, email, display_rank, profile_image in context[
-                        "object_list"
-                    ]
+                    self.get_json_result_from_object(user_tuple)
+                    for user_tuple in context["object_list"]
                 ],
                 "more": context["page_obj"].has_next(),
             }
@@ -143,6 +143,11 @@ class UserSearchSelect2View(BaseListView):
     def get_name(self, obj):
         return str(obj)
 
+    def get_profile_image_url(self, profile_image):
+        if profile_image:
+            return urljoin(settings.MEDIA_URL, profile_image)
+        return None
+
 
 class ContestUserSearchSelect2View(UserSearchSelect2View):
     def get_queryset(self):
@@ -169,3 +174,22 @@ class AssigneeSelect2View(UserSearchSelect2View):
         return Profile.objects.filter(
             assigned_tickets__isnull=False, user__username__icontains=self.term
         ).distinct()
+
+
+class ChatUserSearchSelect2View(UserSearchSelect2View):
+    def get_json_result_from_object(self, user_tuple):
+        if not self.request.user.is_authenticated:
+            raise Http404()
+        pk, username, email, display_rank, profile_image = user_tuple
+        return {
+            "text": username,
+            "id": encrypt_url(self.request.profile.id, pk),
+            "gravatar_url": gravatar(
+                None,
+                self.gravatar_size,
+                self.gravatar_default,
+                self.get_profile_image_url(profile_image),
+                email,
+            ),
+            "display_rank": display_rank,
+        }