Ke@sddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z m Z m Z mZmZGddddejZdS)N)AnyDictOptional) urlencode)confighttpsystemutilversionc @seZdZdZdZeejedddZ de e j ddddZ d d Zddddd dee eejd d dZedddZee ejdddZdS)UAServiceClientN)returncCsdS)z@String in subclasses, the UAConfig attribute containing base urlN)selfrr=/usr/lib/python3/dist-packages/uaclient/http/serviceclient.pycfg_url_base_attrsz!UAServiceClient.cfg_url_base_attr)cfgr cCs%|stj|_n ||_dS)N)rUAConfigr)rrrrr__init__szUAServiceClient.__init__cCs%ddjtjddddiS)Nz user-agentz UA-Client/{}Zacceptzapplication/jsonz content-type)formatr Z get_version)rrrrheadersszUAServiceClient.headersT)log_response_bodytimeoutr c Cs|jd}|s!|j}|jddkr]|r]tj|dtjjd}tj t |j |j |}|j |} | r| S|rddt|jD} |dt| 7}|dk r|n|j} tjd |d |d |d |d | d|S)N/z content-typezapplication/jsonclszutf-8cSs+i|]!\}}|dk r||qS)Nr).0kvrrr >s z/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%s,   !  zUAServiceClient.request_url)r cCs|jdk r"|jj|gS|jjjd}|sIi|_n9tjj|sgi|_ntjt j ||_|jj|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/existsr%loadsrZ load_file)rr Zresponse_overlay_pathrrr_get_response_overlayMs    z%UAServiceClient._get_response_overlay)r r c Cs|j|}|sdSt|dkr8|d}n|jd}i}g}|d}t|tru|}nt|tr|}tjd|dd|jdidt j |ddd d |d |S) z7Return response if faked for this URL in uaclient.conf.NrresponsecoderZbodyZ sort_keysT json_dict json_list) r5lenpop isinstancedictlistr HTTPResponser$r%r&)rr Z responsesr7r9r:Zresprrrr+os&     z#UAServiceClient._get_fake_responses)__name__ __module__ __qualname__r.r1propertyabcabstractmethodstrrrrrrrboolintrr@r0r5r+rrrrr s "  "r metaclass)rEr%r2r(ZtypingrrrZ urllib.parserZuaclientrrrr r ABCMetar rrrrs    (