fix: prefix iaso widget with IASO_

Author: nazarfil

examples/pipelines/iaso/pipeline.py

@@ -9,23 +9,39 @@
 @parameter("iaso_con", type=IASOConnection, required=True)
 @parameter(
     "forms",
-    type=str,
+    type=int,
     required=True,
-    widget=IASOWidget.FORMS,
+    widget=IASOWidget.IASO_FORMS,
     multiple=True,
     connection="iaso_con",
 )
-def iaso(iaso_con, forms):
+@parameter(
+    "projects",
+    type=int,
+    required=True,
+    widget=IASOWidget.IASO_PROJECTS,
+    multiple=True,
+    connection="iaso_con",
+)
+@parameter(
+    "org_units",
+    type=int,
+    required=True,
+    widget=IASOWidget.IASO_ORG_UNITS,
+    multiple=True,
+    connection="iaso_con",
+)
+def iaso(iaso_con, forms, org_units, projects):
     """Get forms from IASO."""
-    print_forms(iaso_con, forms)
+    print_forms(iaso_con, forms, org_units, projects)
 
 
 @iaso.task
-def print_forms(iaso_con, forms):
+def print_forms(iaso_con, forms, org_units, projects):
     """Print forms."""
     current_run.log_info("Printing forms")
 
-    current_run.log_info(f"Forms: {forms}")
+    current_run.log_info(f"Forms: {forms}, Org Units: {org_units}, Projects: {projects}")
 
 
 if __name__ == "__main__":

openhexa/sdk/pipelines/parameter.py

@@ -362,9 +362,9 @@ def validate(self, value: typing.Any | None) -> Dataset:
 class IASOWidget(StrEnum):
     """Enum for IASO widgets."""
 
-    FORMS = "IASO_FORMS"
+    IASO_FORMS = "IASO_FORMS"
     IASO_ORG_UNITS = "IASO_ORG_UNITS"
-    PROJECTS = "IASO_PROJECTS"
+    IASO_PROJECTS = "IASO_PROJECTS"
 
 
 class DHIS2Widget(StrEnum):

tests/test_ast.py

@@ -543,7 +543,7 @@ def test_pipeline_with_connection_parameter_for_iaso(self):
                             "code": "projects",
                             "type": "str",
                             "name": "Projects",
-                            "widget": IASOWidget.PROJECTS.value,
+                            "widget": IASOWidget.IASO_PROJECTS.value,
                             "connection": "iaso_con",
                             "default": None,
                             "multiple": False,
@@ -555,7 +555,7 @@ def test_pipeline_with_connection_parameter_for_iaso(self):
                             "code": "forms",
                             "type": "str",
                             "name": "Forms",
-                            "widget": IASOWidget.FORMS.value,
+                            "widget": IASOWidget.IASO_FORMS.value,
                             "connection": "iaso_con",
                             "default": None,
                             "multiple": False,