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

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
from ereuse_devicehub.resources.lot.models import Lot
from ereuse_devicehub.resources.device.models import Computer


class DeliverynoteView(View):

    def post(self):
        # Create delivery note
        dn = request.get_json()
        dlvnote = Deliverynote(**dn)
        # Create a lot
        lot_name = dlvnote.supplier_email + "_" + datetime.datetime.utcnow().strftime("%B-%d-%Y")
        new_lot = Lot(name=lot_name)
        dlvnote.lot_id = new_lot.id
        db.session.add(new_lot)
        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=('transfer_state',
                                                      'ethereum_address'), partial=True)
        d = request.get_json(schema=patch_schema)
        dlvnote = Deliverynote.query.filter_by(id=id).one()
        device_fields = ['transfer_state',  'deliverynote_address']
        computers = [x for x in dlvnote.transferred_devices if isinstance(x, Computer)]
        for key, value in d.items():
            # Transalate ethereum_address attribute
            devKey = key
            if key == 'ethereum_address':
                devKey = 'deliverynote_address'
            setattr(dlvnote, key, value)
            if devKey in device_fields:
                for dev in computers:
                    setattr(dev, devKey, value)
         
        db.session.commit()
        return Response(status=204)

    def one(self, id: uuid.UUID):
        """Get one delivery note"""
        deliverynote = Deliverynote.query.filter_by(id=id).one()  # type Deliverynote
        return self.schema.jsonify(deliverynote)