openmetadata-ingestion 1.6.0.0rc1__py3-none-any.whl → 1.6.0.0rc3__py3-none-any.whl

metadata/ingestion/source/dashboard/{mstr → microstrategy}/metadata.py

@@ -8,7 +8,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
-"""Mstr source module"""
+"""MicroStrategy source module"""
 import traceback
 from typing import Iterable, List, Optional
 
@@ -16,8 +16,8 @@ from metadata.generated.schema.api.data.createChart import CreateChartRequest
 from metadata.generated.schema.api.data.createDashboard import CreateDashboardRequest
 from metadata.generated.schema.api.lineage.addLineage import AddLineageRequest
 from metadata.generated.schema.entity.data.chart import Chart
-from metadata.generated.schema.entity.services.connections.dashboard.mstrConnection import (
-    MstrConnection,
+from metadata.generated.schema.entity.services.connections.dashboard.microStrategyConnection import (
+    MicroStrategyConnection,
 )
 from metadata.generated.schema.entity.services.ingestionPipelines.status import (
     StackTraceError,
@@ -34,7 +34,7 @@ from metadata.ingestion.api.models import Either
 from metadata.ingestion.api.steps import InvalidSourceException
 from metadata.ingestion.ometa.ometa_api import OpenMetadata
 from metadata.ingestion.source.dashboard.dashboard_service import DashboardServiceSource
-from metadata.ingestion.source.dashboard.mstr.models import (
+from metadata.ingestion.source.dashboard.microstrategy.models import (
     MstrDashboard,
     MstrDashboardDetails,
     MstrPage,
@@ -47,9 +47,9 @@ from metadata.utils.logger import ingestion_logger
 logger = ingestion_logger()
 
 
-class MstrSource(DashboardServiceSource):
+class MicrostrategySource(DashboardServiceSource):
     """
-    MSTR Source Class
+    Microstrategy Source Class
     """
 
     @classmethod
@@ -60,10 +60,10 @@ class MstrSource(DashboardServiceSource):
         pipeline_name: Optional[str] = None,
     ):
         config = WorkflowSource.model_validate(config_dict)
-        connection: MstrConnection = config.serviceConnection.root.config
-        if not isinstance(connection, MstrConnection):
+        connection: MicroStrategyConnection = config.serviceConnection.root.config
+        if not isinstance(connection, MicroStrategyConnection):
             raise InvalidSourceException(
-                f"Expected MstrConnection, but got {connection}"
+                f"Expected MicroStrategyConnection, but got {connection}"
             )
         return cls(config, metadata)
 
@@ -75,14 +75,18 @@ class MstrSource(DashboardServiceSource):
 
         if self.client.is_project_name():
             project = self.client.get_project_by_name()
-            dashboards.extend(self.client.get_dashboards_list(project.id, project.name))
-
-        if not self.client.is_project_name():
-            for project in self.client.get_projects_list():
+            if project:
                 dashboards.extend(
                     self.client.get_dashboards_list(project.id, project.name)
                 )
 
+        if not self.client.is_project_name():
+            for project in self.client.get_projects_list():
+                if project:
+                    dashboards.extend(
+                        self.client.get_dashboards_list(project.id, project.name)
+                    )
+
         return dashboards
 
     def get_dashboard_name(self, dashboard: MstrDashboard) -> str:
@@ -91,6 +95,18 @@ class MstrSource(DashboardServiceSource):
         """
         return dashboard.name
 
+    def get_project_name(self, dashboard_details: MstrDashboard) -> Optional[str]:
+        """
+        Get dashboard project name
+        """
+        try:
+            return dashboard_details.projectName
+        except Exception as exc:
+            logger.debug(
+                f"Cannot get project name from dashboard [{dashboard_details.name}] - [{exc}]"
+            )
+        return None
+
     def get_dashboard_details(self, dashboard: MstrDashboard) -> MstrDashboardDetails:
         """
         Get Dashboard Details
@@ -106,40 +122,41 @@ class MstrSource(DashboardServiceSource):
         """
         Method to Get Dashboard Entity
         """
-        try:
-            dashboard_url = (
-                f"{clean_uri(self.service_connection.hostPort)}/MicroStrategyLibrary/app/"
-                f"{dashboard_details.projectId}/{dashboard_details.id}"
-            )
-            dashboard_request = CreateDashboardRequest(
-                name=EntityName(dashboard_details.id),
-                displayName=dashboard_details.name,
-                sourceUrl=SourceUrl(dashboard_url),
-                project=dashboard_details.projectName,
-                charts=[
-                    FullyQualifiedEntityName(
-                        fqn.build(
-                            self.metadata,
-                            entity_type=Chart,
-                            service_name=self.context.get().dashboard_service,
-                            chart_name=chart,
+        if dashboard_details:
+            try:
+                dashboard_url = (
+                    f"{clean_uri(str(self.service_connection.hostPort))}/MicroStrategyLibrary/app/"
+                    f"{dashboard_details.projectId}/{dashboard_details.id}"
+                )
+                dashboard_request = CreateDashboardRequest(
+                    name=EntityName(dashboard_details.id),
+                    displayName=dashboard_details.name,
+                    sourceUrl=SourceUrl(dashboard_url),
+                    project=dashboard_details.projectName,
+                    charts=[
+                        FullyQualifiedEntityName(
+                            fqn.build(
+                                self.metadata,
+                                entity_type=Chart,
+                                service_name=self.context.get().dashboard_service,
+                                chart_name=chart,
+                            )
                         )
+                        for chart in self.context.get().charts or []
+                    ],
+                    service=self.context.get().dashboard_service,
+                    owners=self.get_owner_ref(dashboard_details=dashboard_details),
+                )
+                yield Either(right=dashboard_request)
+                self.register_record(dashboard_request=dashboard_request)
+            except Exception as exc:
+                yield Either(
+                    left=StackTraceError(
+                        name=dashboard_details.id,
+                        error=f"Error yielding dashboard for {dashboard_details}: {exc}",
+                        stackTrace=traceback.format_exc(),
                     )
-                    for chart in self.context.get().charts or []
-                ],
-                service=self.context.get().dashboard_service,
-                owners=self.get_owner_ref(dashboard_details=dashboard_details),
-            )
-            yield Either(right=dashboard_request)
-            self.register_record(dashboard_request=dashboard_request)
-        except Exception as exc:
-            yield Either(
-                left=StackTraceError(
-                    name=dashboard_details.id,
-                    error=f"Error yielding dashboard for {dashboard_details}: {exc}",
-                    stackTrace=traceback.format_exc(),
                 )
-            )
 
     def yield_dashboard_lineage_details(
         self, dashboard_details: MstrDashboardDetails, db_service_name: str
@@ -156,14 +173,15 @@ class MstrSource(DashboardServiceSource):
         Returns:
             Iterable[CreateChartRequest]
         """
-        try:
-            for chapter in dashboard_details.chapters:
-                for page in chapter.pages:
-                    yield from self._yield_chart_from_visualization(page)
+        if dashboard_details:
+            try:
+                for chapter in dashboard_details.chapters:
+                    for page in chapter.pages:
+                        yield from self._yield_chart_from_visualization(page)
 
-        except Exception as exc:
-            logger.debug(traceback.format_exc())
-            logger.warning(f"Error creating dashboard: {exc}")
+            except Exception as exc:
+                logger.debug(traceback.format_exc())
+                logger.warning(f"Error creating dashboard: {exc}")
 
     def _yield_chart_from_visualization(
         self, page: MstrPage
@@ -192,3 +210,8 @@ class MstrSource(DashboardServiceSource):
                         stackTrace=traceback.format_exc(),
                     )
                 )
+
+    def close(self):
+        # close the api session
+        self.client.close_api_session()
+        self.metadata.close()

metadata/ingestion/source/dashboard/{mstr → microstrategy}/models.py

@@ -9,10 +9,10 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 """
-MSTR Models
+MicroStrategy Models
 """
 from datetime import datetime
-from typing import List, Optional
+from typing import Any, List, Optional
 
 from pydantic import BaseModel
 
@@ -69,7 +69,8 @@ class MstrSearchResult(BaseModel):
 
 
 class MstrSearchResultList(BaseModel):
-    results: Optional[List[MstrSearchResult]]
+    totalItems: Optional[int] = 0
+    result: Optional[List[MstrSearchResult]] = None
 
 
 class MstrDashboard(BaseModel):
@@ -142,3 +143,8 @@ class MstrDashboardDetails(BaseModel):
     projectName: str
     currentChapter: str
     chapters: List[MstrChapter]
+
+
+class AuthHeaderCookie(BaseModel):
+    auth_header: dict
+    auth_cookies: Any

metadata/ingestion/source/dashboard/microstrategy/service_spec.py

@@ -0,0 +1,6 @@
+from metadata.ingestion.source.dashboard.microstrategy.metadata import (
+    MicrostrategySource,
+)
+from metadata.utils.service_spec.default import DefaultDatabaseSpec
+
+ServiceSpec = DefaultDatabaseSpec(metadata_source_class=MicrostrategySource)

metadata/ingestion/source/dashboard/powerbi/client.py

@@ -64,11 +64,10 @@ class PowerBiApiClient:
             client_credential=self.config.clientSecret.get_secret_value(),
             authority=self.config.authorityURI + self.config.tenantId,
         )
-        self.auth_token = self.get_auth_token()
         client_config = ClientConfig(
             base_url="https://api.powerbi.com",
             api_version="v1.0",
-            auth_token=lambda: self.auth_token,
+            auth_token=self.get_auth_token,
             auth_header="Authorization",
             allow_redirects=True,
             retry_codes=[429],

metadata/ingestion/source/dashboard/superset/db_source.py

@@ -201,7 +201,7 @@ class SupersetDBSource(SupersetSourceMixin):
             except Exception as exc:
                 yield Either(
                     left=StackTraceError(
-                        name=chart_json.id,
+                        name=str(chart_json.id),
                         error=f"Error yielding Chart [{chart_json.id} - {chart_json.slice_name}]: {exc}",
                         stackTrace=traceback.format_exc(),
                     )

metadata/ingestion/source/dashboard/tableau/metadata.py

@@ -69,6 +69,7 @@ from metadata.ingestion.source.dashboard.tableau.models import (
     ChartUrl,
     DataSource,
     DatasourceField,
+    TableAndQuery,
     TableauDashboard,
     TableauTag,
     UpstreamTable,
@@ -149,7 +150,7 @@ class TableauSource(DashboardServiceSource):
         return None
 
     @staticmethod
-    def _get_data_models_tags(dataModels: [DataSource]) -> Set[str]:
+    def _get_data_models_tags(dataModels: List[DataSource]) -> Set[str]:
         """
         Get the tags from the data model in the upstreamDatasources
         """
@@ -382,9 +383,12 @@ class TableauSource(DashboardServiceSource):
                         column=column.id,
                     )
                     for to_column in to_columns:
-                        column_lineage.append(
-                            ColumnLineage(fromColumns=[from_column], toColumn=to_column)
-                        )
+                        if from_column and to_column:
+                            column_lineage.append(
+                                ColumnLineage(
+                                    fromColumns=[from_column], toColumn=to_column
+                                )
+                            )
             return column_lineage
         except Exception as exc:
             logger.debug(f"Error to get column lineage: {exc}")
@@ -451,17 +455,22 @@ class TableauSource(DashboardServiceSource):
                 column.id
                 for field in upstream_data_model.fields
                 for column in field.upstreamColumns
+                if column is not None
             }
             for table in datamodel.upstreamTables or []:
                 om_tables = self._get_database_tables(db_service_entity, table)
-                for om_table in om_tables or []:
+                for om_table_and_query in om_tables or []:
                     column_lineage = self._get_column_lineage(
-                        table, om_table, upstream_data_model_entity, upstream_col_set
+                        table,
+                        om_table_and_query.table,
+                        upstream_data_model_entity,
+                        upstream_col_set,
                     )
                     yield self._get_add_lineage_request(
                         to_entity=upstream_data_model_entity,
-                        from_entity=om_table,
+                        from_entity=om_table_and_query.table,
                         column_lineage=column_lineage,
+                        sql=om_table_and_query.query,
                     )
         except Exception as err:
             yield Either(
@@ -698,7 +707,7 @@ class TableauSource(DashboardServiceSource):
 
     def _get_table_entities_from_api(
         self, db_service_entity: DatabaseService, table: UpstreamTable
-    ) -> Optional[List[Table]]:
+    ) -> Optional[List[TableAndQuery]]:
         """
         In case we get the table details from the Graphql APIs we process them
         """
@@ -734,7 +743,7 @@ class TableauSource(DashboardServiceSource):
                     fqn=table_fqn,
                 )
                 if table_entity:
-                    return [table_entity]
+                    return [TableAndQuery(table=table_entity)]
         except Exception as exc:
             logger.debug(traceback.format_exc())
             logger.warning(f"Error to get tables for lineage using GraphQL Apis: {exc}")
@@ -742,7 +751,7 @@ class TableauSource(DashboardServiceSource):
 
     def _get_table_entities_from_query(
         self, db_service_entity: DatabaseService, table: UpstreamTable
-    ) -> Optional[List[Table]]:
+    ) -> Optional[List[TableAndQuery]]:
         """
         In case we get the table details from the Graphql APIs we process them
         """
@@ -778,7 +787,12 @@ class TableauSource(DashboardServiceSource):
                         database_schema=schema_name,
                         table=table_name,
                     )
-                    tables_list.extend(from_entities)
+                    tables_list.extend(
+                        [
+                            TableAndQuery(table=table, query=custom_sql_table.query)
+                            for table in from_entities
+                        ]
+                    )
 
         except Exception as exc:
             logger.debug(traceback.format_exc())
@@ -787,7 +801,7 @@ class TableauSource(DashboardServiceSource):
 
     def _get_database_tables(
         self, db_service_entity: DatabaseService, table: UpstreamTable
-    ) -> Optional[List[Table]]:
+    ) -> Optional[List[TableAndQuery]]:
         """
         Get the table entities for lineage
         """
@@ -879,6 +893,9 @@ class TableauSource(DashboardServiceSource):
         try:
             return dashboard_details.project.name
         except Exception as exc:
+            logger.info(
+                f"Cannot parse project name for dashboard:{dashboard_details.id} from Tableau server"
+            )
             logger.debug(traceback.format_exc())
             logger.warning(
                 f"Error fetching project name for {dashboard_details.id}: {exc}"

metadata/ingestion/source/dashboard/tableau/models.py

@@ -18,6 +18,7 @@ from typing import Any, Dict, List, Optional, Union
 from pydantic import BaseModel, ConfigDict, Field, validator
 
 from metadata.generated.schema.entity.data.chart import ChartType
+from metadata.generated.schema.entity.data.table import Table
 
 
 class TableauBaseModel(BaseModel):
@@ -172,3 +173,12 @@ class TableauDashboard(TableauBaseModel):
     webpageUrl: Optional[str] = None
     charts: Optional[List[TableauChart]] = None
     dataModels: List[DataSource] = []
+
+
+class TableAndQuery(BaseModel):
+    """
+    Wrapper class for Table entity and associated Query for lineage
+    """
+
+    table: Table
+    query: Optional[str] = None

metadata/ingestion/source/database/athena/metadata.py

@@ -157,6 +157,14 @@ class AthenaSource(ExternalTableLineageMixin, CommonDbSourceService):
             return True, partition_details
         return False, None
 
+    def get_location_path(self, table_name: str, schema_name: str) -> Optional[str]:
+        """
+        Method to fetch the location path of the table
+        """
+        return self.external_location_map.get(
+            (self.context.get().database, schema_name, table_name)
+        )
+
     def yield_tag(
         self, schema_name: str
     ) -> Iterable[Either[OMetaTagAndClassification]]:

metadata/ingestion/source/database/azuresql/service_spec.py

@@ -1,10 +1,12 @@
 from metadata.ingestion.source.database.azuresql.lineage import AzuresqlLineageSource
 from metadata.ingestion.source.database.azuresql.metadata import AzuresqlSource
 from metadata.ingestion.source.database.azuresql.usage import AzuresqlUsageSource
+from metadata.sampler.sqlalchemy.azuresql.sampler import AzureSQLSampler
 from metadata.utils.service_spec.default import DefaultDatabaseSpec
 
 ServiceSpec = DefaultDatabaseSpec(
     metadata_source_class=AzuresqlSource,
     lineage_source_class=AzuresqlLineageSource,
     usage_source_class=AzuresqlUsageSource,
+    sampler_class=AzureSQLSampler,
 )

metadata/ingestion/source/database/bigquery/profiler/profiler.py

@@ -22,7 +22,7 @@ class BigQueryProfiler(BigQueryProfilerInterface):
         **kwargs,
     ) -> List[SystemProfile]:
         return self.system_metrics_computer.get_system_metrics(
-            table=runner.table,
+            table=runner.dataset,
             usage_location=self.service_connection_config.usageLocation,
         )
 

metadata/ingestion/source/database/bigquery/profiler/system.py

@@ -13,6 +13,7 @@ from metadata.profiler.metrics.system.system import (
     SQASessionProvider,
     SystemMetricsComputer,
 )
+from metadata.profiler.processor.runner import QueryRunner
 from metadata.utils.logger import profiler_logger
 from metadata.utils.time_utils import datetime_to_timestamp
 
@@ -25,11 +26,11 @@ class BigQuerySystemMetricsSource(
     """BigQuery system metrics source class"""
 
     def get_kwargs(self, **kwargs):
-        table = kwargs.get("table")
+        runner: QueryRunner = kwargs.get("runner")
         return {
-            "table": table.__table__.name,
-            "dataset_id": table.__table_args__["schema"],
-            "project_id": super().get_session().get_bind().url.host,
+            "table": runner.table_name,
+            "database": runner.session.get_bind().url.database,
+            "schema": runner.schema_name,
             "usage_location": kwargs.get("usage_location"),
         }
 

metadata/ingestion/source/database/bigquery/service_spec.py

@@ -4,6 +4,7 @@ from metadata.ingestion.source.database.bigquery.profiler.profiler import (
     BigQueryProfiler,
 )
 from metadata.ingestion.source.database.bigquery.usage import BigqueryUsageSource
+from metadata.sampler.sqlalchemy.bigquery.sampler import BigQuerySampler
 from metadata.utils.service_spec.default import DefaultDatabaseSpec
 
 ServiceSpec = DefaultDatabaseSpec(
@@ -11,4 +12,5 @@ ServiceSpec = DefaultDatabaseSpec(
     lineage_source_class=BigqueryLineageSource,
     usage_source_class=BigqueryUsageSource,
     profiler_class=BigQueryProfiler,
+    sampler_class=BigQuerySampler,
 )

metadata/ingestion/source/database/common_nosql_source.py

@@ -52,6 +52,7 @@ from metadata.utils.constants import DEFAULT_DATABASE
 from metadata.utils.datalake.datalake_utils import DataFrameColumnParser
 from metadata.utils.filters import filter_by_schema, filter_by_table
 from metadata.utils.logger import ingestion_logger
+from metadata.utils.ssl_manager import check_ssl_and_init
 
 logger = ingestion_logger()
 
@@ -73,7 +74,13 @@ class CommonNoSQLSource(DatabaseServiceSource, ABC):
         )
         self.metadata = metadata
         self.service_connection = self.config.serviceConnection.root.config
+        self.ssl_manager = check_ssl_and_init(self.service_connection)
+        if self.ssl_manager:
+            self.service_connection = self.ssl_manager.setup_ssl(
+                self.service_connection
+            )
         self.connection_obj = get_connection(self.service_connection)
+
         self.test_connection()
 
     def prepare(self):

metadata/ingestion/source/database/databricks/metadata.py

@@ -131,7 +131,7 @@ def _get_table_columns(self, connection, table_name, schema, db_name):
         query = DATABRICKS_GET_TABLE_COMMENTS.format(
             database_name=db_name, schema_name=schema, table_name=table_name
         )
-        cursor = get_table_comment_result(
+        rows = get_table_comment_result(
             self,
             connection=connection,
             query=query,
@@ -140,8 +140,6 @@ def _get_table_columns(self, connection, table_name, schema, db_name):
             schema=schema,
         )
 
-        rows = cursor.fetchall()
-
     except exc.OperationalError as e:
         # Does the table exist?
         regex_fmt = r"TExecuteStatementResp.*SemanticException.*Table not found {}"

metadata/ingestion/source/database/databricks/service_spec.py

@@ -1,3 +1,6 @@
+from metadata.data_quality.interface.sqlalchemy.databricks.test_suite_interface import (
+    DatabricksTestSuiteInterface,
+)
 from metadata.ingestion.source.database.databricks.lineage import (
     DatabricksLineageSource,
 )
@@ -13,4 +16,5 @@ ServiceSpec = DefaultDatabaseSpec(
     lineage_source_class=DatabricksLineageSource,
     usage_source_class=DatabricksUsageSource,
     profiler_class=DatabricksProfilerInterface,
+    test_suite_class=DatabricksTestSuiteInterface,
 )

metadata/ingestion/source/database/datalake/clients/azure_blob.py

@@ -26,9 +26,6 @@ from metadata.utils.constants import DEFAULT_DATABASE
 
 
 class DatalakeAzureBlobClient(DatalakeBaseClient):
-    def __init__(self, client: BlobServiceClient):
-        self._client = client
-
     @classmethod
     def from_config(cls, config: AzureConfig) -> "DatalakeAzureBlobClient":
         try:

metadata/ingestion/source/database/datalake/clients/base.py

@@ -13,10 +13,19 @@
 Datalake Base Client
 """
 from abc import ABC, abstractmethod
-from typing import Callable, Iterable, Optional
+from typing import Any, Callable, Iterable, Optional
 
 
 class DatalakeBaseClient(ABC):
+    """Base DL client implementation"""
+
+    def __init__(self, client: Any, **kwargs):
+        self._client = client
+
+    @property
+    def client(self) -> Any:
+        return self._client
+
     @classmethod
     @abstractmethod
     def from_config(cls, config) -> "DatalakeBaseClient":

metadata/ingestion/source/database/datalake/clients/gcs.py

@@ -33,9 +33,11 @@ from metadata.utils.credentials import GOOGLE_CREDENTIALS, set_google_credential
 
 class DatalakeGcsClient(DatalakeBaseClient):
     def __init__(
-        self, client: storage.Client, temp_credentials_file_path_list: List[str]
+        self,
+        client: storage.Client,
+        temp_credentials_file_path_list: List[str],
     ):
-        self._client = client
+        super().__init__(client=client)
         self._temp_credentials_file_path_list = temp_credentials_file_path_list
 
     @property

metadata/ingestion/source/database/datalake/clients/s3.py

@@ -25,9 +25,6 @@ from metadata.utils.s3_utils import list_s3_objects
 
 
 class DatalakeS3Client(DatalakeBaseClient):
-    def __init__(self, client):
-        self._client = client
-
     @classmethod
     def from_config(cls, config: S3Config) -> "DatalakeS3Client":
         if not config.securityConfig:

metadata/ingestion/source/database/datalake/service_spec.py

@@ -1,9 +1,16 @@
+from metadata.data_quality.interface.pandas.pandas_test_suite_interface import (
+    PandasTestSuiteInterface,
+)
 from metadata.ingestion.source.database.datalake.metadata import DatalakeSource
 from metadata.profiler.interface.pandas.profiler_interface import (
     PandasProfilerInterface,
 )
+from metadata.sampler.pandas.sampler import DatalakeSampler
 from metadata.utils.service_spec.default import DefaultDatabaseSpec
 
 ServiceSpec = DefaultDatabaseSpec(
-    metadata_source_class=DatalakeSource, profiler_class=PandasProfilerInterface
+    metadata_source_class=DatalakeSource,
+    profiler_class=PandasProfilerInterface,
+    test_suite_class=PandasTestSuiteInterface,
+    sampler_class=DatalakeSampler,
 )

metadata/ingestion/source/database/db2/connection.py

@@ -12,6 +12,7 @@
 """
 Source connection handler
 """
+from pathlib import Path
 from typing import Optional
 
 from sqlalchemy.engine import Engine
@@ -32,13 +33,25 @@ from metadata.ingestion.connections.builders import (
 )
 from metadata.ingestion.connections.test_connections import test_connection_db_common
 from metadata.ingestion.ometa.ometa_api import OpenMetadata
-from metadata.utils.constants import THREE_MIN
+from metadata.utils.constants import THREE_MIN, UTF_8
 
 
 def get_connection(connection: Db2Connection) -> Engine:
     """
     Create connection
     """
+    # prepare license
+    # pylint: disable=import-outside-toplevel
+    if connection.license and connection.licenseFileName:
+        import clidriver
+
+        with open(
+            Path(clidriver.__path__[0], "license", connection.licenseFileName),
+            "w",
+            encoding=UTF_8,
+        ) as file:
+            file.write(connection.license)
+
     return create_generic_db_connection(
         connection=connection,
         get_connection_url_fn=get_connection_url_common,