zhengruifeng commented on code in PR #44349:
URL: https://github.com/apache/spark/pull/44349#discussion_r1427526487
##########
python/pyspark/errors/exceptions/connect.py:
##########
@@ -374,3 +425,37 @@ class
SparkNoSuchElementException(SparkConnectGrpcException, BaseNoSuchElementEx
"""
No such element exception.
"""
+
+
+class QueryContext(BaseQueryContext):
+ def __init__(self, q: pb2.FetchErrorDetailsResponse.QueryContext):
+ self._q = q
+
+ def contextType(self) -> QueryContextType:
+ context_type = self._q.context_type
+
+ if int(context_type) == QueryContextType.DataFrame.value:
+ return QueryContextType.DataFrame
+ else:
+ return QueryContextType.SQL
+
+ def objectType(self) -> str:
+ return str(self._q.object_type)
+
+ def objectName(self) -> str:
+ return str(self._q.object_name)
+
+ def startIndex(self) -> int:
+ return int(self._q.start_index)
+
+ def stopIndex(self) -> int:
+ return int(self._q.stop_index)
+
+ def fragment(self) -> str:
+ return str(self._q.fragment)
+
+ def callSite(self) -> str:
+ return str(self._q.callSite)
Review Comment:
probably should rename the protobuf field to `call_site`
--
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]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
