pds-2021-g2-nest/nest_backend/database/tables/BoolOperation.py

"""
This module defines the BoolOperation database class.
"""

from ..base import ext
from .Enums import OperationType
from sqlalchemy.orm import backref


class BoolOperation(ext.Model):
    __tablename__ = "bool_operation"

    id = ext.Column(ext.Integer, primary_key=True)
    operation = ext.Column(ext.Enum(OperationType), nullable=False)
    is_root = ext.Column(ext.Boolean, default=False, nullable=False)
    # Foreign Keys
    condition_id = ext.Column(ext.Integer, ext.ForeignKey("condition.id"))
    node_1_id = ext.Column(ext.Integer, ext.ForeignKey("bool_operation.id", ondelete="SET NULL"))
    node_2_id = ext.Column(ext.Integer, ext.ForeignKey("bool_operation.id", ondelete="SET NULL"))
    alert_id = ext.Column(ext.Integer, ext.ForeignKey("alert.id", ondelete="CASCADE"))
    # Relationships
    condition = ext.relationship("Condition", back_populates="operations")
    node_1 = ext.relationship("BoolOperation", primaryjoin=("bool_operation.c.node_1_id==bool_operation.c.id"),
                              remote_side="BoolOperation.id", backref=backref("father_1", uselist=False))
    node_2 = ext.relationship("BoolOperation", primaryjoin=("bool_operation.c.node_2_id==bool_operation.c.id"),
                              remote_side="BoolOperation.id", backref=backref("father_2", uselist=False))
    alert = ext.relationship("Alert", back_populates="operations")

    def to_json(self):
        return {"id": self.id,
                "operation": self.operation,
                "is_root": self.is_root,
                "alert_id": self.alert_id,
                "condition": self.condition.to_json() if self.condition else None,
                "node_1": self.node_1.to_json() if self.node_1 else None,
                "node_2": self.node_2.to_json() if self.node_2 else None
                }

    def get_chain_ids(self, lista):
        if self.id in lista:
            # Loop detected!
            return -1
        lista.append(self.id)
        self.node_1.get_chain_ids(lista)
        self.node_1.get_chain_ids(lista)