o ?DiV@s2ddlZddlmZddlmZGdddZdS)N)Dict)ERROR_CLASSES_MAPc@sJeZdZdZd ddZdedeeefdefdd Zdedefd d ZdS) ErrorClassesReaderzC A reader to load error information from error_classes.py. returnNcCs t|_dS)N)rerror_info_map)selfr _/var/www/Datamplify/venv/lib/python3.10/site-packages/duckdb/experimental/spark/errors/utils.py__init__s zErrorClassesReader.__init__ error_classmessage_parameterscCs\||}td|}t|t|ksJd|d|tdd}||jdi|S)zm Returns the completed error message by applying message parameters to the message template. z<([a-zA-Z0-9_-]+)>z3Undefined error message parameter for error class: z. Parameters: z<>z{}Nr )get_message_templaterefindallsetstr maketrans translateformat)rr r message_template message_parameters_from_templatetabler r r get_error_message s   z$ErrorClassesReader.get_error_messagec Cs|d}t|}|dvsJ|d}||jvr|j|}ntd|dd|d}|dk}|s7|}|S|d } |d } | | vrH| | } ntd | dd| d} |d | }|S) a Returns the message template for corresponding error class from error_classes.py. For example, when given `error_class` is "EXAMPLE_ERROR_CLASS", and corresponding error class in error_classes.py looks like the below: .. code-block:: python "EXAMPLE_ERROR_CLASS" : { "message" : [ "Problem because of ." ] } In this case, this function returns: "Problem because of ." For sub error class, when given `error_class` is "EXAMPLE_ERROR_CLASS.SUB_ERROR_CLASS", and corresponding error class in error_classes.py looks like the below: .. code-block:: python "EXAMPLE_ERROR_CLASS" : { "message" : [ "Problem because of ." ], "sub_class" : { "SUB_ERROR_CLASS" : { "message" : [ "Do to fix the problem." ] } } } In this case, this function returns: "Problem because . Do to fix the problem." .)rrzCannot find main error class '' messagerr sub_classzCannot find sub error class ' )splitlenr ValueErrorjoin) rr error_classeslen_error_classesmain_error_classmain_error_class_info_mapmain_message_template has_sub_classrsub_error_class"main_error_class_subclass_info_mapsub_error_class_info_mapsub_message_templater r r r/s( (      z'ErrorClassesReader.get_message_template)rN) __name__ __module__ __qualname____doc__r rrrrr r r r rs  r)rtypingrr%rrr r r r s