31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
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'])
|