Code format

alibaba · May 16, 2024 · ad04bd3 · ad04bd3
1 parent 36f4417
commit ad04bd3
Show file tree

Hide file tree

Showing 26 changed files with 370 additions and 692 deletions.
diff --git a/flex/coordinator/gs_flex_coordinator/core/__init__.py b/flex/coordinator/gs_flex_coordinator/core/__init__.py
@@ -27,5 +27,5 @@
     level=logging.INFO,
 )
 
-from gs_flex_coordinator.core.client_wrapper import client_wrapper
-from gs_flex_coordinator.core.utils import handle_api_exception
+from gs_flex_coordinator.core.client_wrapper import client_wrapper  # noqa: F401, E402
+from gs_flex_coordinator.core.utils import handle_api_exception  # noqa: F401, E402
diff --git a/flex/coordinator/gs_flex_coordinator/core/alert/alert_manager.py b/flex/coordinator/gs_flex_coordinator/core/alert/alert_manager.py
@@ -20,17 +20,19 @@
 import logging
 import os
 import pickle
-from typing import List, Union
+from typing import List
+from typing import Union
 
 from gs_flex_coordinator.core.alert.alert_receiver import DingTalkReceiver
-from gs_flex_coordinator.core.alert.builtin_rules import \
-    init_builtin_alert_rules
-from gs_flex_coordinator.core.alert.message_collector import \
-    AlertMessageCollector
+from gs_flex_coordinator.core.alert.builtin_rules import init_builtin_alert_rules
+from gs_flex_coordinator.core.alert.message_collector import AlertMessageCollector
 from gs_flex_coordinator.core.config import ALERT_WORKSPACE
 from gs_flex_coordinator.core.scheduler import schedule
-from gs_flex_coordinator.core.utils import decode_datetimestr, encode_datetime
-from gs_flex_coordinator.models import AlertMessage, AlertReceiver, AlertRule
+from gs_flex_coordinator.core.utils import decode_datetimestr
+from gs_flex_coordinator.core.utils import encode_datetime
+from gs_flex_coordinator.models import AlertMessage
+from gs_flex_coordinator.models import AlertReceiver
+from gs_flex_coordinator.models import AlertRule
 
 
 class AlertManager(object):
@@ -146,7 +148,10 @@ def list_alert_messages(
         return rlt
 
     def update_alert_messages(
-        self, messages: List[AlertMessage], batch_status: str, batch_delete: bool
+        self,
+        messages: List[AlertMessage],
+        batch_status: str,
+        batch_delete: bool,
     ):
         for message in messages:
             date = decode_datetimestr(message.trigger_time)

diff --git a/flex/coordinator/gs_flex_coordinator/core/alert/alert_rule.py b/flex/coordinator/gs_flex_coordinator/core/alert/alert_rule.py
@@ -18,10 +18,12 @@
 
 import datetime
 import logging
-from abc import ABCMeta, abstractmethod
+from abc import ABCMeta
+from abc import abstractmethod
 
 from gs_flex_coordinator.core.alert.alert_message import AlertMessage
-from gs_flex_coordinator.core.scheduler import cancel_job, schedule
+from gs_flex_coordinator.core.scheduler import cancel_job
+from gs_flex_coordinator.core.scheduler import schedule
 
 
 class AlertRule(metaclass=ABCMeta):

diff --git a/flex/coordinator/gs_flex_coordinator/core/alert/builtin_rules.py b/flex/coordinator/gs_flex_coordinator/core/alert/builtin_rules.py
@@ -25,9 +25,9 @@
 
 from gs_flex_coordinator.core import client_wrapper
 from gs_flex_coordinator.core.alert.alert_rule import AlertRule
-from gs_flex_coordinator.core.alert.message_collector import \
-    AlertMessageCollector
-from gs_flex_coordinator.core.config import CLUSTER_TYPE, SOLUTION
+from gs_flex_coordinator.core.alert.message_collector import AlertMessageCollector
+from gs_flex_coordinator.core.config import CLUSTER_TYPE
+from gs_flex_coordinator.core.config import SOLUTION
 
 
 class HighDiskUtilizationAlert(AlertRule):

diff --git a/flex/coordinator/gs_flex_coordinator/core/alert/message_collector.py b/flex/coordinator/gs_flex_coordinator/core/alert/message_collector.py
@@ -23,7 +23,8 @@
 
 from gs_flex_coordinator.core.alert.alert_message import AlertMessage
 from gs_flex_coordinator.core.config import ALERT_WORKSPACE
-from gs_flex_coordinator.core.scheduler import cancel_job, schedule
+from gs_flex_coordinator.core.scheduler import cancel_job
+from gs_flex_coordinator.core.scheduler import schedule
 from gs_flex_coordinator.core.utils import decode_datetimestr
 
 
@@ -59,7 +60,9 @@ def _pickle_messages_impl(self):
             self.dump_to_disk()
         except Exception as e:
             logging.warn(
-                "Failed to dump alert message on date %s: %s", str(self._date), str(e)
+                "Failed to dump alert message on date %s: %s",
+                str(self._date),
+                str(e),
             )
 
     def _try_to_recover_from_disk(self):
@@ -109,7 +112,8 @@ def clean(self):
                 cancel_job(self._pickle_messages_job, delete_scheduler=True)
                 self._pickle_messages_job = None
                 logging.info(
-                    "%s: current alert message collector cleaned", str(self._date)
+                    "%s: current alert message collector cleaned",
+                    str(self._date),
                 )
         except:  # noqa: E722, B110
             pass