From 1de0e7dc2c99176d33045b2f7ddbe2c197ad624c Mon Sep 17 00:00:00 2001
From: Diego Imbert <diego@windmill.dev>
Date: Thu, 27 Nov 2025 18:51:50 +0100
Subject: [PATCH] ducklake client python

---
 python-client/wmill/wmill/client.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/python-client/wmill/wmill/client.py b/python-client/wmill/wmill/client.py
index 3557fa3518..0219b483f8 100644
--- a/python-client/wmill/wmill/client.py
+++ b/python-client/wmill/wmill/client.py
@@ -1018,6 +1018,9 @@ class Windmill:
     def datatable(self, name: str = "main"):
         return DataTableClient(self, name)
 
+    def ducklake(self, name: str = "main"):
+        return DucklakeClient(self, name)
+
 
 
 def init_global_client(f):
@@ -1575,7 +1578,7 @@ def datatable(name: str = "main"):
 
 @init_global_client
 def ducklake(name: str = "main"):
-    return _client.datatable(name)
+    return _client.ducklake(name)
 
 def task(*args, **kwargs):
     from inspect import signature
@@ -1690,4 +1693,22 @@ class DataTableClient:
             content=sql,
             language="postgresql",
             args={"database": f"datatable://{self.name}", **args_dict},
+        )
+
+class DucklakeClient:
+    def __init__(self, client: Windmill, name: str):
+        self.client = client
+        self.name = name
+    def fetch(self, sql: str, *args):
+        args_dict = {}
+        args_def = ""
+        for i, arg in enumerate(args):
+            args_dict[f"arg{i+1}"] = arg
+            args_def += f"-- $arg{i+1}\n"
+        attach = f"ATTACH 'ducklake://{self.name}' AS dl;USE dl;\n"
+        sql = args_def + attach + sql
+        return self.client.run_inline_script_preview(
+            content=sql,
+            language="duckdb",
+            args=args_dict,
         )
\ No newline at end of file