From cdf32eb15051b75b17ed2187daefabc0a5fff43a Mon Sep 17 00:00:00 2001
From: Toke Stuart Jepsen <tokejepsen@gmail.com>
Date: Wed, 26 Aug 2020 10:16:20 +0100
Subject: [PATCH] Containerize audio loading.

---
 pype/plugins/maya/load/load_audio.py | 49 +++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

diff --git a/pype/plugins/maya/load/load_audio.py b/pype/plugins/maya/load/load_audio.py
index e1860d0ca6..ca38082ed0 100644
--- a/pype/plugins/maya/load/load_audio.py
+++ b/pype/plugins/maya/load/load_audio.py
@@ -1,6 +1,9 @@
 from maya import cmds, mel
+import pymel.core as pc
 
 from avalon import api
+from avalon.maya.pipeline import containerise
+from avalon.maya import lib
 
 
 class AudioLoader(api.Loader):
@@ -24,4 +27,48 @@ class AudioLoader(api.Loader):
             displaySound=True
         )
 
-        return [sound_node]
+        asset = context["asset"]["name"]
+        namespace = namespace or lib.unique_namespace(
+            asset + "_",
+            prefix="_" if asset[0].isdigit() else "",
+            suffix="_",
+        )
+
+        return containerise(
+            name=name,
+            namespace=namespace,
+            nodes=[sound_node],
+            context=context,
+            loader=self.__class__.__name__
+        )
+
+    def update(self, container, representation):
+        audio_node = None
+        for node in pc.PyNode(container["objectName"]).members():
+            if node.nodeType() == "audio":
+                audio_node = node
+
+        assert audio_node is not None, "Audio node not found."
+
+        path = api.get_representation_path(representation)
+        audio_node.filename.set(path)
+        cmds.setAttr(
+            container["objectName"] + ".representation",
+            str(representation["_id"]),
+            type="string"
+        )
+
+    def switch(self, container, representation):
+        self.update(container, representation)
+
+    def remove(self, container):
+        members = cmds.sets(container['objectName'], query=True)
+        cmds.lockNode(members, lock=False)
+        cmds.delete([container['objectName']] + members)
+
+        # Clean up the namespace
+        try:
+            cmds.namespace(removeNamespace=container['namespace'],
+                           deleteNamespaceContent=True)
+        except RuntimeError:
+            pass