67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
from flask import current_app as app
|
|
from marshmallow import Schema as MarshmallowSchema, ValidationError, validates_schema
|
|
from marshmallow.fields import Boolean, DateTime, Integer, Nested, String, UUID
|
|
from marshmallow.validate import Length
|
|
from sqlalchemy.util import OrderedSet
|
|
from teal.marshmallow import SanitizedStr, URL
|
|
from teal.resource import Schema
|
|
|
|
from ereuse_devicehub.marshmallow import NestedOn
|
|
from ereuse_devicehub.resources.proof import models as m
|
|
from ereuse_devicehub.resources.models import STR_BIG_SIZE, STR_SIZE
|
|
from ereuse_devicehub.resources.schemas import Thing
|
|
from ereuse_devicehub.resources.action import schemas as s_action
|
|
from ereuse_devicehub.resources.device import schemas as s_device
|
|
|
|
|
|
class Proof(Thing):
|
|
__doc__ = m.Proof.__doc__
|
|
id = UUID(dump_only=True)
|
|
ethereum_hash = SanitizedStr(default='', validate=Length(max=STR_BIG_SIZE),
|
|
data_key="ethereumHash", required=True)
|
|
url = URL(dump_only=True, description=m.Proof.url.__doc__)
|
|
devices = NestedOn(s_device.Device,
|
|
many=True,
|
|
required=True, # todo test ensuring len(devices) >= 1
|
|
only_query='id',
|
|
data_key='deviceIDs',
|
|
collection_class=OrderedSet)
|
|
|
|
|
|
class ProofTransfer(Proof):
|
|
__doc__ = m.ProofTransfer.__doc__
|
|
transfer = NestedOn(s_action.DisposeProduct,
|
|
required=True,
|
|
data_key='transferID',
|
|
only_query='id')
|
|
|
|
|
|
class ProofDataWipe(Proof):
|
|
__doc__ = m.ProofDataWipe.__doc__
|
|
erasure_type = SanitizedStr(default='', data_key='erasureType')
|
|
date = DateTime('iso', required=True)
|
|
result = Boolean(required=True)
|
|
erasure = NestedOn(s_action.EraseBasic, only_query='id', data_key='erasureID')
|
|
|
|
|
|
class ProofFunction(Proof):
|
|
__doc__ = m.ProofFunction.__doc__
|
|
disk_usage = Integer(data_key='diskUsage')
|
|
rate = NestedOn(s_action.Rate, required=True,
|
|
only_query='id', data_key='rateID')
|
|
|
|
|
|
class ProofReuse(Proof):
|
|
__doc__ = m.ProofReuse.__doc__
|
|
price = Integer()
|
|
|
|
|
|
class ProofRecycling(Proof):
|
|
__doc__ = m.ProofRecycling.__doc__
|
|
collection_point = SanitizedStr(default='', data_key='collectionPoint', required=True)
|
|
date = DateTime('iso', required=True)
|
|
contact = SanitizedStr(default='', required=True)
|
|
ticket = SanitizedStr(default='', required=True)
|
|
gps_location = SanitizedStr(default='', data_key='gpsLocation', required=True)
|
|
recycler_code = SanitizedStr(default='', data_key='recyclerCode', required=True)
|