24 lines
710 B
Python
24 lines
710 B
Python
from django.contrib.auth.models import User
|
|
from rest_framework import viewsets
|
|
from rest_framework.generics import RetrieveUpdateAPIView
|
|
|
|
from gym.models import Room, Equipment
|
|
from gym.serializers import RoomSerializer, EquipmentSerializer, UserSerializer
|
|
|
|
|
|
class MeView(RetrieveUpdateAPIView):
|
|
serializer_class = UserSerializer
|
|
queryset = User.objects.none()
|
|
|
|
def get_object(self):
|
|
return self.request.user
|
|
|
|
|
|
class RoomViewSet(viewsets.ModelViewSet):
|
|
queryset = Room.objects.all().order_by('name')
|
|
serializer_class = RoomSerializer
|
|
|
|
|
|
class EquipmentViewSet(viewsets.ModelViewSet):
|
|
queryset = Equipment.objects.all().order_by('name')
|
|
serializer_class = EquipmentSerializer |