Feature 10923: Autohealing

[osm/POL.git] / osm_policy_module / core / database.py

diff --git a/osm_policy_module/core/database.py b/osm_policy_module/core/database.py
index b7a0b4f..e596d74 100644 (file)
--- a/osm_policy_module/core/database.py
+++ b/osm_policy_module/core/database.py
@@ -110,6 +110,22 @@ class AlarmAction(BaseModel):
     alarm = ForeignKeyField(VnfAlarm, related_name="actions", on_delete="CASCADE")
 
 
+class HealingAction(BaseModel):
+    alarm_id = CharField()
+    recovery_action = CharField()
+    alarm_uuid = CharField(unique=True)
+    nsr_id = CharField()
+    vnfinstance_id = CharField()
+    vnf_member_index = CharField()
+    vdur_name = CharField()
+    vdu_id = CharField()
+    cooldown_time = IntegerField()
+    count_index = IntegerField()
+    last_heal = DateTimeField(default=datetime.datetime.now)
+    last_status = CharField(default='insufficient-data')
+    day1 = BooleanField(default=False)
+
+
 class DatabaseManager:
     def __init__(self, config: Config):
         db.initialize(connect(config.get("sql", "database_uri")))
@@ -224,3 +240,18 @@ class AlarmActionRepository:
     @staticmethod
     def create(**query) -> AlarmAction:
         return AlarmAction.create(**query)
+
+
+class HealingActionRepository:
+    @staticmethod
+    def list(*expressions) -> Iterable[HealingAction]:
+        log.info("### Printing healing action db alarm {}".format(HealingAction.select().where(*expressions)))
+        return HealingAction.select().where(*expressions)
+
+    @staticmethod
+    def get(*expressions) -> HealingAction:
+        return HealingAction.select().where(*expressions).get()
+
+    @staticmethod
+    def create(**query) -> HealingAction:
+        return HealingAction.create(**query)