From ea0bf51ced259daba0c2fa7fef340d72fb8574da Mon Sep 17 00:00:00 2001 From: Roy Nieterau Date: Wed, 6 Mar 2024 22:18:53 +0100 Subject: [PATCH] Add repair action for edit mode validation --- .../zbrush/plugins/publish/validate_edit_mode.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/client/ayon_core/hosts/zbrush/plugins/publish/validate_edit_mode.py b/client/ayon_core/hosts/zbrush/plugins/publish/validate_edit_mode.py index 7f8e387c30..12bab992b9 100644 --- a/client/ayon_core/hosts/zbrush/plugins/publish/validate_edit_mode.py +++ b/client/ayon_core/hosts/zbrush/plugins/publish/validate_edit_mode.py @@ -3,9 +3,10 @@ import pyblish.api from ayon_core.pipeline.publish import ( OptionalPyblishPluginMixin, ValidateContentsOrder, - PublishValidationError + PublishValidationError, + RepairContextAction ) -from ayon_core.hosts.zbrush.api.lib import is_in_edit_mode +from ayon_core.hosts.zbrush.api.lib import is_in_edit_mode, execute_zscript class ValidateEditMode(pyblish.api.ContextPlugin, @@ -19,6 +20,7 @@ class ValidateEditMode(pyblish.api.ContextPlugin, families = ["model"] hosts = ["zbrush"] optional = True + actions = [RepairContextAction] def process(self, context): edit_mode = is_in_edit_mode() @@ -27,3 +29,8 @@ class ValidateEditMode(pyblish.api.ContextPlugin, "Zbrush is not in edit mode, " "please make sure it is in edit mode before extraction." ) + + @classmethod + def repair(cls, context): + # Enable Transform:Edit state + execute_zscript("[ISet, Transform:Edit, 1]")