8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122 | class CreateWorkfile(plugin.HoudiniCreatorBase, AutoCreator):
"""Workfile auto-creator."""
settings_category = "houdini"
identifier = "io.openpype.creators.houdini.workfile"
label = "Workfile"
product_type = "workfile"
icon = "fa5.file"
default_variant = "Main"
is_mandatory = False
def create(self):
variant = self.default_variant
current_instance = next(
(
instance for instance in self.create_context.instances
if instance.creator_identifier == self.identifier
), None)
project_entity = self.create_context.get_current_project_entity()
folder_entity = self.create_context.get_current_folder_entity()
task_entity = self.create_context.get_current_task_entity()
project_name = project_entity["name"]
folder_path = folder_entity["path"]
task_name = task_entity["name"]
host_name = self.create_context.host_name
if current_instance is None:
product_name = self.get_product_name(
project_name=project_name,
project_entity=project_entity,
folder_entity=folder_entity,
task_entity=task_entity,
variant=variant,
host_name=host_name,
)
data = {
"folderPath": folder_path,
"task": task_name,
"variant": variant,
}
data.update(
self.get_dynamic_data(
project_name,
folder_entity,
task_entity,
variant,
host_name,
current_instance)
)
self.log.info("Auto-creating workfile instance...")
current_instance = CreatedInstance(
self.product_type, product_name, data, self
)
self._add_instance_to_context(current_instance)
elif (
current_instance["folderPath"] != folder_path
or current_instance["task"] != task_name
):
# Update instance context if is not the same
product_name = self.get_product_name(
project_name=project_name,
project_entity=project_entity,
folder_entity=folder_entity,
task_entity=task_entity,
variant=variant,
host_name=host_name,
)
current_instance["folderPath"] = folder_path
current_instance["task"] = task_name
current_instance["productName"] = product_name
# write workfile information to context container.
context_node = self.host.get_context_node()
if not context_node:
context_node = self.host.create_context_node()
if hasattr(current_instance, "set_mandatory"):
current_instance.set_mandatory(self.is_mandatory)
workfile_data = {"workfile": current_instance.data_to_store()}
imprint(context_node, workfile_data)
def collect_instances(self):
context_node = self.host.get_context_node()
if not context_node:
return
instance = read(context_node)
if not instance:
return
workfile = instance.get("workfile")
if not workfile:
return
# Convert legacy creator_identifier
creator_identifier = workfile.get("creator_identifier")
if creator_identifier:
workfile["creator_identifier"] = (
plugin.REMAP_CREATOR_IDENTIFIERS.get(creator_identifier,
creator_identifier)
)
created_instance = CreatedInstance.from_existing(
workfile, self
)
self._add_instance_to_context(created_instance)
def update_instances(self, update_list):
context_node = self.host.get_context_node()
for created_inst, _changes in update_list:
if created_inst["creator_identifier"] == self.identifier:
workfile_data = {"workfile": created_inst.data_to_store()}
imprint(context_node, workfile_data, update=True)
|