o 'iU@s&ddlZddlmZGdddZdS)N)ERROR_CLASSES_MAPc@sJeZdZdZd ddZdedeeefdefdd Zdedefd d ZdS) ErrorClassesReaderz9A reader to load error information from error_classes.py.returnNcCs t|_dS)N)rerror_info_map)selfrd/home/ubuntu/transcripts/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)z[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_templatetablerrr get_error_messages   z$ErrorClassesReader.get_error_messagecCs|d}t|}|dvsJ|d}||jvr|j|}n d|d}t|d|d}|dk}|s9|} | S|d } |d } | | vrJ| | } n d | d}t|d| d} |d | } | S) aReturns 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_mapmsgmain_message_template has_sub_classrsub_error_class"main_error_class_subclass_info_mapsub_error_class_info_mapsub_message_templaterrr r )s, '       z'ErrorClassesReader.get_message_template)rN) __name__ __module__ __qualname____doc__r rdictrr rrrr rs   r)rr$rrrrrr s