78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
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
|