Gergő Jedlicska
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
"""This module contains the business logic of the function.
|
|
|
|
use the automation_context module to wrap your function in an Autamate context helper
|
|
"""
|
|
from automation_context import (
|
|
AutomateBase,
|
|
AutomationContext,
|
|
execute_automate_function,
|
|
)
|
|
from flatten import flatten_base
|
|
|
|
|
|
class FunctionInputs(AutomateBase):
|
|
"""These are function author defined values.
|
|
|
|
Automate will make sure to supply them matching the types specified here.
|
|
"""
|
|
|
|
forbidden_speckle_type: str
|
|
|
|
|
|
def automate_function(
|
|
automate_context: AutomationContext,
|
|
function_inputs: FunctionInputs,
|
|
) -> None:
|
|
"""Hey, trying the automate sdk experience here."""
|
|
version_root_object = automate_context.receive_version()
|
|
|
|
count = 0
|
|
for b in flatten_base(version_root_object):
|
|
if b.speckle_type == function_inputs.forbidden_speckle_type:
|
|
if not b.id:
|
|
raise ValueError("Cannot operate on objects without their id's.")
|
|
automate_context.add_object_error(
|
|
b.id,
|
|
"This project should not contain the type: "
|
|
f"{function_inputs.forbidden_speckle_type}",
|
|
)
|
|
count += 1
|
|
|
|
if count > 0:
|
|
automate_context.mark_run_failed(
|
|
"Automation failed: "
|
|
f"Found {count} object that have a forbidden speckle type: "
|
|
f"{function_inputs.forbidden_speckle_type}"
|
|
)
|
|
|
|
else:
|
|
automate_context.mark_run_success("No forbidden types found.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
execute_automate_function(automate_function, FunctionInputs)
|