Skip to content

Commit 3b00ac8

Browse files
nuttinutti
committed
WIP
1 parent 97b2393 commit 3b00ac8

1 file changed

Lines changed: 55 additions & 56 deletions

File tree

‎src/gen_modfile/gen_node_inputs_outputs_for_node_class.py‎

Lines changed: 55 additions & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ class GenerationConfig:
4545
output_format = "rst"
4646

4747

48+
# pylint: disable=C0209
4849
def analyze() -> Dict:
4950
results = {
5051
"new": [],
@@ -98,37 +99,41 @@ def analyze() -> Dict:
9899
"description": None,
99100
"type": "method",
100101
"parameters": ["key"],
101-
"parameter_details": [{
102-
"name": "key",
103-
"description": None,
104-
"data_type": "typing.Literal[{}] | typing.Literal[\"{}\"]"
105-
.format(i, node_input.name),
106-
"mod_option": "skip-refine",
107-
}],
108-
"return": {
109-
"type": "return",
110-
"description": None,
111-
"data_type": ":class:`bpy.types.{}`"
112-
.format(node_input.__class__.__name__)
113-
},
102+
"parameter_details": [],
103+
"return": {},
114104
"options": ["overload"],
115105
}
106+
new_func_def["parameter_details"].append({
107+
"name": "key",
108+
"description": None,
109+
"data_type": 'typing.Literal[{}] | typing.Literal["{}"]'
110+
.format(i, node_input.name)),
111+
"mod_option": "skip-refine",
112+
})
113+
dtype = ":class:`bpy.types.{}`".format(
114+
node_input.__class__.__name__)
115+
new_func_def["return"] = {
116+
"type": "return",
117+
"description": None,
118+
"data_type": ":class:`bpy.types.{}`"
119+
.format(node_input.__class__.__name__),
120+
}
116121
new_class_def["methods"].append(new_func_def)
117122

118123
results["new"].append(new_class_def)
119124

120125
# Add attribute for input.
121-
append_class_def["attributes"].append(
122-
{
123-
"type": "attribute",
124-
"name": "inputs",
125-
"description": None,
126-
"class": node_class_name,
127-
"module": "bpy.types",
128-
"data_type": ":class:`bpy.types.{}`".format(fake_class_name),
129-
"mod_option": "skip-refine",
130-
}
131-
)
126+
new_attribute = {
127+
"type": "attribute",
128+
"name": "inputs",
129+
"description": None,
130+
"class": node_class_name,
131+
"module": "bpy.types",
132+
"data_type": ":class:`bpy.types.{}`"
133+
.format(fake_class_name),
134+
"mod_option": "skip-refine",
135+
}
136+
append_class_def["attributes"].append(new_attribute)
132137
results["append"].append(append_class_def)
133138

134139
if node.outputs:
@@ -150,50 +155,44 @@ def analyze() -> Dict:
150155
"description": None,
151156
"type": "method",
152157
"parameters": ["key"],
153-
"parameter_details": [
154-
{
155-
"name": "key",
156-
"description": None,
157-
"data_type": "typing.Literal[{}] | typing.Literal[\"{}\"]"
158-
.format(i, node_output.name),
159-
"mod_option": "skip-refine",
160-
}
161-
],
162-
"return": {
163-
"type": "return",
164-
"description": None,
165-
"data_type": ":class:`bpy.types.{}`"
166-
.format(node_output.__class__.__name__)
167-
},
158+
"parameter_details": [],
159+
"return": {},
168160
"options": ["overload"],
169161
}
162+
new_func_def["parameter_details"].append({
163+
"name": "key",
164+
"description": None,
165+
"data_type": 'typing.Literal[{}] | typing.Literal["{}"]'
166+
.format(i, node_output.name),
167+
"mod_option": "skip-refine",
168+
})
169+
new_func_def["return"] = {
170+
"type": "return",
171+
"description": None,
172+
"data_type": ":class:`bpy.types.{}`"
173+
.format(node_output.__class__.__name__)
174+
}
170175
new_class_def["methods"].append(new_func_def)
171176

172177
results["new"].append(new_class_def)
173178

174179
# Add attribute for output.
175-
append_class_def["attributes"].append(
176-
{
177-
"type": "attribute",
178-
"name": "outputs",
179-
"description": None,
180-
"class": node_class_name,
181-
"module": "bpy.types",
182-
"data_type": ":class:`bpy.types.{}`".format(fake_class_name),
183-
}
184-
)
180+
new_attribute_def = {
181+
"type": "attribute",
182+
"name": "outputs",
183+
"description": None,
184+
"class": node_class_name,
185+
"module": "bpy.types",
186+
"data_type": ":class:`bpy.types.{}`"
187+
.format(fake_class_name),
188+
}
189+
append_class_def["attributes"].append(new_attribute_def)
185190
results["append"].append(append_class_def)
186191

187192
return results
188193

189194

190-
def write_description(f: TextIO, description: str, indent: str) -> None:
191-
lines = description.split("\n")
192-
for line in lines: # noqa: FURB122
193-
f.write("{}{}\n".format(indent, line))
194-
f.write("\n")
195-
196-
195+
# pylint: disable=C0209
197196
def write_class_info(f: TextIO, mod_kind: str, class_info: Dict) -> None:
198197
f.write(".. mod-type:: {}\n\n".format(mod_kind))
199198
f.write(".. module:: {}\n\n".format(class_info["module"]))

0 commit comments

Comments
 (0)