diff --git a/pipelines/exemplo/__init__.py b/pipelines/exemplo/__init__.py index 1337dbc..940d399 100644 --- a/pipelines/exemplo/__init__.py +++ b/pipelines/exemplo/__init__.py @@ -1,2 +1,3 @@ # -*- coding: utf-8 -*- from pipelines.exemplo.nome_do_objetivo.flows import * # noqa +from pipelines.exemplo.objetivo2.flows import * # noqa diff --git a/pipelines/exemplo/objetivo2/__init__.py b/pipelines/exemplo/objetivo2/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/pipelines/exemplo/objetivo2/flows.py b/pipelines/exemplo/objetivo2/flows.py new file mode 100644 index 0000000..659492e --- /dev/null +++ b/pipelines/exemplo/objetivo2/flows.py @@ -0,0 +1,28 @@ +# -*- coding: utf-8 -*- +from prefect import Parameter +from prefect.run_configs import KubernetesRun +from prefect.storage import GCS +from prefeitura_rio.pipelines_utils.custom import Flow +from prefeitura_rio.pipelines_utils.state_handlers import handler_inject_bd_credentials + +from pipelines.constants import constants +from pipelines.exemplo.objetivo2.tasks import hello_someone + +with Flow( + name="SANDBOX", + state_handlers=[handler_inject_bd_credentials], + skip_if_running=True, + parallelism=5, +) as rj_escritorio__exemplo__objetivo2__flow: + # Parameters + name = Parameter("name", default="world") + + # Flow + hello_someone(name=name) + + +rj_escritorio__exemplo__objetivo2__flow.storage = GCS(constants.GCS_FLOWS_BUCKET.value) +rj_escritorio__exemplo__objetivo2__flow.run_config = KubernetesRun( + image=constants.DOCKER_IMAGE.value, + labels=[constants.RJ_ESCRITORIO_AGENT_LABEL.value], +) diff --git a/pipelines/exemplo/objetivo2/tasks.py b/pipelines/exemplo/objetivo2/tasks.py new file mode 100644 index 0000000..5c115da --- /dev/null +++ b/pipelines/exemplo/objetivo2/tasks.py @@ -0,0 +1,8 @@ +# -*- coding: utf-8 -*- +from prefect import task +from prefeitura_rio.pipelines_utils.logging import log + + +@task +def hello_someone(name: str) -> None: + log(f"Hello, {name}!", "info")