mirror of
https://github.com/pds-nest/nest.git
synced 2024-11-30 00:14:19 +00:00
45 lines
2 KiB
Python
45 lines
2 KiB
Python
"""
|
|
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)
|
|
isRoot = 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"))
|
|
# 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, l):
|
|
if self.id in l:
|
|
# Loop detected!
|
|
return -1
|
|
l.append(self.id)
|
|
self.get_chain_ids(self.node_1, l)
|
|
self.get_chain_ids(self.node_2, l)
|