acryl-datahub 0.15.0.5rc8__py3-none-any.whl → 0.15.0.5rc10__py3-none-any.whl

datahub/metadata/schemas/DashboardInfo.avsc

@@ -258,16 +258,16 @@
     {
       "Relationship": {
         "/*/destinationUrn": {
-          "createdActor": "datasetEdges/*/created/actor",
-          "createdOn": "datasetEdges/*/created/time",
+          "createdActor": "dashboards/*/created/actor",
+          "createdOn": "dashboards/*/created/time",
           "entityTypes": [
             "dashboard"
           ],
           "isLineage": true,
           "name": "DashboardContainsDashboard",
-          "properties": "datasetEdges/*/properties",
-          "updatedActor": "datasetEdges/*/lastModified/actor",
-          "updatedOn": "datasetEdges/*/lastModified/time"
+          "properties": "dashboards/*/properties",
+          "updatedActor": "dashboards/*/lastModified/actor",
+          "updatedOn": "dashboards/*/lastModified/time"
         }
       },
       "type": {

datahub/metadata/schemas/MetadataChangeEvent.avsc

@@ -2049,16 +2049,16 @@
                       {
                         "Relationship": {
                           "/*/destinationUrn": {
-                            "createdActor": "datasetEdges/*/created/actor",
-                            "createdOn": "datasetEdges/*/created/time",
+                            "createdActor": "dashboards/*/created/actor",
+                            "createdOn": "dashboards/*/created/time",
                             "entityTypes": [
                               "dashboard"
                             ],
                             "isLineage": true,
                             "name": "DashboardContainsDashboard",
-                            "properties": "datasetEdges/*/properties",
-                            "updatedActor": "datasetEdges/*/lastModified/actor",
-                            "updatedOn": "datasetEdges/*/lastModified/time"
+                            "properties": "dashboards/*/properties",
+                            "updatedActor": "dashboards/*/lastModified/actor",
+                            "updatedOn": "dashboards/*/lastModified/time"
                           }
                         },
                         "type": {

datahub/specific/dashboard.py

@@ -161,7 +161,7 @@ class DashboardPatchBuilder(
                 lastModified=self._mint_auditstamp(),
             )
 
-        self._ensure_urn_type("dataset", [chart_edge], "add_chart_edge")
+        self._ensure_urn_type("chart", [chart_edge], "add_chart_edge")
         self._add_patch(
             DashboardInfo.ASPECT_NAME,
             "add",
@@ -271,6 +271,48 @@ class DashboardPatchBuilder(
 
         return self
 
+    def add_dashboard(
+        self, dashboard: Union[Edge, Urn, str]
+    ) -> "DashboardPatchBuilder":
+        """
+        Adds an dashboard to the DashboardPatchBuilder.
+
+        Args:
+            dashboard: The dashboard, which can be an Edge object, Urn object, or a string.
+
+        Returns:
+            The DashboardPatchBuilder instance.
+
+        Raises:
+            ValueError: If the dashboard is not a Dashboard urn.
+
+        Notes:
+            If `dashboard` is an Edge object, it is used directly. If `dashboard` is a Urn object or string,
+            it is converted to an Edge object and added with default audit stamps.
+        """
+        if isinstance(dashboard, Edge):
+            dashboard_urn: str = dashboard.destinationUrn
+            dashboard_edge: Edge = dashboard
+        elif isinstance(dashboard, (Urn, str)):
+            dashboard_urn = str(dashboard)
+            if not dashboard_urn.startswith("urn:li:dashboard:"):
+                raise ValueError(f"Input {dashboard} is not a Dashboard urn")
+
+            dashboard_edge = Edge(
+                destinationUrn=dashboard_urn,
+                created=self._mint_auditstamp(),
+                lastModified=self._mint_auditstamp(),
+            )
+
+        self._ensure_urn_type("dashboard", [dashboard_edge], "add_dashboard")
+        self._add_patch(
+            DashboardInfo.ASPECT_NAME,
+            "add",
+            path=("dashboards", dashboard_urn),
+            value=dashboard_edge,
+        )
+        return self
+
     def set_dashboard_url(
         self, dashboard_url: Optional[str]
     ) -> "DashboardPatchBuilder":

datahub/upgrade/upgrade.py

@@ -55,11 +55,19 @@ async def get_client_version_stats():
         async with session.get(pypi_url) as resp:
             response_json = await resp.json()
             try:
-                releases = response_json.get("releases", [])
-                sorted_releases = sorted(releases.keys(), key=lambda x: Version(x))
-                latest_cli_release_string = [
-                    x for x in sorted_releases if "rc" not in x
-                ][-1]
+                releases = response_json.get("releases", {})
+                filtered_releases = {
+                    version: release_files
+                    for version, release_files in releases.items()
+                    if not all(
+                        release_file.get("yanked") for release_file in release_files
+                    )
+                    and "rc" not in version
+                }
+                sorted_releases = sorted(
+                    filtered_releases.keys(), key=lambda x: Version(x)
+                )
+                latest_cli_release_string = sorted_releases[-1]
                 latest_cli_release = Version(latest_cli_release_string)
                 current_version_info = releases.get(current_version_string)
                 current_version_date = None