from flask import Flask
from crudest import RestApi, CreateResource, RetrieveResource, UpdateResource, DeleteResource, \
    extra_args, fields, create_access_token, jwt_required

from models import Cat  # e.g. ndb models
from schemas import SessionSchema, CatSchema, CatActionSchema  # marshmallow schemas


app = Flask(__name__)
api = RestApi(app, title='Cat API', version='v1')


@api.resource('/sessions/<int:session_id>', name='Session', schema=SessionSchema)
class SessionResource(CreateResource):

    @extra_args({'username': fields.Str(required=True), 'password': fields.Str(required=True)})
    def create(self, username, password):
        # check username and password here
        return {'access_token': create_access_token(identity=username)}


@api.resource('/cats/<int:cat_id>', name='Cat', schema=CatSchema)
class CatResource(CreateResource, RetrieveResource, UpdateResource, DeleteResource):

    @jwt_required
    def create(self, **kwargs):
        cat = Cat(**kwargs)
        cat.put()
        return cat

    @jwt_required
    def list(self):
        return Cat.query()

    @jwt_required
    def retrieve(self, cat_id):
        # ...

    @jwt_required
    def update(self, cat_id, **kwargs):
        # ...

    @jwt_required
    def delete(self, cat_id):
        # ...


@api.resource('/cats/<int:cat_id>/actions/<int:action_id>', name='CatAction', schema=CatActionSchema)
class CatActionResource(CreateResource):

    @jwt_required
    def create(self, cat_id, **kwargs):
        # ...