o ?Di@sddlmZmZmZddlmZGdddeZGdddeZGdd d eZ Gd d d eZ Gd d d eZ GdddeZ GdddeZ GdddeZGdddeZGdddeZGddde ZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#eZGd$d%d%eZGd&d'd'eeZGd(d)d)eeZGd*d+d+eeZGd,d-d-eeZGd.d/d/ee Z!Gd0d1d1ee"Z#Gd2d3d3ee$Z%d4S)5)DictOptionalcast)ErrorClassesReaderc @seZdZdZ   ddeedeedeeeeffddZdeefd d Zdeeeeffd d Z dd dZ defddZ dS)PySparkExceptionzD Base Exception for handling errors generated from PySpark. Nmessage error_classmessage_parameterscCs|dur |dur |dus|dur|dur|dusJt|_|dur5|jtt|ttttf||_n||_||_||_dS)N) r error_readerget_error_messagerstrrrr r )selfrr r ri/var/www/Datamplify/venv/lib/python3.10/site-packages/duckdb/experimental/spark/errors/exceptions/base.py__init__ s  zPySparkException.__init__returncC|jS)z Returns an error class as a string. .. versionadded:: 3.4.0 See Also -------- :meth:`PySparkException.getMessageParameters` :meth:`PySparkException.getSqlState` )r rrrr getErrorClass# zPySparkException.getErrorClasscCr)z Returns a message parameters as a dictionary. .. versionadded:: 3.4.0 See Also -------- :meth:`PySparkException.getErrorClass` :meth:`PySparkException.getSqlState` )r rrrrgetMessageParameters0rz%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__Ls zPySparkException.__str__)NNN)rN) __name__ __module__ __qualname____doc__rr rrrrrrrrrrrs   rc@eZdZdZdS)AnalysisExceptionz- Failed to analyze a SQL query plan. Nrrrrrrrrr Rr c@r)SessionNotSameExceptionzA Performed the same operation on different SparkSession. Nr!rrrrr#Xr"r#c@r)TempTableAlreadyExistsExceptionz@ Failed to create temp view since it is already exists. Nr!rrrrr$^r"r$c@r)ParseExceptionz( Failed to parse a SQL command. Nr!rrrrr%dr"r%c@r)IllegalArgumentExceptionz6 Passed an illegal or inappropriate argument. Nr!rrrrr&jr"r&c@r)ArithmeticExceptionzE Arithmetic exception thrown from Spark with an error class. Nr!rrrrr'pr"r'c@r)UnsupportedOperationExceptionzP Unsupported operation exception thrown from Spark with an error class. Nr!rrrrr(vr"r(c@r)ArrayIndexOutOfBoundsExceptionzT Array index out of bounds exception thrown from Spark with an error class. Nr!rrrrr)|r"r)c@r)DateTimeExceptionzC Datetime exception thrown from Spark with an error class. Nr!rrrrr*r"r*c@r)NumberFormatExceptionzH Number format exception thrown from Spark with an error class. Nr!rrrrr+r"r+c@r)StreamingQueryExceptionz; Exception that stopped a :class:`StreamingQuery`. Nr!rrrrr,r"r,c@r)QueryExecutionExceptionz$ Failed to execute a query. Nr!rrrrr-r"r-c@r)PythonExceptionz0 Exceptions thrown from Python workers. Nr!rrrrr.r"r.c@r)SparkRuntimeExceptionzB Runtime exception thrown from Spark with an error class. Nr!rrrrr/r"r/c@r)SparkUpgradeExceptionz4 Exception thrown because of Spark upgrade. Nr!rrrrr0r"r0c@r)UnknownExceptionz' None of the above exceptions. Nr!rrrrr1r"r1c@r)PySparkValueErrorz@ Wrapper class for ValueError to support error classes. Nr!rrrrr2r"r2c@r)PySparkIndexErrorz@ Wrapper class for IndexError to support error classes. Nr!rrrrr3r"r3c@r)PySparkTypeErrorz? Wrapper class for TypeError to support error classes. Nr!rrrrr4r"r4c@r)PySparkAttributeErrorzD Wrapper class for AttributeError to support error classes. Nr!rrrrr5r"r5c@r)PySparkRuntimeErrorzB Wrapper class for RuntimeError to support error classes. Nr!rrrrr6r"r6c@r)PySparkAssertionErrorzD Wrapper class for AssertionError to support error classes. Nr!rrrrr7r"r7c@r)PySparkNotImplementedErrorzI Wrapper class for NotImplementedError to support error classes. Nr!rrrrr8r"r8N)&typingrrrutilsr Exceptionrr r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1 ValueErrorr2 IndexErrorr3 TypeErrorr4AttributeErrorr5 RuntimeErrorr6AssertionErrorr7NotImplementedErrorr8rrrrs4 M