Improved code example

This commit is contained in:
Pablo Martin 2023-01-23 14:37:12 +01:00
parent e355e771b0
commit ee489edd61

View file

@ -41,11 +41,15 @@ lolaconfig = build_lolaconfig(
```python
from lolafect.connections import connect_to_trino, close_trino_connection
with Flow(...) as flow:
connection = connect_to_trino.run(
trino_credentials=my_trino_credentials #You can probably try to fetch this from lolaconfig.TRINO_CREDENTIALS
)
connection.cursor().execute("SELECT 1")
close_trino_connection.run(trino_connection=connection)
task_result = some_trino_related_task(trino_connection=connection)
close_trino_connection.run(
trino_connection=connection,
upstream_tasks=[task_result]
)
```