scripts-blender/addons/asset_pipeline_2/merge/transfer_data/transfer_ui.py

@@ -1,6 +1,6 @@
 import bpy
 from ... import constants
-from ..task_layer import get_local_task_layers
+from ..task_layer import draw_task_layer_selection
 
 
 def draw_transfer_data_type(
@@ -14,27 +14,11 @@ def draw_transfer_data_type(
     box.label(text=name, icon=icon)
     scene = bpy.context.scene
     for transfer_data_item in transfer_data:
-        row = layout.row()
+        row = box.row()
         row.label(text=f"{transfer_data_item.name}: ")
-        if transfer_data_item.owner not in [
-            tl.name for tl in scene.asset_pipeline.local_task_layers
-        ]:
-            row.enabled = False
-            row.prop_search(
-                transfer_data_item,
-                'owner',
-                scene.asset_pipeline,
-                'all_task_layers',
-                text="",
-            )
-        else:
-            row.prop_search(
-                transfer_data_item,
-                'owner',
-                scene.asset_pipeline,
-                'local_task_layers',
-                text="",
-            )
+        draw_task_layer_selection(
+            row, scene, transfer_data_item, transfer_data_item.owner, "owner"
+        )
 
 
 def draw_transfer_data(