o i@sddlmZddlmZGdddeZGdddeZGdd d eZGd d d eZGd d d eZ GdddeZ GdddeZ GdddeZ GdddeZ GdddeZGddde ZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#eZGd$d%d%eZGd&d'd'eeZGd(d)d)eeZGd*d+d+eeZGd,d-d-eeZGd.d/d/eeZGd0d1d1ee Z!Gd2d3d3ee"Z#d4S)5)cast)ErrorClassesReaderc @seZdZdZ   ddedBdedBdeeefdBddfddZdedBfd d ZdeeefdBfd d Zdd dZ defddZ dS)PySparkExceptionz:Base Exception for handling errors generated from PySpark.Nmessage error_classmessage_parametersreturncCsx|dur |dur |dus|dur|dur|dusJt|_|dur1|jtd|td||_n||_||_||_dS)Nstrzdict[str, str])r error_readerget_error_messagerrrr)selfrrrrd/home/ubuntu/.local/lib/python3.10/site-packages/duckdb/experimental/spark/errors/exceptions/base.py__init__ s  zPySparkException.__init__cC|jS)zReturns an error class as a string. .. versionadded:: 3.4.0 See Also: -------- :meth:`PySparkException.getMessageParameters` :meth:`PySparkException.getSqlState` )rr rrr getErrorClass" zPySparkException.getErrorClasscCr)zReturns a message parameters as a dictionary. .. versionadded:: 3.4.0 See Also: -------- :meth:`PySparkException.getErrorClass` :meth:`PySparkException.getSqlState` )rrrrrgetMessageParameters.rz%PySparkException.getMessageParameterscCsdS)a$Returns an SQLSTATE as a string. Errors generated in Python have no SQLSTATE, so it always returns None. .. versionadded:: 3.4.0 See Also: -------- :meth:`PySparkException.getErrorClass` :meth:`PySparkException.getMessageParameters` Nrrrrr getSqlState:s zPySparkException.getSqlStatecCs(|durd|d|jS|jS)N[z] )rrrrrr__str__Hs zPySparkException.__str__)NNN)r N) __name__ __module__ __qualname____doc__r dictrrrrrrrrrrs$  rc@eZdZdZdS)AnalysisExceptionz#Failed to analyze a SQL query plan.NrrrrrrrrrOrc@r)SessionNotSameExceptionz7Performed the same operation on different SparkSession.Nr rrrrr"Sr!r"c@r)TempTableAlreadyExistsExceptionz6Failed to create temp view since it is already exists.Nr rrrrr#Wr!r#c@r)ParseExceptionzFailed to parse a SQL command.Nr rrrrr$[r!r$c@r)IllegalArgumentExceptionz,Passed an illegal or inappropriate argument.Nr rrrrr%_r!r%c@r)ArithmeticExceptionz;Arithmetic exception thrown from Spark with an error class.Nr rrrrr&cr!r&c@r)UnsupportedOperationExceptionzFUnsupported operation exception thrown from Spark with an error class.Nr rrrrr'gr!r'c@r)ArrayIndexOutOfBoundsExceptionzJArray index out of bounds exception thrown from Spark with an error class.Nr rrrrr(kr!r(c@r)DateTimeExceptionz9Datetime exception thrown from Spark with an error class.Nr rrrrr)or!r)c@r)NumberFormatExceptionz>Number format exception thrown from Spark with an error class.Nr rrrrr*sr!r*c@r)StreamingQueryExceptionz1Exception that stopped a :class:`StreamingQuery`.Nr rrrrr+wr!r+c@r)QueryExecutionExceptionzFailed to execute a query.Nr rrrrr,{r!r,c@r)PythonExceptionz&Exceptions thrown from Python workers.Nr rrrrr-r!r-c@r)SparkRuntimeExceptionz8Runtime exception thrown from Spark with an error class.Nr rrrrr.r!r.c@r)SparkUpgradeExceptionz*Exception thrown because of Spark upgrade.Nr rrrrr/r!r/c@r)UnknownExceptionzNone of the above exceptions.Nr rrrrr0r!r0c@r)PySparkValueErrorz6Wrapper class for ValueError to support error classes.Nr rrrrr1r!r1c@r)PySparkIndexErrorz6Wrapper class for IndexError to support error classes.Nr rrrrr2r!r2c@r)PySparkTypeErrorz5Wrapper class for TypeError to support error classes.Nr rrrrr3r!r3c@r)PySparkAttributeErrorz:Wrapper class for AttributeError to support error classes.Nr rrrrr4r!r4c@r)PySparkRuntimeErrorz8Wrapper class for RuntimeError to support error classes.Nr rrrrr5r!r5c@r)PySparkAssertionErrorz:Wrapper class for AssertionError to support error classes.Nr rrrrr6r!r6c@r)PySparkNotImplementedErrorz?Wrapper class for NotImplementedError to support error classes.Nr rrrrr7r!r7N)$typingrutilsr Exceptionrrr"r#r$r%r&r'r(r)r*r+r,r-r.r/r0 ValueErrorr1 IndexErrorr2 TypeErrorr3AttributeErrorr4 RuntimeErrorr5AssertionErrorr6NotImplementedErrorr7rrrrs4 I