Fix mypy errors

Author: rdheekonda

dreadnode/credential_manager.py

@@ -4,7 +4,7 @@
 from typing import TYPE_CHECKING, TypeVar
 
 from botocore.exceptions import ClientError
-from s3fs import S3FileSystem
+from s3fs import S3FileSystem  # type: ignore[import-untyped]
 
 from dreadnode.constants import FS_CREDENTIAL_REFRESH_BUFFER
 from dreadnode.util import logger, resolve_endpoint
@@ -29,7 +29,7 @@ def __init__(self, credential_fetcher: Callable[[], "UserDataCredentials"]):
         self._credential_fetcher = credential_fetcher
         self._credentials: UserDataCredentials | None = None
         self._credentials_expiry: datetime | None = None
-        self._filesystem = None
+        self._filesystem: S3FileSystem | None = None
         self._prefix = ""
 
     def initialize(self) -> None:
@@ -40,6 +40,7 @@ def get_filesystem(self) -> S3FileSystem:
         """Get current filesystem, refreshing credentials if needed."""
         if self._needs_refresh():
             self._refresh_credentials()
+        assert self._filesystem is not None  # noqa: S101
         return self._filesystem
 
     def get_prefix(self) -> str:

dreadnode/main.py

@@ -347,13 +347,15 @@ def initialize(self) -> None:
             #         )
             #     )
             # )
-            self._credential_manager = CredentialManager(
-                credential_fetcher=lambda: self._api.get_user_data_credentials()
-            )
-            self._credential_manager.initialize()
+            if self._api is not None:
+                api = self._api
+                self._credential_manager = CredentialManager(
+                    credential_fetcher=lambda: api.get_user_data_credentials()
+                )
+                self._credential_manager.initialize()
 
-            self._fs = self._credential_manager.get_filesystem()
-            self._fs_prefix = self._credential_manager.get_prefix()
+                self._fs = self._credential_manager.get_filesystem()
+                self._fs_prefix = self._credential_manager.get_prefix()
 
         self._logfire = logfire.configure(
             local=not self.is_default,