import datetime import uuid from collections import deque from enum import Enum from typing import Dict, List, Set, Union import marshmallow as ma import teal.cache from flask import Response, jsonify, request from marshmallow import Schema as MarshmallowSchema, fields as f from teal.marshmallow import EnumField from teal.resource import View from sqlalchemy.orm import joinedload from ereuse_devicehub.db import db from ereuse_devicehub.query import things_response from ereuse_devicehub.resources.deliverynote.models import DeliveryNote class DeliveryNoteView(View): class FindArgs(MarshmallowSchema): """Allowed arguments for the ``find`` method (GET collection) endpoint """ search = f.Str(missing=None) def post(self): l = request.get_json() dlvnote = DeliveryNote(**l) db.session.add(dlvnote) db.session().final_flush() ret = self.schema.jsonify(dlvnote) ret.status_code = 201 db.session.commit() return ret # def patch(self, id): # patch_schema = self.resource_def.SCHEMA(only=('name', 'description', 'transfer_state', 'receiver_address', 'deposit', 'deliverynote_address', 'devices', 'owner_address'), partial=True) # d = request.get_json(schema=patch_schema) # dlvnote = DeliveryNote.query.filter_by(id=id).one() # device_fields = ['transfer_state', 'receiver_address', 'deposit', 'deliverynote_address', 'owner_address'] # computers = [x for x in dlvnote.all_devices if isinstance(x, Computer)] # for key, value in d.items(): # setattr(dlvnote, key, value) # if key in device_fields: # for dev in computers: # setattr(dev, key, value) # db.session.commit() # return Response(status=204) def one(self, id: uuid.UUID): """Gets one action.""" import pdb; pdb.set_trace() deliverynote = DeliveryNote.query.filter_by(id=id).one() # type: DeliveryNote return self.schema.jsonify(deliverynote) @teal.cache.cache(datetime.timedelta(minutes=5)) def find(self, args: dict): """Gets deliverynotes. By passing the value `UiTree` in the parameter `format` of the query you get a recursive nested suited for ui-tree:: [ {title: 'lot1', nodes: [{title: 'child1', nodes:[]}] ] Note that in this format filters are ignored. Otherwise it just returns the standard flat view of lots that you can filter. """ query = DeliveryNote.query if args['search']: query = query.filter(DeliveryNote.name.ilike(args['search'] + '%')) dlvnote = query.paginate(per_page=6 if args['search'] else 30) return things_response( self.schema.dump(dlvnote.items, many=True, nested=0), dlvnote.page, dlvnote.per_page, dlvnote.total, dlvnote.prev_num, dlvnote.next_num ) def delete(self, id): dlvnote = DeliveryNote.query.filter_by(id=id).one() dlvnote.delete() db.session.commit() return Response(status=204)