Merge pull request #4 from NERC-CEH/add_profiling

add profiling option for duckdb

src/driutils/io/duckdb.py

@@ -51,14 +51,17 @@ class DuckDBS3Reader(DuckDBReader):
     """Concrete Implementation of a DuckDB reader for reading
     data from an S3 endpoint"""
 
-    def __init__(self, auth_type: str, endpoint_url: Optional[str] = None, use_ssl: bool = True) -> None:
+    def __init__(
+        self, auth_type: str, endpoint_url: Optional[str] = None, use_ssl: bool = True, profiling: bool = False
+    ) -> None:
         """Initializes
 
         Args:
             auth_type: The type of authentication to request. May
             be one of ["auto", "sts", "custom_endpoint"]
             endpoint_url: Custom s3 endpoint
             use_ssl: Flag for using ssl (https connections).
+            profiling: Profile all duckdb queries. False by default.
         """
 
         super().__init__()
@@ -76,6 +79,9 @@ def __init__(self, auth_type: str, endpoint_url: Optional[str] = None, use_ssl:
 
         self._authenticate(auth_type, endpoint_url, use_ssl)
 
+        if profiling:
+            self._connection.execute("SET enable_profiling = query_tree;")
+
     def _authenticate(self, method: str, endpoint_url: Optional[str] = None, use_ssl: bool = True) -> None:
         """Handles authentication selection