From db3f5c60c68e47b210dcd644ea69f754c554b8ee Mon Sep 17 00:00:00 2001
From: Jakub Trllo <43494761+iLLiCiTiT@users.noreply.github.com>
Date: Mon, 24 Jun 2024 17:38:18 +0200
Subject: [PATCH] implemented logic to update to latest approved

---
 client/ayon_core/tools/sceneinventory/view.py | 84 ++++++++++++++++++-
 1 file changed, 82 insertions(+), 2 deletions(-)

diff --git a/client/ayon_core/tools/sceneinventory/view.py b/client/ayon_core/tools/sceneinventory/view.py
index c8cc3299a2..017a9597f1 100644
--- a/client/ayon_core/tools/sceneinventory/view.py
+++ b/client/ayon_core/tools/sceneinventory/view.py
@@ -233,10 +233,18 @@ def _build_item_menu_for_selection(self, menu, indexes, active_index):
         has_outdated = False
         has_loaded_hero_versions = False
         has_available_hero_version = False
-        for version_items_by_id in version_items_by_product_id.values():
+        has_outdated_approved = False
+        last_version_by_product_id = {}
+        for product_id, version_items_by_id in (
+            version_items_by_product_id.items()
+        ):
+            _has_outdated_approved = False
+            _last_approved_version_item = None
             for version_item in version_items_by_id.values():
                 if version_item.is_hero:
                     has_available_hero_version = True
+                if version_item.is_last_approved:
+                    _last_approved_version_item = version_item
 
                 if version_item.version_id not in version_ids:
                     continue
@@ -245,6 +253,17 @@ def _build_item_menu_for_selection(self, menu, indexes, active_index):
 
                 elif not version_item.is_latest:
                     has_outdated = True
+                elif not version_item.is_last_approved:
+                    _has_outdated_approved = True
+
+            if (
+                _has_outdated_approved
+                and _last_approved_version_item is not None
+            ):
+                last_version_by_product_id[product_id] = (
+                    _last_approved_version_item
+                )
+                has_outdated_approved = True
 
         switch_to_versioned = None
         if has_loaded_hero_versions:
@@ -261,6 +280,41 @@ def _build_item_menu_for_selection(self, menu, indexes, active_index):
                 lambda: self._on_switch_to_versioned(item_ids)
             )
 
+        update_to_last_approved_action = None
+        if has_outdated_approved:
+            approved_version_by_item_id = {}
+            for container_item in container_items_by_id.values():
+                repre_id = container_item.representation_id
+                repre_info = repre_info_by_id.get(repre_id)
+                if not repre_info or not repre_info.is_valid:
+                    continue
+                version_item = last_version_by_product_id.get(
+                    repre_info.product_id
+                )
+                if (
+                    version_item is None
+                    or version_item.id == repre_info.version_id
+                ):
+                    continue
+                approved_version_by_item_id[container_item.item_id] = (
+                    version_item.version
+                )
+
+            update_icon = qtawesome.icon(
+                "fa.angle-double-up",
+                color="#00f0b4"
+            )
+            update_to_last_approved_action = QtWidgets.QAction(
+                update_icon,
+                "Update to last approved",
+                menu
+            )
+            update_to_last_approved_action.triggered.connect(
+                lambda: self._update_containers_to_approved_versions(
+                    approved_version_by_item_id
+                )
+            )
+
         update_to_latest_action = None
         if has_outdated or has_loaded_hero_versions:
             update_icon = qtawesome.icon(
@@ -299,7 +353,9 @@ def _build_item_menu_for_selection(self, menu, indexes, active_index):
         # set version
         set_version_action = None
         if active_repre_id is not None:
-            set_version_icon = qtawesome.icon("fa.hashtag", color=DEFAULT_COLOR)
+            set_version_icon = qtawesome.icon(
+                "fa.hashtag", color=DEFAULT_COLOR
+            )
             set_version_action = QtWidgets.QAction(
                 set_version_icon,
                 "Set version",
@@ -323,6 +379,9 @@ def _build_item_menu_for_selection(self, menu, indexes, active_index):
         if switch_to_versioned:
             menu.addAction(switch_to_versioned)
 
+        if update_to_last_approved_action:
+            menu.addAction(update_to_last_approved_action)
+
         if update_to_latest_action:
             menu.addAction(update_to_latest_action)
 
@@ -970,3 +1029,24 @@ def _update_containers_to_version(self, item_ids, version):
         """
         versions = [version for _ in range(len(item_ids))]
         self._update_containers(item_ids, versions)
+
+    def _update_containers_to_approved_versions(
+        self, approved_version_by_item_id
+    ):
+        """Helper to update items to given version (or version per item)
+
+        If at least one item is specified this will always try to refresh
+        the inventory even if errors occurred on any of the items.
+
+        Arguments:
+            approved_version_by_item_id (Dict[str, int]): Version to set by
+                item id.
+
+        """
+        versions = []
+        item_ids = []
+        for item_id, version in approved_version_by_item_id.items():
+            item_ids.append(item_id)
+            versions.append(version)
+
+        self._update_containers(item_ids, versions)