o ߥi@s@ddlZddlmZddlZddlmZddlmZddlm Z e ej dZ Gddde Z Gd d d e ZGd d d e ZGd dde ZGddde ZGddde ZGddde ZGddde ZGddde ZdejfddZddZdejfddZd d!Zdejfd"d#Zd$d%Zd&ejfd'd(Zd)d*ZdS)+N) HTTPStatus) HTTPError)MODELSCOPE_REQUEST_ID) get_logger) log_levelc@ eZdZdS)NotSupportErrorN__name__ __module__ __qualname__r r I/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/hub/errors.pyrrc@r)NoValidRevisionErrorNr r r r rrrrc@r) NotExistErrorNr r r r rrrrc@r) RequestErrorNr r r r rrrrc@r)GitErrorNr r r r rrrrc@r)InvalidParameterNr r r r rr#rrc@r)NotLoginExceptionNr r r r rr'rrc@r)FileIntegrityErrorNr r r r rr+rrc@r)FileDownloadErrorNr r r r rr/rrresponsecCst|jjvr |jjtSdS)N)rrequestheadersrr r rget_request_id3s  rcCs|dtjko |dS)z Check the request is ok Args: rsp (Response): The request response body Returns: bool: `True` if success otherwise `False`. CodeSuccess)rOKrspr r ris_ok:s r#cCs.d|jddvr|}|S|jd}|S)Nzapplication/jsonz content-typerutf-8)rgetjsoncontentdecode)rmessager r r_decode_response_errorFs  r*c CsJz|WdSty$}zt|}td|||t|f|d}~ww)NzHRequest %s with body: %s exception, Response details: %s, request id: %s)raise_for_statusrr*r)rurl request_bodyerrorr)r r rhandle_http_post_errorNsr/cCsd}t|jtr z|jd}Wnty|jd}Ynw|j}t|}d|jkr1d|}n?d|jkrA|dur>d} n2d}n/d |jkrLd krYnn d |j|||jf}nd |jkrdd krpnn d |j|||jf}|r}||t ||ddS)Nrr$ iso-8859-1z%The request model: %s does not exist!z%Authentication token does not exist, zGThe authentication token is invalid, failed to access model {model_id}./%s Client Error: %s, Request id: %s for url: %sX0%s Server Error: %s, Request id: %s, for url: %sr) isinstancereasonbytesr(UnicodeDecodeErrorr status_coder,r.r)rloggercookiesmodel_idhttp_error_msgr9 request_idr r rhandle_http_responseYs:         rBcCs|dtjkr dSt|d)zIf response error, raise exception Args: rsp (_type_): The server response Raises: RequestError: the response error message. Returns: bool: True if request is OK, otherwise raise `RequestError` exception. rTMessage)rr rr!r r rraise_on_errorys  rD http_responsec CsF|dtjkr dS|d}td|d|d|dd|dd ) aQIf response error, raise exception Args: url (str): The request url rsp (HTTPResponse): The server response. http_response: the origin http response. Raises: RequestError: the http request error. Returns: bool: `True` if request is OK, otherwise raise `RequestError` exception. rT RequestIdzUrl = z , Request id=z Code = z Message = rCzC, Please specify correct dataset_name and namespace.)r%rr r)r,r"rErAr r rdatahub_raise_on_errors $rGcCs d}t|jtr z|jd}Wnty|jd}Ynw|j}t|}d|jkr6d}d|j|fn7d|jkr>d}n/d |jkrId krVnn d |j|||jf}nd |jkrad krmnn d |j|||jf}|r|j}|j dkr~d||j f}t ||ddS)zAttempt to decode utf-8 first since some servers localize reason strings, for invalid utf-8, fall back to decoding with iso-8859-1. Args: rsp: The http response. Raises: HTTPError: The http error info. rr$r0r1z7The request resource(model or dataset) does not exist!,zurl: %s, reason: %sr2z/Authentication token does not exist or invalid.r3r4r5r6r7POSTz %s, body: %srN) r8r9r:r(r;rr<r,rmethodbodyr)r"r@r9rAreqr r rraise_for_http_statuss8        rL)logginghttprrequestsrequests.exceptionsrmodelscope.hub.constantsrmodelscope.utils.loggerrWARNINGr= ExceptionrrrrrrrrrResponserr#r*r/rBrDrGrLr r r rs0