devicehub-teal/ereuse_devicehub/resources/tradedocument/schemas.py

import base64

from marshmallow.fields import DateTime, Integer, Raw
from teal.marshmallow import SanitizedStr
from marshmallow import ValidationError, validates_schema

from ereuse_devicehub.marshmallow import NestedOn
from ereuse_devicehub.resources.schemas import Thing
from ereuse_devicehub.resources.tradedocument import models as m
from ereuse_devicehub.resources.lot import schemas as s_lot


class TradeDocument(Thing):
    __doc__ = m.TradeDocument.__doc__
    id = Integer(description=m.TradeDocument.id.comment, dump_only=True)
    date = DateTime(required=False, description=m.TradeDocument.date.comment)
    id_document = SanitizedStr(default='', description=m.TradeDocument.id_document.comment)
    description = SanitizedStr(default='', description=m.TradeDocument.description.comment)
    file_name = SanitizedStr(default='', description=m.TradeDocument.file_name.comment)
    file = Raw(type='file')
    lot = NestedOn(s_lot.Lot, only_query='id', description=m.TradeDocument.lot.__doc__)


    @validates_schema
    def validate_filestream(self, data):
        if not data.get('file'):
            txt = 'Error, no there are any file for save'
            raise ValidationError(txt)

        data['file'] = base64.b64decode(data['file'])