Enable black and pylint in tox.ini

[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 e596d74..6423b5e 100644 (file)
--- a/osm_policy_module/core/database.py
+++ b/osm_policy_module/core/database.py
@@ -98,7 +98,7 @@ class VnfAlarm(BaseModel):
     nsr_id = CharField()
     vnf_member_index = CharField()
     vdu_name = CharField()
-    last_action = CharField(default='insufficient-data')
+    last_action = CharField(default="insufficient-data")
     id_suffix = IntegerField()
     ok_ack = BooleanField(default=False)
     alarm_ack = BooleanField(default=False)
@@ -122,7 +122,7 @@ class HealingAction(BaseModel):
     cooldown_time = IntegerField()
     count_index = IntegerField()
     last_heal = DateTimeField(default=datetime.datetime.now)
-    last_status = CharField(default='insufficient-data')
+    last_status = CharField(default="insufficient-data")
     day1 = BooleanField(default=False)
 
 
@@ -141,7 +141,7 @@ class DatabaseManager:
 class ScalingAlarmRepository:
     @staticmethod
     def list(*expressions) -> Iterable[ScalingAlarm]:
-        return ScalingAlarm.select().where(*expressions)
+        return ScalingAlarm.select().where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions, join_classes: List = None) -> ScalingAlarm:
@@ -159,7 +159,7 @@ class ScalingAlarmRepository:
 class ScalingGroupRepository:
     @staticmethod
     def list(*expressions) -> Iterable[ScalingGroup]:
-        return ScalingGroup.select().where(*expressions)
+        return ScalingGroup.select().where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions) -> ScalingGroup:
@@ -177,7 +177,7 @@ class ScalingPolicyRepository:
         if join_classes:
             for join_class in join_classes:
                 query = query.join(join_class)
-        return query.where(*expressions)
+        return query.where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions, join_classes: List = None) -> ScalingPolicy:
@@ -199,7 +199,7 @@ class ScalingCriteriaRepository:
         if join_classes:
             for join_class in join_classes:
                 query = query.join(join_class)
-        return query.where(*expressions)
+        return query.where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions, join_classes: List = None) -> ScalingCriteria:
@@ -217,7 +217,7 @@ class ScalingCriteriaRepository:
 class VnfAlarmRepository:
     @staticmethod
     def list(*expressions) -> Iterable[VnfAlarm]:
-        return VnfAlarm.select().where(*expressions)
+        return VnfAlarm.select().where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions) -> VnfAlarm:
@@ -231,7 +231,7 @@ class VnfAlarmRepository:
 class AlarmActionRepository:
     @staticmethod
     def list(*expressions) -> Iterable[AlarmAction]:
-        return AlarmAction.select().where(*expressions)
+        return AlarmAction.select().where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions) -> AlarmAction:
@@ -245,8 +245,12 @@ class AlarmActionRepository:
 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)
+        log.info(
+            "### Printing healing action db alarm {}".format(
+                HealingAction.select().where(*expressions)
+            )
+        )
+        return HealingAction.select().where(*expressions).__iter__()
 
     @staticmethod
     def get(*expressions) -> HealingAction: