python-tdd/src/apps/api/views.py

from django.shortcuts import get_object_or_404
from rest_framework.views import APIView
from rest_framework.response import Response

from apps.api.serializers import LineSerializer, PostSerializer, UserSerializer
from apps.dashboard.models import Line, Post
from apps.lyric.models import User


class PostDetailAPI(APIView):
    def get(self, request, post_id):
        post = get_object_or_404(Post, id=post_id)
        serializer = PostSerializer(post)
        return Response(serializer.data)

class PostLinesAPI(APIView):
    def post(self, request, post_id):
        post = get_object_or_404(Post, id=post_id)
        serializer = LineSerializer(data=request.data, context={"post": post})
        if serializer.is_valid():
            serializer.save(post=post)
            return Response(serializer.data, status=201)
        return Response(serializer.errors, status=400)

class PostsAPI(APIView):
    def get(self, request):
        posts = Post.objects.filter(owner=request.user)
        serializer = PostSerializer(posts, many=True)
        return Response(serializer.data)

    def post(self, request):
        post = Post.objects.create(owner=request.user)
        line = Line.objects.create(text=request.data.get("text", ""), post=post)
        serializer = PostSerializer(post)
        return Response(serializer.data, status=201)

class UserSearchAPI(APIView):
    def get(self, request):
        q = request.query_params.get("q", "")
        users = User.objects.filter(
            username__icontains=q,
            searchable=True,
        )
        serializer = UserSerializer(users, many=True)
        return Response(serializer.data)