U lHJe@stddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z m Z m Z mZmZGdddejdZdS)N)AnyDictOptional) urlencode)confighttpsystemutilversionc@seZdZdZdZeejedddZ de e j ddddZ d d Zdee eejd d dZedddZee ejdddZdS)UAServiceClientN)returncCsdS)z@String in subclasses, the UAConfig attribute containing base urlNselfrr=/usr/lib/python3/dist-packages/uaclient/http/serviceclient.pycfg_url_base_attrsz!UAServiceClient.cfg_url_base_attr)cfgr cCs|st|_n||_dSN)rUAConfigr)rrrrr__init__s zUAServiceClient.__init__cCsdtdddS)Nz UA-Client/{}application/json)z user-agentZaccept content-type)formatr Z get_versionrrrrheaderss zUAServiceClient.headersT)log_response_bodytimeoutr c Cs|d}|s|}|ddkr>|r>tj|tjdd}t t |j |j |}| |} | rf| S|rddt|D} |dt| 7}|dk r|n|j} tj||||| |d S) N/rr)clszutf-8cSsi|]\}}|dk r||qSrr).0kvrrr >sz/UAServiceClient.request_url..?)urldatarmethodrr)lstriprgetjsondumpsr ZDatetimeAwareJSONEncoderencode posixpathjoingetattrrr_get_fake_responsessorteditemsr url_timeoutrZreadurl) rpathr%rr&Z query_paramsrrr$Z fake_responseZfiltered_paramsZtimeout_to_userrr request_url%s2   zUAServiceClient.request_url)r$cCsf|jdk r|j|gS|jjd}|s2i|_n&tj|sFi|_ntt ||_|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(rZfeaturesosr3existsr)loadsrZ load_file)rr$Zresponse_overlay_pathrrr_get_response_overlayMs  z%UAServiceClient._get_response_overlay)r$r cCs||}|sdSt|dkr(|d}n |d}i}g}|d}t|trR|}nt|tr`|}tj|d|dit j |ddd||d S) z7Return response if faked for this URL in uaclient.conf.NrresponsecoderT)Z sort_keys)r<rZbody json_dict json_list) r9lenpop isinstancedictlistr HTTPResponser(r)r*)rr$Z responsesr;r=r>Zresprrrr/os(       z#UAServiceClient._get_fake_responses)N)NNNNTN)__name__ __module__ __qualname__r2r5propertyabcabstractmethodstrrrrrrrboolintrrDr4r9r/rrrrr s(  ("r ) metaclass)rIr)r6r,typingrrrZ urllib.parserZuaclientrrrr r ABCMetar rrrrs