This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
devicehub-teal/ereuse_devicehub/resources/allocate/views.py

75 lines
2.4 KiB
Python
Raw Normal View History

2020-11-18 17:13:13 +00:00
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
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 rent """
res_json = request.get_json()
assigned = Allocate(**res_json)
db.session.add(assigned)
db.session().final_flush()
ret = self.schema.jsonify(assigned)
ret.status_code = 201
db.session.commit()
return ret
def find(self, args: dict):
rents = Allocate.query.filter_by(author=g.user) \
.order_by(Allocate.created.desc()) \
.paginate(per_page=200)
return things_response(
self.schema.dump(rents.items, many=True, nested=0),
rents.page, rents.per_page, rents.total, rents.prev_num, rents.next_num
)
def one(self, id: uuid.UUID):
"""Gets one action."""
assigned = Allocate.query.filter_by(id=id, author=g.user).one()
return self.schema.jsonify(assigned, 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 rent """
res_json = request.get_json()
assigned = Allocate(**res_json)
db.session.add(assigned)
db.session().final_flush()
ret = self.schema.jsonify(assigned)
ret.status_code = 201
db.session.commit()
return ret
def find(self, args: dict):
rents = Allocate.query.filter_by(author=g.user) \
.order_by(Allocate.created.desc()) \
.paginate(per_page=200)
return things_response(
self.schema.dump(rents.items, many=True, nested=0),
rents.page, rents.per_page, rents.total, rents.prev_num, rents.next_num
)
def one(self, id: uuid.UUID):
"""Gets one action."""
assigned = Allocate.query.filter_by(id=id, author=g.user).one()
return self.schema.jsonify(assigned, nested=2)