2dZddlmZddlmZddlmZddlmZddlZddlmZ ddlm Z ddl mZ dd l mZejd Zejd ZGd d e j$ZGdde j$ZGdde j$ZGddeZGddeZGddeZGddeZGddeZGddee j6ZGddeeZGd d!eeZGd"d#eeZGd$d%e j@Z!Gd&d'ee j6Z" d*d(Z# d*d)Z$y)+z;API interface for interacting with cloud storage providers.)absolute_import) annotations)division)unicode_literalsN) exceptions)resource)urllibz(b/(?P.*)/o/(?P.*?)(\?|$)z`^b/(?P[a-z0-9\-_\.]+)(/)?(iam|channels|lockRetentionPolicy|iam/testPermissions|o)?(\?|$)c eZdZy) CloudApiErrorN__name__ __module__ __qualname__,lib/googlecloudsdk/api_lib/storage/errors.pyr r /rr c eZdZy)InsightApiErrorNr rrrrr3rrrc eZdZy)StorageBatchOperationsApiErrorNr rrrrr7rrrceZdZdZy)RetryableApiErrorz3Error raised to indicate a transient network error.Nr rr__doc__rrrrr;s;rrceZdZdZy) NotFoundErrorzError raised when the requested resource does not exist. Both GCS and S3 APIs should raise this error if a resource does not exist so that the caller can handle the error in an API agnostic manner. Nrrrrrr@s  rrceZdZdZy)PreconditionFailedError,Raised when a precondition is not satisfied.NrrrrrrJ4rrceZdZdZy) ConflictErrorDRaised when a resource cannot be created because one already exists.Nrrrrr#r#NLrr#ceZdZdZy)ResumableUploadAbortErrorz5Raised when a resumable upload needs to be restarted.Nrrrrr'r'Rs=rr'c"eZdZdZeddZy) GcsApiErrorzBase class for GCS API errors.ct|dr%|jr t|jddSyy#t$rYywxYw)zEReturns the status code of the error, or None if it is not available.error status_codeN)hasattrr+getattrKeyError)selfs rr,zGcsApiError.status_codeZsHtW$**tzz=$77 #- s 3 ??N)returnz int | None)r rrrpropertyr,rrrr)r)Ws&  rr)c.eZdZdZfdZdZdZxZS)GcsNotFoundErrorzwError raised when the requested GCS resource does not exist. Implements custom formatting to avoid messy default. c@~~tt| |d|jsyd}t j |j\}}}t j|}|r|j||ytj|}|r|j||yy)Nz)HTTPError {status_code}: {status_message}) error_formatz.gs://{instance_name} not found: {status_code}.) superr4__init__urlrSplitEndpointUrlOBJECT_RESOURCE_PATH_PATTERNsearch_custom_format_object_errorBUCKET_RESOURCE_PATH_PATTERN_custom_format_bucket_error) r0r+argskwargs+custom_error_format_for_buckets_and_objects_ resource_pathmatch_object_resource_pathmatch_bucket_resource_path __class__s rr8zGcsNotFoundError.__init__ls f D* A+ B 99  90#33EII>Aq-!=!D!D"! && $ 57 !=!D!D"! && $ 57"rcR||_|jd|j_y)a*Sets custom error formatting for buckets resource paths. Args: match_bucket_resource_path (re.Match): Match object that contains the result of searching regex BUCKET_RESOURCE_PATH_PATTERN in a resource path. error_format (str): Custom error format for buckets. bucketN)r6grouppayload instance_name)r0rFr6s rr?z,GcsNotFoundError._custom_format_bucket_errors#%D!;!A!A(!KDLLrc |j}tjj|}d|vr d|ddz}nd}||_dj |j d|j d||j_y) a)Sets custom error formatting for object resource paths. Args: match_object_resource_path (re.Match): Match object that contains the result of searching regex OBJECT_RESOURCE_PATH_PATTERN in a resource path. error_format (str): Custom error format for objects. generation#rz{}/{}{}rIobjectN) stringr parseparse_qsr6formatrJrKrL)r0rEr6rDparamsgeneration_strings rr=z,GcsNotFoundError._custom_format_object_errors/55M \\ " "= 1Fv| 4Q 77$D!*!1!1"((2"((24E"GDLLr)r rrrr8r?r= __classcell__rGs@rr4r4fs "7H LGrr4ceZdZdZy)GcsPreconditionFailedErrorr Nrrrrr[r[r!rr[ceZdZdZy)GcsConflictErrorr$Nrrrrr]r]r%rr]c"eZdZdZfdZxZS)S3ErrorPayloada^Allows using format strings to create strings from botocore ClientErrors. Format strings of the form '{field_name}' will be populated from class attributes. Strings of the form '{.field_name}' will be populated from the self.content JSON dump. See api_lib.util.HttpErrorPayload for more detail and sample usage. Attributes: content (dict): The dumped JSON content. message (str): The human readable error message. status_code (int): The HTTP status code number. status_description (str): The status_code description. status_message (str): Context specific status message. ctt| |t||_t |dr|j |_d|j vr$|j djdd|_ d|j vr=|j d}|jdd|_ |jdd|_ |j|_ y y ) zInitializes an S3ErrorPayload instance. Args: client_error (Union[botocore.exceptions.ClientError, str]): The error thrown by botocore, or a string that will be displayed as the error message. responseResponseMetadataHttpStatusCoderErrorCoderPMessageN)r7r_r8strbotocore_error_stringr-racontentgetr,status_descriptionstatus_message_MakeGenericMessagemessage)r0 client_errorr+rGs rr8zS3ErrorPayload.__init__s .$(6"%\!2D|Z(!**dl |44 4'001CDHH a! L)) )%%g."'))FB"7#ii 26--/dl)rr rrrr8rXrYs@rr_r_s 00rr_c$eZdZdZdfd ZxZS) XmlApiErrorzTranslates a botocore ClientError and allows formatting. Attributes: error: The original ClientError. error_format: An S3ErrorPayload format string. payload: The S3ErrorPayload object. c:tt| ||ty)N)r6 payload_class)r7rrr8r_)r0r+r6rGs rr8zXmlApiError.__init__s +t% L&Hr)z{botocore_error_string}rprYs@rrrrrsHHrrrct|d}|D]-\}}}t||s|| |||k(s$|||cS|S)aTranslates error or returns original error if no matches. Note, an error will be translated if it is a child class of a value in translation_list. Also, translations earlier in the list take priority. Args: error (Exception): Error to translate. translation_list (list): List of (Exception, int|None, Exception) tuples. Translates errors that are instances of first error type to second if the status code of the first exception matches the integer value. If the status code argument is None, the entry will translate errors of any status code. format_str (str|None): An api_lib.util.exceptions.FormattableErrorPayload format string. Note that any properties that are accessed here are on the FormattableErrorPayload object, not the object returned from the server. status_code_getter (Exception -> int|None): Function that gets a status code from the exception type raised by the underlying client, e.g. apitools_exceptions.HttpError. If None, only entries with a null status code in `translation_list` will translate errors. Returns: Error (Exception). Translated if match. Else, original error. cy)Nr)rCs rz!translate_error..s4r) isinstance)r+translation_list format_strstatus_code_getteruntranslated_erroruntranslated_status_codetranslated_errors rtranslate_errorrsa4'  %+, (  % e $(@ @ eZ 00  ,rcfd}|S)aDecorator catches an error and raises CloudApiError with a custom message. Args: translation_list (list): List of (Exception, int|None, Exception) tuples. Translates errors that are instances of first error type to second if the status code of the first exception matches the integer value. If the status code argument is None, the entry will translate errors of any status code. format_str (str|None): An api_lib.util.exceptions.FormattableErrorPayload format string. Note that any properties that are accessed here are on the FormattableErrorPayload object, not the object returned from the server. status_code_getter (Exception -> int|None): Function that gets a status code from the exception type raised by the underlying client, e.g. apitools_exceptions.HttpError. If None, only entries with a null status code in `translation_list` will translate errors. Returns: A decorator that catches errors and raises a CloudApiError with a customizable error message. Example: @catch_error_raise_cloud_api_error( [(apitools_exceptions.HttpError, GcsApiError)], 'Error [{status_code}]') def some_func_that_might_throw_an_error(): cfd}|S)Nc  |i|S#t$r,}tjt|Yd}~yd}~wwxYw)N)rzr{) Exceptioncore_exceptionsreraiser)r@rAerzfunctionr{rys rwrapperzYcatch_error_raise_cloud_api_error..translate_api_error_decorator..wrapperEsO  (((     %#5      s A";Ar)rrrzr{rys` rtranslate_api_error_decoratorzHcatch_error_raise_cloud_api_error..translate_api_error_decoratorBs   Nrr)ryrzr{rs``` r!catch_error_raise_cloud_api_errorr%s:( '&r)NN)%r __future__rrrrregooglecloudsdk.api_lib.utilrapi_exceptionsrgooglecloudsdk.corer six.movesr compiler;r>rdr rrrrrr#r' HttpExceptionr)r4r[r]FormattableErrorPayloadr_rrrrrrrrsNB&"' D0= *rzz/ 1 *rzz% & O))o++_%:%:  M5m5MMM  -!=!= NG{MNGb5.E5M{MM%0^;;%0P H-!=!= H(BF(X;?1'r