You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
150 lines
4.8 KiB
150 lines
4.8 KiB
from django.contrib.auth.models import User, Group |
|
from django.db.models import Max |
|
from rest_framework import viewsets, views, permissions |
|
from rest_framework.response import Response |
|
from rest_auth.registration.views import RegisterView |
|
from fuzzywuzzy import fuzz, process |
|
from collections import OrderedDict |
|
|
|
from . import models, serializers |
|
|
|
class AllowMetadata(permissions.BasePermission): |
|
def has_permission(self, request, view): |
|
return request.method in ['OPTIONS', 'HEAD'] |
|
|
|
|
|
|
|
search_strings = {} |
|
def gen_search_strings(): |
|
import time |
|
start = time.time() |
|
|
|
for m in models.Member.objects.all(): |
|
string = '{} {}'.format( |
|
m.preferred_name, |
|
m.last_name, |
|
).lower() |
|
search_strings[string] = m.id |
|
|
|
print('Generated search strings in {} s'.format(time.time() - start)) |
|
gen_search_strings() |
|
|
|
class SearchViewSet(viewsets.ViewSet): |
|
permission_classes = [AllowMetadata | permissions.IsAuthenticated] |
|
serializer_class = serializers.OtherMemberSerializer |
|
|
|
def get_queryset(self): |
|
NUM_SEARCH_RESULTS = 10 |
|
|
|
queryset = models.Member.objects.all() |
|
search = self.request.data.get('q', '').lower() |
|
|
|
if len(search): |
|
choices = search_strings.keys() |
|
|
|
# get exact starts with matches |
|
results = [x for x in choices if x.startswith(search)] |
|
# then get exact substring matches |
|
results += [x for x in choices if search in x] |
|
|
|
if len(results) == 0 and len(search) >= 3: |
|
# then get fuzzy matches |
|
fuzzy_results = process.extract(search, choices, limit=NUM_SEARCH_RESULTS, scorer=fuzz.token_set_ratio) |
|
results += [x[0] for x in fuzzy_results] |
|
|
|
# remove dupes, truncate list |
|
results = list(OrderedDict.fromkeys(results))[:NUM_SEARCH_RESULTS] |
|
|
|
result_ids = [search_strings[x] for x in results] |
|
result_objects = [queryset.get(id=x) for x in result_ids] |
|
|
|
queryset = result_objects |
|
else: |
|
queryset = queryset.order_by('-vetted_date')[:NUM_SEARCH_RESULTS] |
|
|
|
return queryset |
|
|
|
# must POST so query string doesn't change so preflight request is cached |
|
def create(self, request): |
|
try: |
|
seq = int(request.data.get('seq', 0)) |
|
except ValueError: |
|
seq = 0 |
|
|
|
queryset = self.get_queryset() |
|
serializer = self.serializer_class(queryset, many=True) |
|
return Response({'seq': seq, 'results': serializer.data}) |
|
|
|
|
|
class MemberViewSet(viewsets.ModelViewSet): |
|
permission_classes = [AllowMetadata | permissions.IsAuthenticated] |
|
http_method_names = ['options', 'head', 'get', 'put', 'patch'] |
|
|
|
def get_queryset(self): |
|
objects = models.Member.objects.all() |
|
if self.request.user.is_staff: |
|
return objects.order_by('id') |
|
else: |
|
return objects.filter(user=self.request.user) |
|
|
|
def get_serializer_class(self): |
|
if self.request.user.is_staff: |
|
return serializers.AdminMemberSerializer |
|
else: |
|
return serializers.MemberSerializer |
|
|
|
def update(self, request, *args, **kwargs): |
|
gen_search_strings() |
|
return super().update(request, *args, **kwargs) |
|
def partial_update(self, request, *args, **kwargs): |
|
gen_search_strings() |
|
return super().partial_update(request, *args, **kwargs) |
|
|
|
|
|
class CourseViewSet(viewsets.ModelViewSet): |
|
permission_classes = [AllowMetadata | permissions.IsAuthenticated] |
|
queryset = models.Course.objects.annotate(date=Max('sessions__datetime')).order_by('-date') |
|
|
|
def get_serializer_class(self): |
|
if self.action == 'retrieve': |
|
return serializers.CourseDetailSerializer |
|
else: |
|
return serializers.CourseSerializer |
|
|
|
|
|
class SessionViewSet(viewsets.ModelViewSet): |
|
permission_classes = [AllowMetadata | permissions.IsAuthenticated] |
|
|
|
def get_queryset(self): |
|
if self.action == 'list': |
|
return models.Session.objects.order_by('-datetime')[:20] |
|
else: |
|
return models.Session.objects.all() |
|
|
|
def get_serializer_class(self): |
|
#if self.action == 'retrieve': |
|
# return serializers.CourseDetailSerializer |
|
#else: |
|
return serializers.SessionSerializer |
|
|
|
|
|
class MyUserView(views.APIView): |
|
permission_classes = [AllowMetadata | permissions.IsAuthenticated] |
|
|
|
def get(self, request): |
|
serializer = serializers.UserSerializer(request.user) |
|
return Response(serializer.data) |
|
|
|
|
|
class RegistrationViewSet(RegisterView): |
|
serializer_class = serializers.RegistrationSerializer |
|
|
|
#def create(self, request): |
|
# data = request.data.copy() |
|
# data['username'] = '{}.{}'.format( |
|
# data['first_name'], |
|
# data['last_name'] |
|
# ).lower() |
|
# request._full_data = data |
|
# return super().create(request) |
|
|
|
|