counterweight/lolafect
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve logging.

Browse source
This commit is contained in:
Pablo Martin 2023-01-19 17:45:57 +01:00
parent caa69d85cd
commit 28831bea8c
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
lolafect/connections.py
View file

@ -1,5 +1,6 @@
import datetime import datetime
import prefect
from prefect import task from prefect import task
from trino.auth import BasicAuthentication from trino.auth import BasicAuthentication
import trino import trino
@ -14,12 +15,15 @@ def connect_to_trino(
""" """
Open a connection to the specified trino instance and return it. Open a connection to the specified trino instance and return it.
:param trino_credentials: a dict with the host, port, user and password. :param trino_credentials: a dict with the host, port, user and password.
:param http_schema: which http schema to use in the connection. :param http_schema: which http schema to use in the connection.
:return: :return:
""" """
print("Connecting to Trino.") logger = prefect.context.get("logger")
logger.info(
f"Connecting to Trino at {trino_credentials['host']}:{trino_credentials['port']}."
)
connection = trino.dbapi.connect( connection = trino.dbapi.connect(
host=trino_credentials["host"], host=trino_credentials["host"],
port=trino_credentials["port"], port=trino_credentials["port"],
@ -30,6 +34,8 @@ def connect_to_trino(
trino_credentials["password"], trino_credentials["password"],
), ),
) )
print("Connected to Trino.") logger.info(
f"Connected to Trino at {trino_credentials['host']}:{trino_credentials['port']}."
)
return connection return connection