Improve logging.

lolafect/connections.py

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