Skip to content

Commit e95a12e

Browse files
committed
Add aliases to operator signature
1 parent 1537c46 commit e95a12e

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

src/ansys/dpf/core/operators/build.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@ def build_pin_data(pins, output=False):
102102
"document_pin_docstring": document_pin_docstring,
103103
"ellipsis": 0 if specification.ellipsis else -1,
104104
"has_aliases": len(specification.aliases) > 0,
105+
"aliases_list": [dict(("alias", alias)) for alias in specification.aliases],
105106
"aliases": str(specification.aliases),
106107
}
107108

@@ -129,13 +130,13 @@ def build_operator(
129130
input_pins = []
130131
if specification.inputs:
131132
input_pins = build_pin_data(specification.inputs)
132-
has_input_aliases = any(len(pin["aliases"]) > len("[]") for pin in input_pins)
133+
has_input_aliases = any(len(pin["aliases_list"]) > 0 for pin in input_pins)
133134

134135
output_pins = []
135136
if specification.outputs:
136137
output_pins = build_pin_data(specification.outputs, output=True)
137138
multiple_output_types = any(pin["multiple_types"] for pin in output_pins)
138-
has_output_aliases = any(len(pin["aliases"]) > len("[]") for pin in output_pins)
139+
has_output_aliases = any(len(pin["aliases_list"]) > 0 for pin in output_pins)
139140

140141
docstring = build_docstring(specification_description)
141142

src/ansys/dpf/core/operators/operator.mustache

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,13 +82,19 @@ class {{class_name}}(Operator):
8282
{{/outputs}}
8383
"""
8484

85-
def __init__(self, {{#input_pins}}{{name}}=None, {{/input_pins}}config=None, server=None):
85+
def __init__(self, {{#input_pins}}{{name}}=None, {{/input_pins}}config=None, server=None{{#has_input_aliases}}, {{#input_pins}}{{#aliases_list}}{{alias}}=None, {{/aliases_list}}{{/input_pins}}{{/has_input_aliases}}):
8686
super().__init__(name="{{operator_name}}", config=config, server=server)
8787
self._inputs = Inputs{{capital_class_name}}(self)
8888
self._outputs = Outputs{{capital_class_name}}(self)
8989
{{#input_pins}}
9090
if {{name}} is not None:
9191
self.inputs.{{name}}.connect({{name}})
92+
{{#has_input_aliases}}
93+
{{#aliases_list}}
94+
elif {{alias}} is not None:
95+
self.inputs.{{name}}.connect({{alias}})
96+
{{/aliases_list}}
97+
{{/has_input_aliases}}
9298
{{/input_pins}}
9399

94100
@staticmethod

0 commit comments

Comments
 (0)