Skip to content

Commit ae630f6

Browse files
committed
feat: only write geo node tool settings if tool
1 parent e6ad11c commit ae630f6

File tree

2 files changed

+17
-10
lines changed

2 files changed

+17
-10
lines changed

geometry/operator.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -119,15 +119,22 @@ def _process_node_tree(self, node_tree: GeometryNodeTree,
119119
self._write("\n")
120120

121121
if bpy.app.version >= (4, 0, 0):
122-
geo_node_tree_flags = ["is_mode_edit",
123-
"is_mode_sculpt",
124-
"is_modifier",
125-
"is_tool",
126-
"is_type_curve",
127-
"is_type_mesh",
128-
"is_type_point_cloud"]
129-
for flag in geo_node_tree_flags:
130-
self._write(f"{inner}{nt_var}.{flag} = {getattr(node_tree, flag)}\n")
122+
is_mod = node_tree.is_modifier
123+
is_tool = node_tree.is_tool
124+
if is_mod:
125+
self._write(f"{inner}{nt_var}.is_modifier = True\n")
126+
if is_tool:
127+
self._write(f"{inner}{nt_var}.is_tool = True\n")
128+
129+
tool_flags = ["is_mode_edit",
130+
"is_mode_sculpt",
131+
"is_type_curve",
132+
"is_type_mesh",
133+
"is_type_point_cloud"]
134+
135+
for flag in tool_flags:
136+
self._write(f"{inner}{nt_var}.{flag} = "
137+
f"{getattr(node_tree, flag)}\n")
131138
self._write("\n")
132139

133140
#initialize nodes

utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def clean_string(string: str, lower: bool = True) -> str:
6262
string (str): The input string
6363
6464
Returns:
65-
string (str): The input string ready to be used as a variable
65+
string (str): The input string ready to be used as a variable/file
6666
"""
6767

6868
if lower:

0 commit comments

Comments
 (0)