1 ##############################################################################
2 # Copyright (c) 2016 Max Breitenfeldt and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
11 from django.contrib.auth.decorators import login_required
12 from django.shortcuts import redirect
13 from django.utils.decorators import method_decorator
14 from django.views import View
15 from rest_framework import viewsets
16 from rest_framework.authtoken.models import Token
18 from api.serializers import *
19 from booking.models import Booking
20 from dashboard.models import Resource, Server, ResourceStatus
23 class BookingViewSet(viewsets.ModelViewSet):
24 queryset = Booking.objects.all()
25 serializer_class = BookingSerializer
26 filter_fields = ('resource', 'id')
29 class ServerViewSet(viewsets.ModelViewSet):
30 queryset = Server.objects.all()
31 serializer_class = ServerSerializer
32 filter_fields = ('resource', 'name')
35 class ResourceViewSet(viewsets.ModelViewSet):
36 queryset = Resource.objects.all()
37 serializer_class = ResourceSerializer
38 filter_fields = ('name', 'id')
40 class ResourceStatusViewSet(viewsets.ModelViewSet):
41 queryset = ResourceStatus.objects.all()
42 serializer_class = ResourceStatusSerializer
45 @method_decorator(login_required, name='dispatch')
46 class GenerateTokenView(View):
47 def get(self, request, *args, **kwargs):
48 user = self.request.user
49 token, created = Token.objects.get_or_create(user=user)
52 Token.objects.create(user=user)
53 return redirect('account:settings')