OpServ-Monitoring/opserv-backend

from .component_type_metrics_table_management import ComponentTypeMetricsTableManagement
from .general_table_management import GeneralTableManagement


class ComponentMetricsTableManagement(GeneralTableManagement):
    @classmethod
    def TABLE_NAME(cls) -> str:
        return "component_metrics_table"

    __KEY_COMPONENT_TYPE_FK = "component_type_fk"
    __KEY_COMPONENT_ARG = "component_arg"
    __KEY_COMPONENT_METRIC_FK = "component_metric_fk"
    __KEY_COMPONENT_GATHERING_RATE = "component_gathering_rate"

    @classmethod
    def KEY_COMPONENT_TYPE_FK(cls):
        return ComponentMetricsTableManagement.__KEY_COMPONENT_TYPE_FK

    @classmethod
    def KEY_COMPONENT_ARG(cls):
        return ComponentMetricsTableManagement.__KEY_COMPONENT_ARG

    @classmethod
    def KEY_COMPONENT_METRIC_FK(cls):
        return ComponentMetricsTableManagement.__KEY_COMPONENT_METRIC_FK

    @classmethod
    def KEY_COMPONENT_GATHERING_RATE(cls):
        return ComponentMetricsTableManagement.__KEY_COMPONENT_GATHERING_RATE

    @classmethod
    def _get_columns(cls) -> list:
        type_integer_gr_or_eq_500 = \
            "INTEGER CHECK({0} IS NULL OR (typeof({0}) = 'integer' and ({0} == 0 or {0} >= 500)))".format(
                ComponentMetricsTableManagement.KEY_COMPONENT_GATHERING_RATE()
            )

        return [
            (ComponentMetricsTableManagement.KEY_COMPONENT_TYPE_FK(), GeneralTableManagement._type_text_not_null),
            (ComponentMetricsTableManagement.KEY_COMPONENT_ARG(), GeneralTableManagement._type_text_not_null),
            (ComponentMetricsTableManagement.KEY_COMPONENT_METRIC_FK(), GeneralTableManagement._type_text_not_null),
            (ComponentMetricsTableManagement.KEY_COMPONENT_GATHERING_RATE(), type_integer_gr_or_eq_500)
        ]

    @classmethod
    def _get_primary_key(cls) -> list:
        return [
            ComponentMetricsTableManagement.KEY_COMPONENT_TYPE_FK(),
            ComponentMetricsTableManagement.KEY_COMPONENT_ARG(),
            ComponentMetricsTableManagement.KEY_COMPONENT_METRIC_FK()
        ]

    @classmethod
    def _get_foreign_keys(cls) -> list:
        return [
            GeneralTableManagement._build_foreign_key(
                [
                    ComponentMetricsTableManagement.KEY_COMPONENT_TYPE_FK(),
                    ComponentMetricsTableManagement.KEY_COMPONENT_METRIC_FK()
                ],
                ComponentTypeMetricsTableManagement.TABLE_NAME(),
                [
                    ComponentTypeMetricsTableManagement.KEY_COMPONENT_TYPE_FK(),
                    ComponentTypeMetricsTableManagement.KEY_METRIC_FK()
                ]
            )
        ]