class ParameterSchema(TaskMethodsMixin, ObjectSchema):
class Meta:
object_class = lambda: prefect.core.task.Parameter # type: ignore
exclude_fields = ["id", "type", "outputs"]
id = UUID()
type = fields.Function(lambda task: to_qualified_name(type(task)), lambda x: x)
name = fields.String(required=True)
default = JSONCompatible(allow_none=True)
required = fields.Boolean(allow_none=True)
description = fields.String(allow_none=True)
tags = fields.List(fields.String())
outputs = fields.Method("load_outputs", allow_none=True)
class TaskSchema(TaskMethodsMixin, ObjectSchema):
class Meta:
object_class = lambda: prefect.core.Task
exclude_fields = ["id", "type", "inputs", "outputs"]
id = UUID()
type = fields.Function(lambda task: to_qualified_name(type(task)),
lambda x: x)
name = fields.String(allow_none=True)
slug = fields.String(allow_none=True)
description = fields.String(allow_none=True)
tags = fields.List(fields.String())
max_retries = fields.Integer(allow_none=True)
retry_delay = fields.TimeDelta(allow_none=True)
inputs = fields.Method("load_inputs", allow_none=True)
outputs = fields.Method("load_outputs", allow_none=True)
timeout = fields.Integer(allow_none=True)
trigger = FunctionReference(
valid_functions=[
prefect.triggers.all_finished,
prefect.triggers.manual_only,
prefect.triggers.always_run,
prefect.triggers.all_successful,
prefect.triggers.all_failed,
prefect.triggers.any_successful,
prefect.triggers.any_failed,
],
# don't reject custom functions, just leave them as strings
reject_invalid=False,
allow_none=True,
)
skip_on_upstream_skip = fields.Boolean(allow_none=True)
cache_for = fields.TimeDelta(allow_none=True)
cache_validator = FunctionReference(
valid_functions=[
prefect.engine.cache_validators.never_use,
prefect.engine.cache_validators.duration_only,
prefect.engine.cache_validators.all_inputs,
prefect.engine.cache_validators.all_parameters,
prefect.engine.cache_validators.partial_inputs_only,
prefect.engine.cache_validators.partial_parameters_only,
],
# don't reject custom functions, just leave them as strings
reject_invalid=False,
allow_none=True,
)
class Schema(marshmallow.Schema):
u = UUID()