o K&b@sddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZmZmZGdddejd ZdS) N)join)AnyDictOptionalTupleType)error) urlencode)config exceptionsutilversionc @seZdZdZdZeejdee fddZ eejde fddZ dde ejddfdd Zd d Z dd d Zde fddZde dee ee efe ee e fffddZdS)UAServiceClientNreturncCdS)z1Set in subclasses to the type of API error raisedNselfrr8/usr/lib/python3/dist-packages/uaclient/serviceclient.py api_error_clszUAServiceClient.api_error_clscCr)z@String in subclasses, the UAConfig attribute containing base urlNrrrrrcfg_url_base_attrrz!UAServiceClient.cfg_url_base_attrcfgcCs|s t|_dS||_dSN)r UAConfigr)rrrrr__init__s zUAServiceClient.__init__cCsdtdddS)Nz UA-Client/{}application/json)z user-agentZaccept content-type)formatr Z get_versionrrrrheaders%s zUAServiceClient.headersc CsX|d}|s |}|ddkr|rt|d}tt|j|j |}| |\}}|r3||fS|rHddt | D} |dt | 7}ztj|||||jd\} }W| |fStjy} zDd} t| d rn| j} n t| d rz| d} | rz tj| tjd } Wn tyd} Ynw| r|| | tj| t| d d||d d} ~ ww)N/rrzutf-8cSsi|] \}}|dur||qSrr).0kvrrr :s  z/UAServiceClient.request_url..?)urldatarmethodZtimeoutbodyread)clscoder,rr&)lstriprgetjsondumpsencodeurljoingetattrrr_get_fake_responsessorteditemsr r Zreadurl url_timeoutrURLErrorhasattrr)r*decodeloadsZDatetimeAwareJSONDecoder ValueErrorrr UrlError)rpathr'rr(Z query_paramsr&Z fake_responseZ fake_headersZfiltered_paramsresponseer)Z error_detailsrrr request_url,sX       zUAServiceClient.request_urlr&cCsf|jdur |j|gS|jjd}|si|_ntj|s#i|_n tt ||_|j|gS)a.Return a list of fake response dicts for a given URL. serviceclient_url_responses in uaclient.conf should be a path to a json file which contains a dictionary keyed by full URL path. Each value will be a list of dicts representing each faked response for the given URL. The response dict item will have a code: and response: "some string of content". The JSON string below fakes the available_resources URL on the contract server: '{"https://contracts.canonical.com/v1/resources": [{"code": 200, "response": {"key": "val1", "key2": "val2"}}]}' :return: List of dicts for each faked response matching the url, or and empty list when no matching url found. NZserviceclient_url_responses) _response_overlayr/rZfeaturesosr?existsr0r<r Z load_file)rr&Zresponse_overlay_pathrrr_get_response_overlayZs  z%UAServiceClient._get_response_overlaycCs||}|s difSt|dkr|d}n|d}|ddkr+|d|difSt|d}tj||d|di|d) aReturn response and headers if faked for this URL in uaclient.conf. :return: A tuple of response and header dicts if the URL has an overlay response defined. Return (None, {}) otherwise. :raises exceptions.URLError: When faked response "code" is != 200. URLError reason will be "response" value and any optional "headers" provided. Nrr,r@rr-)rFlenpopr/rr9r r>)rr&Z responsesr@rArrrr5|s      z#UAServiceClient._get_fake_responsesr)NNNN)__name__ __module__ __qualname__r8rCpropertyabcabstractmethodr Exceptionrstrrrr rrrrBrFrrrr5rrrrr s& .""r) metaclass)rOr0rD posixpathrr3typingrrrrrZurllibrZ urllib.parser Zuaclientr r r r ABCMetarrrrrs