mirror of https://github.com/fspc/workstand.git
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.
109 lines
3.7 KiB
109 lines
3.7 KiB
import json
|
|
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.core.urlresolvers import reverse
|
|
from django.http import HttpResponse, HttpResponseRedirect, JsonResponse
|
|
from django.shortcuts import get_object_or_404
|
|
from django.template.response import TemplateResponse
|
|
from django.utils.decorators import method_decorator
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
from django.views.generic import TemplateView, View
|
|
from haystack.query import SearchQuerySet
|
|
from rest_framework import viewsets
|
|
from rest_framework.renderers import JSONRenderer
|
|
from rest_framework.serializers import ModelSerializer
|
|
|
|
from core.models import Visit
|
|
from registration.utils import signin_member, get_signed_in_members
|
|
from .serializers import MemberSerializer
|
|
from .forms import MemberForm
|
|
from .models import Member
|
|
|
|
|
|
@method_decorator(login_required, name='dispatch')
|
|
class MemberFormView(View):
|
|
def get(self, request, member_id=None):
|
|
try:
|
|
member = Member.objects.get(id=member_id)
|
|
form = MemberForm(instance=member)
|
|
except Member.DoesNotExist:
|
|
form = MemberForm()
|
|
member = None
|
|
|
|
context = dict(form=form)
|
|
if member:
|
|
context['member'] = member
|
|
return TemplateResponse(request, 'edit_member_form.html', context=context)
|
|
|
|
return TemplateResponse(request, 'member_form.html', context=context)
|
|
|
|
def post(self, request, member_id=None):
|
|
try:
|
|
member = Member.objects.get(id=member_id)
|
|
form = MemberForm(request.POST, instance=member)
|
|
except Member.DoesNotExist:
|
|
member = None
|
|
form = MemberForm(request.POST)
|
|
|
|
if form.is_valid():
|
|
member_instance = form.save()
|
|
return HttpResponseRedirect(reverse('member_edit', kwargs=dict(member_id=member_instance.id)))
|
|
|
|
context = {'form': form}
|
|
if member:
|
|
context['member'] = member
|
|
return TemplateResponse(request, 'member_form.html', context=context)
|
|
|
|
|
|
class MemberSearchView(View):
|
|
def get(self, request, query):
|
|
sqs = SearchQuerySet().models(Member).autocomplete(text=query)[:5]
|
|
results = [dict(name=result.object.get_full_name(), email=result.object.email, id=result.object.id)
|
|
for result in sqs]
|
|
|
|
data = json.dumps(dict(results=results))
|
|
|
|
return HttpResponse(data, content_type='application/json')
|
|
|
|
|
|
class VisitSerializer(ModelSerializer):
|
|
member = MemberSerializer()
|
|
|
|
class Meta:
|
|
model = Visit
|
|
fields = ('created_at', 'purpose', 'member')
|
|
depth = 1
|
|
|
|
|
|
class MemberSignIn(View):
|
|
@method_decorator(csrf_exempt)
|
|
def dispatch(self, request, *args, **kwargs):
|
|
return super(MemberSignIn, self).dispatch(request, *args, **kwargs)
|
|
|
|
def post(self, request):
|
|
member = get_object_or_404(Member, id=request.POST.get('id'))
|
|
visit = signin_member(member, request.POST.get('purpose'))
|
|
data = json.dumps(dict(results=dict(id=member.id, created_at=visit.created_at.isoformat())))
|
|
|
|
return JsonResponse(data=data, safe=False, status=201)
|
|
|
|
def get(self, request):
|
|
visits = get_signed_in_members().prefetch_related()
|
|
serializer = VisitSerializer(visits, many=True)
|
|
results_json = JSONRenderer().render(serializer.data)
|
|
|
|
return JsonResponse(data=results_json.decode(), safe=False, status=200)
|
|
|
|
|
|
@method_decorator(login_required, name='dispatch')
|
|
class Members(TemplateView):
|
|
template_name = 'members.html'
|
|
|
|
def get(self, request):
|
|
members = Member.objects.all()
|
|
return self.render_to_response(dict(members=members))
|
|
|
|
|
|
class MemberViewSet(viewsets.ModelViewSet):
|
|
queryset = Member.objects.all()
|
|
serializer_class = MemberSerializer
|
|
|