import uuid from datetime import datetime from typing import Iterable from boltons import urlutils from citext import CIText from flask import g from sqlalchemy.dialects.postgresql import JSONB, UUID from ereuse_devicehub.db import db from ereuse_devicehub.resources.enums import TransferState from ereuse_devicehub.resources.lot.models import Lot from ereuse_devicehub.resources.models import Thing from ereuse_devicehub.resources.user.models import User from ereuse_devicehub.teal.db import IntEnum, check_range from ereuse_devicehub.teal.resource import url_for_resource class Deliverynote(Thing): id = db.Column( UUID(as_uuid=True), primary_key=True ) # uuid is generated on init by default document_id = db.Column(CIText(), nullable=False) creator_id = db.Column( UUID(as_uuid=True), db.ForeignKey(User.id), nullable=False, default=lambda: g.user.id, ) creator = db.relationship(User, primaryjoin=creator_id == User.id) supplier_email = db.Column( CIText(), db.ForeignKey(User.email), nullable=False, default=lambda: g.user.email, ) supplier = db.relationship( User, primaryjoin=lambda: Deliverynote.supplier_email == User.email ) receiver_address = db.Column( CIText(), db.ForeignKey(User.email), nullable=False, default=lambda: g.user.email, ) receiver = db.relationship( User, primaryjoin=lambda: Deliverynote.receiver_address == User.email ) date = db.Column(db.DateTime, nullable=False, default=datetime.utcnow) date.comment = 'The date the DeliveryNote initiated' amount = db.Column(db.Integer, check_range('amount', min=0, max=100), default=0) # The following fields are supposed to be 0:N relationships # to SnapshotDelivery entity. # At this stage of implementation they will treated as a # comma-separated string of the devices expexted/transfered expected_devices = db.Column(JSONB, nullable=False) # expected_devices = db.Column(db.ARRAY(JSONB, dimensions=1), nullable=False) transferred_devices = db.Column(db.ARRAY(db.Integer, dimensions=1), nullable=True) transfer_state = db.Column( IntEnum(TransferState), default=TransferState.Initial, nullable=False ) transfer_state.comment = TransferState.__doc__ lot_id = db.Column(UUID(as_uuid=True), db.ForeignKey(Lot.id), nullable=False) lot = db.relationship( Lot, backref=db.backref('deliverynote', uselist=False, lazy=True), lazy=True, primaryjoin=Lot.id == lot_id, ) def __init__( self, document_id: str, amount: str, date, supplier_email: str, expected_devices: Iterable, transfer_state: TransferState, ) -> None: """Initializes a delivery note""" super().__init__( id=uuid.uuid4(), document_id=document_id, amount=amount, date=date, supplier_email=supplier_email, expected_devices=expected_devices, transfer_state=transfer_state, ) @property def type(self) -> str: return self.__class__.__name__ @property def url(self) -> urlutils.URL: """The URL where to GET this action.""" return urlutils.URL(url_for_resource(Deliverynote, item_id=self.id)) def delete(self): """Deletes the deliverynote. This method removes the delivery note. """ db.session.delete(self) def __repr__(self) -> str: return ''.format(self)