almet/umap
1
0
Fork 0
mirror of https://github.com/umap-project/umap.git synced 2025-04-28 19:42:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fixup: order users autocomplete by username length

Browse source 
cf #2591
This commit is contained in:
Yohan Boniface 2025-04-08 15:47:06 +02:00
parent f4ae51ae7e
commit 14663bc56e
2 changed files with 12 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
umap/autocomplete.py
View file

@ -2,7 +2,7 @@ from agnocomplete.core import AgnocompleteModel
from agnocomplete.register import register
from django.conf import settings
from django.contrib.auth import get_user_model
from django.db.models import Q
from django.db.models.functions import Length
@register
@ -15,23 +15,10 @@ class AutocompleteUser(AgnocompleteModel):
data["url"] = current_item.get_url()
return data
def get_queryset_filters(self, query):
if len(query) <= 3 and self._force_exact:
conditions = Q()
for field_name in self.fields:
if not field_name[0].isalnum():
field_name = field_name[1:]
# Force exact match
conditions |= Q(**{self._construct_qs_filter(f"={field_name}"): query})
return conditions
return super().get_queryset_filters(query)
def build_filtered_queryset(self, query, **kwargs):
self._force_exact = len(query) <= 3
qs = super().build_filtered_queryset(query, **kwargs)
if len(query) <= 3 and not len(qs):
# Not exact match, let's fallback to normal "startswith" query
self._force_exact = False
qs = super().build_filtered_queryset(query, **kwargs)
return qs
def build_extra_filtered_queryset(self, queryset, **kwargs):
order_by = []
for field_name in self.fields:
if not field_name[0].isalnum():
field_name = field_name[1:]
order_by.append(Length(field_name).asc())
return queryset.order_by(*order_by)

5
umap/tests/test_views.py
View file

@ -516,10 +516,13 @@ def test_can_combine_search_and_filter(client, map):
def test_can_find_small_usernames(client):
UserFactory(username="Joe")
UserFactory(username="JoeJoe")
UserFactory(username="Joe3")
UserFactory(username="Joe57")
UserFactory(username="JoeBar")
url = "/agnocomplete/AutocompleteUser/"
response = client.get(url + "?q=joe")
data = json.loads(response.content)["data"]
assert len(data) == 1
assert len(data) == 5
assert data[0]["label"] == "Joe"
response = client.get(url + "?q=joej")
data = json.loads(response.content)["data"]