Skip to content
Snippets Groups Projects
KpiModel.py 1.78 KiB
Newer Older
import logging
from typing import Dict
from common.orm.fields.EnumeratedField import EnumeratedField
from common.orm.fields.FloatField import FloatField
from common.orm.fields.ForeignKeyField import ForeignKeyField
from common.orm.fields.PrimaryKeyField import PrimaryKeyField
from common.orm.fields.StringField import StringField
from common.orm.model.Model import Model
from .DeviceModel import DeviceModel
from .EndPointModel import EndPointModel
from .KpiSampleType import ORM_KpiSampleTypeEnum

LOGGER = logging.getLogger(__name__)

class KpiModel(Model):
    pk = PrimaryKeyField()
    kpi_uuid = StringField(required=True, allow_empty=False)
    kpi_description = StringField(required=False, allow_empty=True)
    kpi_sample_type = EnumeratedField(ORM_KpiSampleTypeEnum, required=True)
    device_fk = ForeignKeyField(DeviceModel)
    endpoint_fk = ForeignKeyField(EndPointModel)
    sampling_duration = FloatField(min_value=0, required=True)
    sampling_interval = FloatField(min_value=0, required=True)

    def dump_id(self) -> Dict:
        return {'kpi_id': {'uuid': self.kpi_uuid}}

    def dump_descriptor(self) -> Dict:
        result = {
            'kpi_description': self.kpi_description,
            'kpi_sample_type': self.kpi_sample_type.value,
        }
        if self.device_fk is not None:
            result['device_id'] = DeviceModel(self.database, self.device_fk).dump_id()
        if self.endpoint_fk is not None:
            result['endpoint_id'] = EndPointModel(self.database, self.endpoint_fk).dump_id()
        return result

    def dump(self) -> Dict:
        return {
            'kpi_id': self.dump_id(),
            'kpi_descriptor': self.dump_descriptor(),
            'sampling_duration_s': self.sampling_duration,
            'sampling_interval_s': self.sampling_interval,
        }