devicehub-teal/ereuse_devicehub/resources/allocate/views.py

import uuid
# from typing import Callable, Iterable, Tuple
from flask import g, request
# from flask.json import jsonify
from teal.resource import View

from ereuse_devicehub import auth
from ereuse_devicehub.db import db
from ereuse_devicehub.query import things_response
from ereuse_devicehub.resources.action.models import Allocate, Deallocate


class AllocateView(View):
    @auth.Auth.requires_auth
    def get(self, id: uuid.UUID) -> Allocate:
        return super().get(id)

    @auth.Auth.requires_auth
    def post(self):
        """ Create one allocate """
        res_json = request.get_json()
        allocate = Allocate(**res_json)
        db.session.add(allocate)
        db.session().final_flush()
        ret = self.schema.jsonify(allocate)
        ret.status_code = 201
        db.session.commit()
        return ret

    def find(self, args: dict):
        allocates = Allocate.query.filter_by(author=g.user) \
            .order_by(Allocate.created.desc()) \
            .paginate(per_page=200)
        return things_response(
            self.schema.dump(allocates.items, many=True, nested=0),
            allocates.page, allocates.per_page, allocates.total,
            allocates.prev_num, allocates.next_num
        )

    def one(self, id: uuid.UUID):
        """Gets one action."""
        allocate = Allocate.query.filter_by(id=id, author=g.user).one()
        return self.schema.jsonify(allocate, nested=2)


class DeAllocateView(View):
    @auth.Auth.requires_auth
    def get(self, id: uuid.UUID) -> Allocate:
        return super().get(id)

    @auth.Auth.requires_auth
    def post(self):
        """ Create one Deallocate """
        res_json = request.get_json()
        deallocate = Deallocate(**res_json)
        db.session.add(deallocate)
        db.session().final_flush()
        ret = self.schema.jsonify(deallocate)
        ret.status_code = 201
        db.session.commit()
        return ret

    def find(self, args: dict):
        deallocates = Deallocate.query.filter_by(author=g.user) \
            .order_by(Deallocate.created.desc()) \
            .paginate(per_page=200)
        return things_response(
            self.schema.dump(deallocates.items, many=True, nested=0),
            deallocates.page, deallocates.per_page, deallocates.total,
            deallocates.prev_num, deallocates.next_num
        )

    def one(self, id: uuid.UUID):
        """Gets one action."""
        deallocate = Deallocate.query.filter_by(id=id, author=g.user).one()
        res = self.schema.jsonify(deallocate, nested=2)
        return res