Add JWT token support via connection string in sqlalchemy

Author: ms32035

README.md

@@ -151,8 +151,11 @@ the [`JWT` authentication type](https://trino.io/docs/current/security/jwt.html)
 
     ```python
     from sqlalchemy import create_engine
-    from trino.auth import JWTAuthentication
 
+    engine = create_engine("trino://<username>@<host>:<port>/<catalog>/<schema>?access_token=<jwt_token>")
+  
+    # or
+    from trino.auth import JWTAuthentication
     engine = create_engine(
         "trino://<username>@<host>:<port>/<catalog>",
         connect_args={

trino/sqlalchemy/dialect.py

@@ -18,7 +18,7 @@
 from sqlalchemy.engine.url import URL
 
 from trino import dbapi as trino_dbapi, logging
-from trino.auth import BasicAuthentication
+from trino.auth import BasicAuthentication, JWTAuthentication
 from trino.dbapi import Cursor
 from trino.exceptions import TrinoUserError
 from trino.sqlalchemy import compiler, datatype, error
@@ -87,6 +87,10 @@ def create_connect_args(self, url: URL) -> Tuple[Sequence[Any], Mapping[str, Any
             kwargs["http_scheme"] = "https"
             kwargs["auth"] = BasicAuthentication(url.username, url.password)
 
+        if "access_token" in url.query:
+            kwargs["http_scheme"] = "https"
+            kwargs["auth"] = JWTAuthentication(url.query["access_token"])
+
         return args, kwargs
 
     def get_columns(self, connection: Connection, table_name: str, schema: str = None, **kw) -> List[Dict[str, Any]]: