Typing, docstrings, refactors.

This commit is contained in:
Pablo Martin 2022-08-22 13:45:46 +02:00
parent 85c4de0d69
commit 3c8d2281ca
2 changed files with 20 additions and 6 deletions

View file

@ -46,7 +46,7 @@ class MySSHTunnel:
self.tunnel.stop()
def get_connection(connection_config: dict) -> Union[trino.dbapi.Connection]:
def get_connection(connection_config: dict) -> Union[trino.dbapi.Connection, mysql.connector.MySQLConnection]:
"""
Pick the right way to build a connection and pass it the connection details.