Merge pull request #43 from pypeclub/bug/PYPE-743_yeti-cache-versions

PYPE-743: fixing yeti cache version updates

pype/plugins/maya/load/load_yeti_cache.py

@@ -6,10 +6,11 @@ from collections import defaultdict
 
 from maya import cmds
 
-from avalon import api
+from avalon import api, io
 from avalon.maya import lib as avalon_lib, pipeline
 from pype.maya import lib
 from pypeapp import config
+from pprint import pprint
 
 
 class YetiCacheLoader(api.Loader):
@@ -101,12 +102,23 @@ class YetiCacheLoader(api.Loader):
 
     def update(self, container, representation):
 
+        io.install()
         namespace = container["namespace"]
         container_node = container["objectName"]
+
+        fur_settings = io.find_one(
+            {"parent": representation["parent"], "name": "fursettings"}
+        )
+
+        pprint({"parent": representation["parent"], "name": "fursettings"})
+        pprint(fur_settings)
+        assert fur_settings is not None, (
+            "cannot find fursettings representation"
+        )
+
+        settings_fname = api.get_representation_path(fur_settings)
         path = api.get_representation_path(representation)
         # Get all node data
-        fname, ext = os.path.splitext(path)
-        settings_fname = "{}.fursettings".format(fname)
         with open(settings_fname, "r") as fp:
             settings = json.load(fp)
 
@@ -147,13 +159,14 @@ class YetiCacheLoader(api.Loader):
 
             cmds.delete(to_remove)
 
+        # replace frame in filename with %04d
+        RE_frame = re.compile(r"(\d+)(\.fur)$")
+        file_name = re.sub(RE_frame, r"%04d\g<2>", os.path.basename(path))
         for cb_id, data in meta_data_lookup.items():
 
             # Update cache file name
-            file_name = data["name"].replace(":", "_")
-            cache_file_path = "{}.%04d.fur".format(file_name)
             data["attrs"]["cacheFileName"] = os.path.join(
-                path, cache_file_path)
+                os.path.dirname(path), file_name)
 
             if cb_id not in scene_lookup:
 
@@ -197,6 +210,12 @@ class YetiCacheLoader(api.Loader):
                     yeti_node = yeti_nodes[0]
 
                     for attr, value in data["attrs"].items():
+                        # handle empty attribute strings. Those are reported
+                        # as None, so their type is NoneType and this is not
+                        # supported on attributes in Maya. We change it to
+                        # empty string.
+                        if value is None:
+                            value = ""
                         lib.set_attribute(attr, value, yeti_node)
 
         cmds.setAttr("{}.representation".format(container_node),

pype/plugins/maya/publish/collect_yeti_cache.py

@@ -49,6 +49,10 @@ class CollectYetiCache(pyblish.api.InstancePlugin):
             attr_data = {}
             for attr in SETTINGS:
                 current = cmds.getAttr("%s.%s" % (shape, attr))
+                # change None to empty string as Maya doesn't support
+                # NoneType in attributes
+                if current is None:
+                    current = ""
                 attr_data[attr] = current
 
             # Get transform data