postgres_task_example.py

from prefect import Flow, Parameter, task
from prefect.tasks.postgres import PostgresExecute
from prefect.tasks.secrets import PrefectSecret

postgres_execute = PostgresExecute(
    db_name="abc", user="peter", host="my_host", port=5432,
)


@task
def define_query_from_param(param: str):
    return f"INSERT INTO {param} VALUES (...)"


with Flow("postgres_example") as flow:
    postgres_pwd = PrefectSecret("POSTGRES_PASSWORD")
    table_name = Parameter("table_name", default="my_schema.my_table")
    query = define_query_from_param(table_name)
    postgres_execute(password=postgres_pwd, query=query)