carom-server/billard/views.py

54 lines
1.8 KiB
Python
Raw Normal View History

2017-01-06 19:11:29 +01:00
from billard.serializers import *
from .models import LocationData
2017-01-03 19:55:01 +01:00
from rest_framework import viewsets
2017-02-04 07:35:17 +01:00
from django.shortcuts import render, redirect
from django.views import generic
2017-01-09 20:42:59 +01:00
from django.views.generic.detail import DetailView
2017-02-03 09:42:23 +01:00
from django.contrib.auth.decorators import login_required
2017-02-04 07:35:17 +01:00
from django.db.models import Min
2017-01-03 19:26:51 +01:00
2017-01-03 19:55:01 +01:00
2017-01-06 20:24:36 +01:00
class LocationDataViewSet(viewsets.ModelViewSet):
queryset = LocationData.objects.all()
serializer_class = LocationDataSerializer
class IndexView(generic.ListView):
model = LocationData
2017-01-09 20:42:59 +01:00
class LocationDataDetailView(DetailView):
model = LocationData
2017-02-01 18:57:29 +01:00
2017-02-03 09:42:23 +01:00
@login_required
2017-02-01 18:57:29 +01:00
def index(request):
2017-02-04 07:35:17 +01:00
if request.method == 'GET':
min_loc = Location.objects.filter(users__id=request.user.id).aggregate(Min('id'))['id__min']
loc = None
if 'loc' in request.GET:
loc = request.GET['loc']
if not Location.objects.filter(users__id=request.user.id).filter(id=loc).exists():
resp = redirect('carom_index')
2017-02-05 12:58:06 +01:00
if min_loc is not None:
resp['Location'] += '?loc={}'.format(str(min_loc))
return resp
else:
return render(request, 'billard/index.html')
2017-02-04 07:35:17 +01:00
if loc is None:
loc = min_loc
locations = Location.objects.filter(users__id=request.user.id).order_by('code')
clients = Client.objects.filter(location_id=loc).order_by('id')
context = {
2017-02-10 19:39:21 +01:00
'range': range(1, 9),
2017-02-04 07:35:17 +01:00
'locations': locations,
'clients': clients,
'location_id': int(loc),
}
return render(request, 'billard/index.html', context=context)
if request.method == 'POST':
2017-02-01 18:57:29 +01:00
loc = request.POST['location-selector']
2017-02-04 07:35:17 +01:00
resp = redirect('carom_index')
resp['Location'] += '?loc={}'.format(str(loc))
return resp