messaging.sensormessages

Complexity

Total Complexity

4

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	SensorSchema.make_sensor()	0	4	1
A	SensorValueSchema.make_sensorvalue()	0	4	1
A	SensorValueMessage.__init__()	0	4	1
A	SensorMessage.__init__()	0	4	1

'''sensor messages and schema'''
from marshmallow import Schema, fields, post_load
from domain.sensors import Sensor, SensorValue

class SensorValueMessage(object):
    """Sensor data in a message"""
    def __init__(self, sensorid, value, sensor_type):
        self.sensorid = sensorid
        self.value = value
        self.type = sensor_type

class SensorMessage(object):
    """Sensor message"""
    def __init__(self, sensorid, sensor_type, location):
        self.sensorid = sensorid
        self.type = sensor_type
        self.location = location

class SensorSchema(Schema):
    '''schema for sensor'''
    sensorid = fields.Str()
    sensortype = fields.Str()
    location = fields.Str()

    @post_load
    def make_sensor(self, data):
        '''reconstitute sensor'''
        return Sensor(**data)

class SensorValueSchema(Schema):
    '''serialized version of sensor value'''
    sensorid = fields.Str()
    value = fields.Str()
    valuetype = fields.Str()
    sensor = fields.Nested(SensorSchema, allow_none=True)
    sensortime = fields.DateTime()

    @post_load
    def make_sensorvalue(self, data):
        '''reconstitute sensorvalue'''
        return SensorValue(**data)