WIP: fixed creds issue

censys · Oct 26, 2023 · e3b327b · e3b327b
1 parent 766b789
commit e3b327b
Show file tree

Hide file tree

Showing 2 changed files with 126 additions and 76 deletions.
diff --git a/src/censys/cloud_connectors/common/connector.py b/src/censys/cloud_connectors/common/connector.py
@@ -49,7 +49,7 @@ def __init__(self, settings: Settings):
         """
         if not self.provider:
             raise ValueError("The provider must be set.")
-        self.label_prefix = self.provider.label() + ": "
+        self.label_prefix = self.get_provider_label_prefix()
         self.settings = settings
         self.logger = get_logger(
             log_name=f"{self.provider.lower()}_cloud_connector",
@@ -79,6 +79,13 @@ def __init__(self, settings: Settings):
         self.cloud_assets = defaultdict(set)
         self.current_service = None
 
+    def get_provider_label_prefix(self):
+        """Get the provider label prefix.
+        Returns:
+            str: Provider label prefix.
+        """
+        return self.provider.label() + ": "
+
     def delete_seeds_by_label(self, label: str):
         """Replace seeds for [label] with an empty list.