o /ѹgW@s\ddlZddlmZddlmZddlZddlmZddlm Z ddl m Z e ej dZ Gdd d eZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZdejfddZddZdejfdd Zd!d"Z #d.dejd$eed%efd&d'Zd(d$Z d)ejfd*d+Z!d,d-Z"dS)/N) HTTPStatus)Optional) HTTPError)MODELSCOPE_REQUEST_ID) get_logger) log_levelc@ eZdZdS)NotSupportErrorN__name__ __module__ __qualname__rrY/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/modelscope/hub/errors.pyr r c@r)NoValidRevisionErrorNr rrrrrrrc@r) NotExistErrorNr rrrrrrrc@r) RequestErrorNr rrrrrrrc@r)GitErrorNr rrrrr rrc@r)InvalidParameterNr rrrrr$rrc@r)NotLoginExceptionNr rrrrr(rrc@r)FileIntegrityErrorNr rrrrr,rrc@r)FileDownloadErrorNr rrrrr0rrresponsecCst|jjvr |jjtSdS)N)rrequestheadersrrrrget_request_id4s  rcCs|dtjko |dS)z Check the request is ok Args: rsp (Response): The request response body Returns: bool: `True` if success otherwise `False`. CodeZSuccessN)rOKrsprrris_ok;s r#cCs.d|jddvr|}|S|jd}|S)Nzapplication/jsonz content-typerutf-8)rgetjsoncontentdecode)rmessagerrr_decode_response_errorGs  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)rrrhandle_http_post_errorOsr/Traise_on_errorreturncCsd}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|r||t ||d|jS)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)rloggercookiesZmodel_idr0http_error_msgr; request_idrrrhandle_http_responseZs:         rCcCs|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. rTMessageN)rr rr!rrrr0s   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. rTZ RequestIdzUrl = z , Request id=z Code = z Message = rDzC, Please specify correct dataset_name and namespace.N)r%rr r)r,r"rErBrrrdatahub_raise_on_errors $rFcCs 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$r2r3z7The request resource(model or dataset) does not exist!,zurl: %s, reason: %sr4z/Authentication token does not exist or invalid.r5r6r7r8r9POSTz %s, body: %srN) r:r;r<r(r=rr>r,rmethodbodyr)r"rAr;rBreqrrrraise_for_http_statuss8        rK)T)#logginghttprtypingrrequestsZrequests.exceptionsrZmodelscope.hub.constantsrZmodelscope.utils.loggerrWARNINGr? Exceptionr rrrrrrrrResponserr#r*r/boolintrCr0rFrKrrrrs<         %