From 9d3a824c3d551d5222dd64a019c5fe68821f0497 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakubjezek001@gmail.com>
Date: Thu, 25 Jul 2024 17:36:30 +0200
Subject: [PATCH] Update get_timeline_media_pool_item function signature and
 add detailed docstring.

- Updated get_timeline_media_pool_item function to accept timeline and root arguments.
- Added detailed docstring specifying the argument types and return type for better clarity.
---
 client/ayon_resolve/api/lib.py                      | 13 +++++++++++--
 .../plugins/load/load_editorial_package.py          | 10 +++-------
 2 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/client/ayon_resolve/api/lib.py b/client/ayon_resolve/api/lib.py
index 807ee60c07..eee403ca1d 100644
--- a/client/ayon_resolve/api/lib.py
+++ b/client/ayon_resolve/api/lib.py
@@ -41,8 +41,17 @@
 self.pype_timeline_name = "OpenPypeTimeline"
 
 
-def get_timeline_media_pool_item(timeline, root=None):
-    """Return MediaPoolItem from Timeline"""
+def get_timeline_media_pool_item(timeline, root=None) -> object:
+    """Return MediaPoolItem from Timeline
+
+
+    Args:
+        timeline (resolve.Timeline): timeline object
+        root (resolve.Folder): root folder / bin object
+
+    Returns:
+        resolve.MediaPoolItem: media pool item from timeline
+    """
 
     # Due to limitations in the Resolve API we can't get
     # the media pool item directly from the timeline.
diff --git a/client/ayon_resolve/plugins/load/load_editorial_package.py b/client/ayon_resolve/plugins/load/load_editorial_package.py
index 013db78350..38905f6d55 100644
--- a/client/ayon_resolve/plugins/load/load_editorial_package.py
+++ b/client/ayon_resolve/plugins/load/load_editorial_package.py
@@ -52,13 +52,9 @@ def load(self, context, name, namespace, data):
         timeline = media_pool.ImportTimelineFromFile(files, import_options)
 
         # get timeline media pool item for metadata update
-        timeline_media_pool_item = None
-        for item in lib.iter_all_media_pool_clips(root=loaded_bin):
-            item_name = item.GetName()
-            if item_name != loaded_timeline_name:
-                continue
-            timeline_media_pool_item = item
-            break
+        timeline_media_pool_item = lib.get_timeline_media_pool_item(
+            timeline, loaded_bin
+        )
 
         # Update the metadata
         clip_data = self._get_container_data(