fix typing and formatting

dbt-adapters/src/dbt/adapters/base/catalog.py

@@ -1,13 +1,14 @@
 import abc
 from dataclasses import dataclass
-from enum import Enum
 from typing import Optional, Dict
 
 from dbt.adapters.contracts.relation import RelationConfig
 from dbt.adapters.relation_configs.formats import TableFormat
 from dbt.adapters.protocol import CatalogIntegrationProtocol
+from dbt_common.dataclass_schema import StrEnum
 
-class CatalogIntegrationType(Enum):
+
+class CatalogIntegrationType(StrEnum):
     managed = "managed"
     iceberg_rest = "iceberg_rest"
     glue = "glue"
@@ -30,7 +31,7 @@ class CatalogIntegration(abc.ABC, CatalogIntegrationProtocol):
     Implements the CatalogIntegrationProtocol as an abstract class should be subclassed by
     specific integrations in the adapters.
 
-    A catalog integration is a platform's way of interacting with an external catalog. 
+    A catalog integration is a platform's way of interacting with an external catalog.
     """
 
     catalog_name: str

dbt-adapters/src/dbt/adapters/base/impl.py

@@ -318,7 +318,9 @@ def add_catalog_integrations(
                 catalogs_client.add_catalog(integration, integration_config.catalog_name)
 
     @available
-    def get_catalog_integration(self, integration_name: str) -> CatalogIntegrationProtocol:
+    def get_catalog_integration(
+        self, integration_name: str
+    ) -> Optional[CatalogIntegrationProtocol]:
         return catalogs_client.get_catalog(integration_name)
 
     ###

dbt-adapters/src/dbt/adapters/clients/catalogs.py

@@ -3,6 +3,7 @@
 
 from typing import Optional
 
+
 class CatalogIntegrations:
     def __init__(self):
         self._integrations = {}
@@ -21,11 +22,13 @@ def add_integration(self, integration: CatalogIntegrationProtocol, catalog_name:
             raise DbtCatalogIntegrationAlreadyExistsError(catalog_name)
         self._integrations[catalog_name] = integration
 
+
 _CATALOG_CLIENT = CatalogIntegrations()
 
 
 def get_catalog(integration_name: str) -> Optional[CatalogIntegrationProtocol]:
     return _CATALOG_CLIENT.get(integration_name)
 
+
 def add_catalog(integration: CatalogIntegrationProtocol, catalog_name: str):
     _CATALOG_CLIENT.add_integration(integration, catalog_name)

dbt-adapters/src/dbt/adapters/exceptions/catalog_integration.py

@@ -1,4 +1,3 @@
-
 from dbt_common.exceptions import DbtRuntimeError
 
 

dbt-adapters/tests/unit/clients/test_catalogs.py

@@ -8,12 +8,14 @@ def test_adding_and_getting_catalog_integration(fake_catalog_integration):
     add_catalog(catalog, catalog_name="fake_catalog")
     assert get_catalog("fake_catalog").catalog_name == catalog.catalog_name
 
+
 def test_adding_catalog_integration_that_already_exists(fake_catalog_integration):
     catalog = fake_catalog_integration
     catalog_name = "fake_catalog_2"
     add_catalog(catalog, catalog_name=catalog_name)
     with pytest.raises(DbtCatalogIntegrationAlreadyExistsError):
         add_catalog(catalog, catalog_name=catalog_name)
 
+
 def test_getting_catalog_integration_that_does_not_exist():
     assert get_catalog("non_existent_catalog") is None