greeter.py

from pydantic import BaseModel
from typing import List, Optional

from bali.decorators import action
from bali.resources import Resource
from bali.schemas import GetRequest, ListRequest
from permissions import IsAuthenticated

GREETERS = [{'id': i, 'content': 'Hi, number %s' % i} for i in range(10)]


class Greeter(BaseModel):
    id: int
    content: str


class GreeterFilter(BaseModel):
    ids: List[int]


class GreeterResource(Resource):

    schema = Greeter
    filters = [
        {'name': str},
        {'title': Optional[str]},
    ]
    permission_classes = [IsAuthenticated]

    @action()
    def get(self, pk=None):
        print('greeter pk: %s' % pk)
        return [g for g in GREETERS if g.get('id') == pk][0]

    @action()
    def list(self, schema_in: ListRequest = None):
        # `list` NOT FULL SUPPORT HTTP REQUEST
        # return GREETERS[:schema_in.limit]
        print(schema_in.filters.get('name'))
        return GREETERS

    @action()
    def create(self, schema_in: schema = None):
        return {'id': schema_in.id, 'content': schema_in.content}

    @action()
    def update(self, schema_in: schema = None, pk=None):
        return {'id': pk, 'content': schema_in.content}

    @action()
    def delete(self, pk=None):
        return {'result': True}

    @action(detail=False)
    def recents(self):
        return GREETERS[:2]

    @action(detail=True)
    def items_recents(self, pk=None):
        return [g for g in GREETERS if g.get('id') == pk]

    @action(detail=False, methods=['post'])
    def custom_create(self, schema_in: GreeterFilter):
        print('schema_in', schema_in)
        return GREETERS[0]