openmetadata-ingestion 1.5.13.1__py3-none-any.whl → 1.5.13.2__py3-none-any.whl

metadata/generated/schema/type/entityUsage.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityUsage.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/filterPattern.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/filterPattern.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/function.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/function.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/include.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/include.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/jdbcConnection.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/jdbcConnection.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/lifeCycle.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/lifeCycle.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/paging.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/paging.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/profile.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/profile.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/queryParserData.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/queryParserData.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/reaction.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/reaction.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/schedule.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/schedule.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/schema.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/schema.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/tableQuery.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/tableQuery.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/tableUsageCount.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/tableUsageCount.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/tagLabel.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/tagLabel.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/usageDetails.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/usageDetails.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/usageRequest.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/usageRequest.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/votes.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/votes.json
-#   timestamp: 2024-12-03T13:40:51+00:00
+#   timestamp: 2024-12-04T09:51:40+00:00
 
 from __future__ import annotations
 

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

@@ -37,7 +37,8 @@ from metadata.utils.logger import ingestion_logger
 logger = ingestion_logger()
 
 API_VERSION = "MicroStrategyLibrary/api"
-LOGIN_MODE_GUEST = 1
+HEADERS = {"Content-Type": "application/json"}
+LOGIN_MODE = 1
 APPLICATION_TYPE = 35
 
 
@@ -57,60 +58,71 @@ class MSTRClient:
     ):
         self.config = config
 
-        self.auth_params: AuthHeaderCookie = self._get_auth_header_and_cookies()
+        self.auth_params = self._get_auth_header_and_cookies()
 
         client_config = ClientConfig(
             base_url=clean_uri(config.hostPort),
             api_version=API_VERSION,
-            extra_headers=self.auth_params.auth_header,
+            extra_headers=self.auth_params.auth_header if self.auth_params else None,
             allow_redirects=True,
-            cookies=self.auth_params.auth_cookies,
+            cookies=self.auth_params.auth_cookies if self.auth_params else None,
         )
 
         self.client = REST(client_config)
-        self._set_api_session()
 
-    def _get_auth_header_and_cookies(self) -> Optional[AuthHeaderCookie]:
+    def get_auth_params(self) -> AuthHeaderCookie:
         """
-        Send a request to authenticate the user and get headers and
-
-        To know about the data params below please visit
-        https://demo.microstrategy.com/MicroStrategyLibrary/api-docs/index.html#/Authentication/postLogin
+        Test whether we can fetch auth_token from the api
         """
         data = {
             "username": self.config.username,
             "password": self.config.password.get_secret_value(),
-            "loginMode": LOGIN_MODE_GUEST,
+            "loginMode": LOGIN_MODE,
             "applicationType": APPLICATION_TYPE,
         }
         response = requests.post(
-            url=self._get_base_url("auth/login"), data=data, timeout=60
+            url=self._get_base_url("auth/login"), json=data, headers=HEADERS, timeout=60
         )
-        if not response.status_code == 204:
-            logger.error(
-                f"Auth Failed with login mode: {LOGIN_MODE_GUEST}, Error: {response.text}"
+        response.raise_for_status()
+        if (
+            not response.ok
+            or response.status_code != 204
+            or "X-MSTR-AuthToken" not in response.headers
+        ):
+            raise SourceConnectionException(
+                f"Failed to Fetch Token, please validate your credentials and login_mode : {response.text}"
             )
-            logger.error("Trying with login mode: 8")
-            data["loginMode"] = 8
-            response = requests.post(
-                url=self._get_base_url("auth/login"), data=data, timeout=60
-            )
-            if not response.status_code == 204:
-                raise SourceConnectionException(
-                    f"Auth Failed with login mode: 8 and {LOGIN_MODE_GUEST}, Error: {response.text}"
-                )
         return AuthHeaderCookie(
             auth_header=response.headers, auth_cookies=response.cookies
         )
 
-    def _set_api_session(self) -> bool:
+    def _get_auth_header_and_cookies(self) -> Optional[AuthHeaderCookie]:
+        """
+        Send a request to authenticate the user and get headers and
+
+        To know about the data params below please visit
+        https://demo.microstrategy.com/MicroStrategyLibrary/api-docs/index.html#/Authentication/postLogin
+        """
+        try:
+            auth_data = self.get_auth_params()
+            if auth_data:
+                self._set_api_session(auth_data)
+                return auth_data
+        except Exception as exc:
+            logger.debug(traceback.format_exc())
+            logger.error(
+                f"Failed to fetch the auth header and cookies due to : [{exc}]"
+            )
+        return None
+
+    def _set_api_session(self, auth_data: AuthHeaderCookie) -> bool:
         """
         Set the user api session to active this will keep the connection alive
         """
         api_session = requests.put(
             url=self._get_base_url("sessions"),
-            headers=self.auth_params.auth_header,
-            cookies=self.auth_params.auth_cookies,
+            headers=auth_data.auth_header,
+            cookies=auth_data.auth_cookies,
             timeout=60,
         )
         if api_session.ok:
@@ -140,24 +152,27 @@ class MSTRClient:
     def is_project_name(self) -> bool:
         return bool(self.config.projectName)
 
+    def list_projects(self) -> List[MstrProject]:
+        """
+        Test if we can get the token and fetch projects
+        """
+        self.get_auth_params()
+        resp_projects = self.client.get(
+            path="/projects",
+        )
+        return resp_projects
+
     def get_projects_list(self) -> List[MstrProject]:
         """
         Get List of all projects
         """
         try:
-            resp_projects = self.client.get(
-                path="/projects",
-            )
-
-            project_list = MstrProjectList(projects=resp_projects)
+            project_list = MstrProjectList(projects=self.list_projects())
             return project_list.projects
-
         except Exception as exc:
             logger.debug(traceback.format_exc())
             logger.warning(f"Failed to fetch the project list due to [{exc}]")
 
-        return []
-
     def get_project_by_name(self) -> Optional[MstrProject]:
         """
         Get Project By Name

metadata/ingestion/source/dashboard/mstr/connection.py

@@ -48,7 +48,9 @@ def test_connection(
     of a metadata workflow or during an Automation Workflow
     """
 
-    test_fn = {"GetProjects": client.get_projects_list}
+    test_fn = {
+        "GetProjects": client.list_projects,
+    }
 
     return test_connection_steps(
         metadata=metadata,