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

Docstrings and typing.

Browse source
This commit is contained in:
Pablo Martin 2023-01-23 14:06:19 +01:00
parent 26d1f10a00
commit 910373f19c
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
lolafect/connections.py
View file

@ -43,7 +43,14 @@ def connect_to_trino(
@task(trigger=all_finished)
def close_trino_connection(trino_connection: trino.dbapi.Connection):
def close_trino_connection(trino_connection: trino.dbapi.Connection) -> None:
"""
Close a Trino connection, or do nothing if what has been passed is not a
Trino connection.
:param trino_connection: a trino connection.
:return: None
"""
logger = prefect.context.get("logger")
if isinstance(trino_connection, trino.dbapi.Connection):
trino_connection.close()