feat(trino): add source application to trino backend

ibis/backends/trino/__init__.py

@@ -74,6 +74,7 @@ def do_connect(
         port: int = 8080,
         database: str | None = None,
         schema: str | None = None,
+        source: str | None = None,
         **connect_args,
     ) -> None:
         """Create an Ibis client connected to a Trino database."""
@@ -85,6 +86,7 @@ def do_connect(
             host=host,
             port=port,
             database=database,
+            query=dict(source="ibis" if source is None else source),
         )
         connect_args.setdefault("timezone", "UTC")
         with warnings.catch_warnings():

ibis/backends/trino/tests/test_client.py

@@ -3,6 +3,12 @@
 import pytest
 
 import ibis
+from ibis.backends.trino.tests.conftest import (
+    TRINO_HOST,
+    TRINO_PASS,
+    TRINO_PORT,
+    TRINO_USER,
+)
 
 
 @pytest.fixture
@@ -56,3 +62,17 @@ def test_list_catalogs(con):
 
 def test_list_schemas(con):
     assert {"information_schema", "sf1"}.issubset(con.list_schemas(database="tpch"))
+
+
+@pytest.mark.parametrize(("source", "expected"), [(None, "ibis"), ("foo", "foo")])
+def test_con_source(source, expected):
+    con = ibis.trino.connect(
+        user=TRINO_USER,
+        host=TRINO_HOST,
+        port=TRINO_PORT,
+        password=TRINO_PASS,
+        database="hive",
+        schema="default",
+        source=source,
+    )
+    assert con.con.url.query["source"] == expected