This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
devicehub-teal/ereuse_devicehub/resources/event/schemas.py

397 lines
13 KiB
Python
Raw Normal View History

import decimal
2018-06-20 21:18:15 +00:00
from flask import current_app as app
2018-07-14 14:41:22 +00:00
from marshmallow import Schema as MarshmallowSchema, ValidationError, validates_schema
from marshmallow.fields import Boolean, DateTime, Decimal, Float, Integer, List, Nested, String, \
TimeDelta, URL, UUID
2018-06-20 21:18:15 +00:00
from marshmallow.validate import Length, Range
from sqlalchemy.util import OrderedSet
2018-06-20 21:18:15 +00:00
2018-04-27 17:16:43 +00:00
from ereuse_devicehub.marshmallow import NestedOn
from ereuse_devicehub.resources.agent.schemas import Agent
from ereuse_devicehub.resources.device.schemas import Component, Computer, Device
2018-06-10 16:47:49 +00:00
from ereuse_devicehub.resources.enums import AppearanceRange, Bios, FunctionalityRange, \
PriceSoftware, RATE_POSITIVE, RatingSoftware, ReceiverRole, SnapshotExpectedEvents, \
SnapshotSoftware, TestHardDriveLength
2018-06-12 14:50:05 +00:00
from ereuse_devicehub.resources.event import models as m
2018-04-27 17:16:43 +00:00
from ereuse_devicehub.resources.models import STR_BIG_SIZE, STR_SIZE
from ereuse_devicehub.resources.schemas import Thing
from ereuse_devicehub.resources.user.schemas import User
from teal.enums import Country, Currency, Subdivision
from teal.marshmallow import EnumField, IP, Version
2018-04-27 17:16:43 +00:00
from teal.resource import Schema
class Event(Thing):
2018-06-15 13:31:03 +00:00
id = UUID(dump_only=True)
2018-08-08 19:25:53 +00:00
name = String(default='', validate=Length(max=STR_BIG_SIZE), description=m.Event.name.comment)
2018-06-26 13:35:13 +00:00
incidence = Boolean(default=False, description=m.Event.incidence.comment)
closed = Boolean(missing=True, description=m.Event.closed.comment)
error = Boolean(default=False, description=m.Event.error.comment)
2018-06-26 13:35:13 +00:00
description = String(default='', description=m.Event.description.comment)
start_time = DateTime(data_key='startTime', description=m.Event.start_time.comment)
end_time = DateTime(data_key='endTime', description=m.Event.end_time.comment)
snapshot = NestedOn('Snapshot', dump_only=True)
agent = NestedOn(Agent, description=m.Event.agent_id.comment)
2018-06-12 14:50:05 +00:00
author = NestedOn(User, dump_only=True, exclude=('token',))
components = NestedOn(Component, dump_only=True, many=True)
parent = NestedOn(Computer, dump_only=True, description=m.Event.parent_id.comment)
2018-04-27 17:16:43 +00:00
class EventWithOneDevice(Event):
device = NestedOn(Device, only_query='id')
2018-04-27 17:16:43 +00:00
class EventWithMultipleDevices(Event):
devices = NestedOn(Device, many=True, only_query='id', collection_class=OrderedSet)
2018-04-27 17:16:43 +00:00
class Add(EventWithOneDevice):
pass
class Remove(EventWithOneDevice):
pass
class Allocate(EventWithMultipleDevices):
to = NestedOn(User,
description='The user the devices are allocated to.')
organization = String(validate=Length(max=STR_SIZE),
2018-04-27 17:16:43 +00:00
description='The organization where the user was when this happened.')
class Deallocate(EventWithMultipleDevices):
from_rel = Nested(User,
2018-04-27 17:16:43 +00:00
data_key='from',
description='The user where the devices are not allocated to anymore.')
organization = String(validate=Length(max=STR_SIZE),
2018-04-27 17:16:43 +00:00
description='The organization where the user was when this happened.')
class EraseBasic(EventWithOneDevice):
2018-07-02 10:52:54 +00:00
zeros = Boolean(required=True, description=m.EraseBasic.zeros.comment)
2018-06-10 16:47:49 +00:00
steps = NestedOn('Step', many=True, required=True)
2018-04-27 17:16:43 +00:00
class EraseSectors(EraseBasic):
pass
class Step(Schema):
2018-06-10 16:47:49 +00:00
type = String(description='Only required when it is nested.')
start_time = DateTime(required=True, data_key='startTime')
end_time = DateTime(required=True, data_key='endTime')
error = Boolean(default=False, description='Did the event fail?')
class StepZero(Step):
pass
class StepRandom(Step):
pass
class Rate(EventWithOneDevice):
rating = Integer(validate=Range(*RATE_POSITIVE),
dump_only=True,
data_key='ratingValue',
description='The rating for the content.')
2018-07-14 14:41:22 +00:00
software = EnumField(RatingSoftware,
dump_only=True,
description='The algorithm used to produce this rating.')
version = Version(dump_only=True,
description='The version of the software.')
2018-06-10 16:47:49 +00:00
appearance = Integer(validate=Range(-3, 5), dump_only=True)
functionality = Integer(validate=Range(-3, 5),
dump_only=True,
data_key='functionalityScore')
class IndividualRate(Rate):
pass
class AggregateRate(Rate):
ratings = NestedOn(IndividualRate, many=True)
class PhotoboxRate(IndividualRate):
num = Integer(dump_only=True)
# todo Image
2018-06-20 21:18:15 +00:00
class PhotoboxUserRate(IndividualRate):
2018-06-10 16:47:49 +00:00
assembling = Integer()
parts = Integer()
buttons = Integer()
dents = Integer()
decolorization = Integer()
scratches = Integer()
tag_adhesive = Integer()
dirt = Integer()
2018-04-27 17:16:43 +00:00
2018-06-20 21:18:15 +00:00
class PhotoboxSystemRate(IndividualRate):
2018-06-10 16:47:49 +00:00
pass
2018-06-20 21:18:15 +00:00
class ManualRate(IndividualRate):
2018-06-10 16:47:49 +00:00
appearance_range = EnumField(AppearanceRange,
required=True,
data_key='appearanceRange',
description='Grades the imperfections that aesthetically '
'affect the device, but not its usage.')
functionality_range = EnumField(FunctionalityRange,
required=True,
data_key='functionalityRange',
2018-07-14 14:41:22 +00:00
description='Grades the defects of a device affecting usage.')
2018-06-20 21:18:15 +00:00
labelling = Boolean(description='Sets if there are labels stuck that should be removed.')
class AppRate(ManualRate):
pass
class WorkbenchRate(ManualRate):
processor = Float()
ram = Float()
data_storage = Float()
graphic_card = Float()
bios = EnumField(Bios, description='How difficult it has been to set the bios to '
'boot from the network.')
2018-06-10 16:47:49 +00:00
2018-07-14 14:41:22 +00:00
class Price(EventWithOneDevice):
currency = EnumField(Currency, required=True)
price = Decimal(places=4, rounding=decimal.ROUND_HALF_EVEN, required=True)
2018-07-14 14:41:22 +00:00
software = EnumField(PriceSoftware, dump_only=True)
version = Version(dump_only=True)
rating = NestedOn(AggregateRate, dump_only=True)
class EreusePrice(Price):
class Service(MarshmallowSchema):
class Type(MarshmallowSchema):
amount = Float()
percentage = Float()
standard = Nested(Type)
warranty2 = Nested(Type)
warranty2 = Float()
refurbisher = Nested(Service)
retailer = Nested(Service)
platform = Nested(Service)
2018-06-10 16:47:49 +00:00
class Install(EventWithOneDevice):
2018-06-19 16:38:42 +00:00
name = String(validate=Length(min=4, max=STR_BIG_SIZE),
2018-04-27 17:16:43 +00:00
required=True,
description='The name of the OS installed.')
elapsed = TimeDelta(precision=TimeDelta.SECONDS, required=True)
class Snapshot(EventWithOneDevice):
"""
The Snapshot updates the state of the device with information about
its components and events performed at them.
See docs for more info.
"""
2018-06-20 21:18:15 +00:00
uuid = UUID()
2018-06-10 16:47:49 +00:00
software = EnumField(SnapshotSoftware,
required=True,
description='The software that generated this Snapshot.')
version = Version(required=True, description='The version of the software.')
events = NestedOn(Event, many=True, dump_only=True)
2018-06-19 16:38:42 +00:00
expected_events = List(EnumField(SnapshotExpectedEvents),
data_key='expectedEvents',
description='Keep open this Snapshot until the following events'
'are performed. Setting this value will activate'
'the async Snapshot.')
2018-06-20 21:18:15 +00:00
elapsed = TimeDelta(precision=TimeDelta.SECONDS)
components = NestedOn(Component,
many=True,
description='A list of components that are inside of the device'
'at the moment of this Snapshot.'
'Order is preserved, so the component num 0 when'
'submitting is the component num 0 when returning it back.')
2018-04-27 17:16:43 +00:00
@validates_schema
def validate_workbench_version(self, data: dict):
2018-06-10 16:47:49 +00:00
if data['software'] == SnapshotSoftware.Workbench:
2018-04-27 17:16:43 +00:00
if data['version'] < app.config['MIN_WORKBENCH']:
raise ValidationError(
2018-07-14 14:41:22 +00:00
'Min. supported Workbench version is '
2018-06-10 16:47:49 +00:00
'{}'.format(app.config['MIN_WORKBENCH']),
2018-04-27 17:16:43 +00:00
field_names=['version']
)
@validates_schema
def validate_components_only_workbench(self, data: dict):
2018-06-10 16:47:49 +00:00
if data['software'] != SnapshotSoftware.Workbench:
2018-06-20 21:18:15 +00:00
if data.get('components', None) is not None:
2018-04-27 17:16:43 +00:00
raise ValidationError('Only Workbench can add component info',
field_names=['components'])
2018-06-20 21:18:15 +00:00
@validates_schema
def validate_only_workbench_fields(self, data: dict):
"""Ensures workbench has ``elapsed`` and ``uuid`` and no others."""
# todo test
if data['software'] == SnapshotSoftware.Workbench:
if not data.get('uuid', None):
raise ValidationError('Snapshots from Workbench must have uuid',
field_names=['uuid'])
if not data.get('elapsed', None):
raise ValidationError('Snapshots from Workbench must have elapsed',
field_names=['elapsed'])
else:
if data.get('uuid', None):
raise ValidationError('Only Snapshots from Workbench can have uuid',
field_names=['uuid'])
if data.get('elapsed', None):
raise ValidationError('Only Snapshots from Workbench can have elapsed',
field_names=['elapsed'])
2018-04-27 17:16:43 +00:00
class Test(EventWithOneDevice):
elapsed = TimeDelta(precision=TimeDelta.SECONDS, required=True)
2018-06-10 16:47:49 +00:00
class TestDataStorage(Test):
2018-04-27 17:16:43 +00:00
length = EnumField(TestHardDriveLength, required=True)
status = String(validate=Length(max=STR_SIZE), required=True)
2018-07-02 10:52:54 +00:00
lifetime = TimeDelta(precision=TimeDelta.DAYS)
2018-06-19 16:38:42 +00:00
assessment = Boolean()
reallocated_sector_count = Integer(data_key='reallocatedSectorCount')
power_cycle_count = Integer(data_key='powerCycleCount')
reported_uncorrectable_errors = Integer(data_key='reportedUncorrectableErrors')
command_timeout = Integer(data_key='commandTimeout')
current_pending_sector_count = Integer(data_key='currentPendingSectorCount')
offline_uncorrectable = Integer(data_key='offlineUncorrectable')
remaining_lifetime_percentage = Integer(data_key='remainingLifetimePercentage')
2018-04-27 17:16:43 +00:00
class StressTest(Test):
pass
2018-06-19 16:38:42 +00:00
class Benchmark(EventWithOneDevice):
elapsed = TimeDelta(precision=TimeDelta.SECONDS)
class BenchmarkDataStorage(Benchmark):
read_speed = Float(required=True, data_key='readSpeed')
write_speed = Float(required=True, data_key='writeSpeed')
class BenchmarkWithRate(Benchmark):
rate = Integer(required=True)
class BenchmarkProcessor(BenchmarkWithRate):
pass
class BenchmarkProcessorSysbench(BenchmarkProcessor):
pass
class BenchmarkRamSysbench(BenchmarkWithRate):
pass
class ToRepair(EventWithMultipleDevices):
pass
class Repair(EventWithMultipleDevices):
pass
class ReadyToUse(EventWithMultipleDevices):
pass
class ToPrepare(EventWithMultipleDevices):
pass
class Prepare(EventWithMultipleDevices):
pass
class Live(EventWithOneDevice):
ip = IP(dump_only=True)
subdivision_confidence = Integer(dump_only=True, data_key='subdivisionConfidence')
subdivision = EnumField(Subdivision, dump_only=True)
country = EnumField(Country, dump_only=True)
city = String(dump_only=True)
city_confidence = Integer(dump_only=True, data_key='cityConfidence')
isp = String(dump_only=True)
organization = String(dump_only=True)
organization_type = String(dump_only=True, data_key='organizationType')
class Organize(EventWithMultipleDevices):
pass
class Reserve(Organize):
pass
class CancelReservation(Organize):
pass
class Trade(EventWithMultipleDevices):
shipping_date = DateTime(data_key='shippingDate')
invoice_number = String(validate=Length(max=STR_SIZE), data_key='invoiceNumber')
price = NestedOn(Price)
to = NestedOn(Agent, only_query='id')
confirms = NestedOn(Organize)
class Sell(Trade):
pass
class Donate(Trade):
pass
class Rent(Trade):
pass
class CancelTrade(Trade):
pass
class ToDisposeProduct(Trade):
pass
class DisposeProduct(Trade):
pass
class Receive(EventWithMultipleDevices):
role = EnumField(ReceiverRole)
class Migrate(EventWithMultipleDevices):
other = URL()
class MigrateTo(Migrate):
pass
class MigrateFrom(Migrate):
pass