forked from mirrors/bookwyrm
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
|
''' search views'''
|
||
|
|
import re
|
||
|
|
|
||
|
|
from django.contrib.postgres.search import TrigramSimilarity
|
||
|
|
from django.db.models.functions import Greatest
|
||
|
|
from django.http import JsonResponse
|
||
|
|
from django.template.response import TemplateResponse
|
||
|
|
from django.views import View
|
||
|
|
|
||
|
|
from bookwyrm import models
|
||
|
|
from bookwyrm.connectors import connector_manager
|
||
|
|
from bookwyrm.utils import regex
|
||
|
|
from .helpers import is_api_request
|
||
|
|
from .helpers import handle_remote_webfinger
|
||
|
|
|
||
|
|
|
||
|
|
# pylint: disable= no-self-use
|
||
|
|
class Search(View):
|
||
|
|
''' search users or books '''
|
||
|
|
def get(self, request):
|
||
|
|
''' that search bar up top '''
|
||
|
|
query = request.GET.get('q')
|
||
|
|
min_confidence = request.GET.get('min_confidence', 0.1)
|
||
|
|
|
||
|
|
if is_api_request(request):
|
||
|
|
# only return local book results via json so we don't cascade
|
||
|
|
book_results = connector_manager.local_search(
|
||
|
|
query, min_confidence=min_confidence)
|
||
|
|
return JsonResponse([r.json() for r in book_results], safe=False)
|
||
|
|
|
||
|
|
# use webfinger for mastodon style account@domain.com username
|
||
|
|
if re.match(r'\B%s' % regex.full_username, query):
|
||
|
|
handle_remote_webfinger(query)
|
||
|
|
|
||
|
|
# do a local user search
|
||
|
|
user_results = models.User.objects.annotate(
|
||
|
|
similarity=Greatest(
|
||
|
|
TrigramSimilarity('username', query),
|
||
|
|
TrigramSimilarity('localname', query),
|
||
|
|
)
|
||
|
|
).filter(
|
||
|
|
similarity__gt=0.5,
|
||
|
|
).order_by('-similarity')[:10]
|
||
|
|
|
||
|
|
book_results = connector_manager.search(
|
||
|
|
query, min_confidence=min_confidence)
|
||
|
|
data = {
|
||
|
|
'title': 'Search Results',
|
||
|
|
'book_results': book_results,
|
||
|
|
'user_results': user_results,
|
||
|
|
'query': query,
|
||
|
|
}
|
||
|
|
return TemplateResponse(request, 'search_results.html', data)
|