+ Query Profile Graph
+
+ ${METRIC_GRID}
+
+
+ ${SQL_QUERY}
+ ${TIMING_TABLE}
+
+ ${TREE}
+
+
+ """ # noqa: E501
+ html = html.replace("${TREEFLEX_CSS}", html_output["treeflex_css"])
+ html = html.replace("${DUCKDB_CSS}", html_output["duckdb_css"])
+ html = html.replace("${METRIC_GRID}", highlight_metric_grid)
+ html = html.replace("${SQL_QUERY}", sql_query_html)
+ html = html.replace("${TIMING_TABLE}", timing_table)
+ html = html.replace("${TREE}", tree_output)
+ f.write(html)
def main() -> None: # noqa: D103
@@ -326,7 +641,7 @@ def main() -> None: # noqa: D103
description="""Given a json profile output, generate a html file showing the query graph and
timings of operators""",
)
- parser.add_argument("profile_input", help="profile input in json")
+ parser.add_argument("--profile_input", help="profile input in json")
parser.add_argument("--out", required=False, default=False)
parser.add_argument("--open", required=False, action="store_true", default=True)
args = parser.parse_args()
@@ -347,8 +662,8 @@ def main() -> None: # noqa: D103
exit(1)
open_output = args.open
-
- translate_json_to_html(input, output)
+ profiling_info = ProfilingInfo(from_file=input)
+ profiling_info.to_html(output_file=output)
if open_output:
webbrowser.open(f"file://{Path(output).resolve()}", new=2)
diff --git a/scripts/connection_methods.json b/scripts/connection_methods.json
index a87b992f..3b02a9b1 100644
--- a/scripts/connection_methods.json
+++ b/scripts/connection_methods.json
@@ -1093,5 +1093,30 @@
}
],
"return": "None"
+ },
+ {
+ "name": "get_profiling_information",
+ "function": "GetProfilingInformation",
+ "docs": "Get profiling information for a query",
+ "args": [
+ {
+ "name": "format",
+ "default": "JSON",
+ "type": "Optional[str]"
+ }
+ ],
+ "return": "str"
+ },
+ {
+ "name": "enable_profiling",
+ "function": "EnableProfiling",
+ "docs": "Enable profiling for a connection",
+ "return": "None"
+ },
+ {
+ "name": "disable_profiling",
+ "function": "DisableProfiling",
+ "docs": "Disable profiling for a connection",
+ "return": "None"
}
]
diff --git a/scripts/generate_connection_stubs.py b/scripts/generate_connection_stubs.py
index d542a047..76c19b36 100644
--- a/scripts/generate_connection_stubs.py
+++ b/scripts/generate_connection_stubs.py
@@ -5,7 +5,7 @@
os.chdir(Path(__file__).parent)
JSON_PATH = "connection_methods.json"
-DUCKDB_STUBS_FILE = Path("..") / "duckdb" / "__init__.pyi"
+DUCKDB_STUBS_FILE = Path("..") / "_duckdb-stubs" / "__init__.pyi"
START_MARKER = " # START OF CONNECTION METHODS"
END_MARKER = " # END OF CONNECTION METHODS"
diff --git a/src/duckdb_py/duckdb_python.cpp b/src/duckdb_py/duckdb_python.cpp
index 1dd3ba17..fedbec5f 100644
--- a/src/duckdb_py/duckdb_python.cpp
+++ b/src/duckdb_py/duckdb_python.cpp
@@ -124,6 +124,34 @@ static void InitializeConnectionMethods(py::module_ &m) {
},
"Check if a filesystem with the provided name is currently registered", py::arg("name"), py::kw_only(),
py::arg("connection") = py::none());
+ m.def(
+ "get_profiling_information",
+ [](const py::str &format, shared_ptr