pankajkoti commented on code in PR #48507:
URL: https://github.com/apache/airflow/pull/48507#discussion_r2024274065
##########
providers/databricks/src/airflow/providers/databricks/hooks/databricks.py:
##########
@@ -709,6 +768,47 @@ def update_job_permission(self, job_id: int, json:
dict[str, Any]) -> dict:
"""
return self._do_api_call(("PATCH",
f"api/2.0/permissions/jobs/{job_id}"), json)
+ def post_sql_statement(self, json: dict[str, Any]) -> str:
+ response = self._do_api_call(("POST", "api/2.0/sql/statements"), json)
+ return response["statement_id"]
+
+ def get_sql_statement_state(self, statement_id: str) -> SQLStatementState:
+ """
+ Retrieve run state of the SQL statement.
+
+ :param statement_id: ID of the SQL statement.
+ :return: state of the SQL statement.
+ """
+ get_statement_endpoint = ("GET",
f"api/2.0/sql/statements/{statement_id}")
Review Comment:
added such a constant and used it
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
