fix: Handle disabled shelter

codecov/settings_base.py

@@ -395,6 +395,7 @@
 )
 
 SHELTER_SHARED_SECRET = get_config("setup", "shelter_shared_secret", default=None)
+SHELTER_ENABLED = get_config("setup", "shelter_enabled", default=True)
 
 SENTRY_ENV = os.environ.get("CODECOV_ENV", False)
 SENTRY_DSN = os.environ.get("SERVICES__SENTRY__SERVER_DSN", None)

codecov/settings_enterprise.py

@@ -72,3 +72,5 @@
 ]
 
 GUEST_ACCESS = get_config("setup", "guest_access", default=True)
+
+SHELTER_ENABLED = get_config("setup", "shelter_enabled", default=False)

utils/shelter.py

@@ -23,6 +23,9 @@
         return cls._instance
 
     def __init__(self) -> None:
+        if not settings.SHELTER_ENABLED:
+            return
+
         if not self.pubsub_publisher:
             self.pubsub_publisher = pubsub_v1.PublisherClient()
         pubsub_project_id: str = settings.SHELTER_PUBSUB_PROJECT_ID
@@ -32,6 +35,9 @@
         self.topic_path = self.pubsub_publisher.topic_path(pubsub_project_id, topic_id)
 
     def publish(self, data: Dict[str, Any]) -> None:
+        if not settings.SHELTER_ENABLED:
+            return
+
         try:
             self.pubsub_publisher.publish(
                 self.topic_path,