o viD@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZddlmZdd lmZmZmZmZmZdd lmZmZd ZGd d d eZGdddeZ GdddeZ!GdddeZ"Gdddej#Z$Gdddej%Z&dS)N)partial) defaultdict)abstractmethodabstractproperty)JSONDecodeError) ProtocolError MaxRetryError) __version__) NotFoundErrorResourceNotFoundErrorResourceNotUniqueError ApiExceptionServiceUnavailableError)Resource ResourceListapisc@seZdZdZddZdddZddZd d Zed d Z e dddZ e ddZ e ddZdddZd ddZddZddZddZd S)! Discovererz A convenient container for storing discovered API resources. Allows easy searching and retrieval of specific resources. Subclasses implement the abstract methods with different loading strategies. cCs||_|jjj}tjr|d}zdtj|dd }Wnt y1dt| }Ynw|p_rKzDiscoverer.discovercCs|jSrI)_Discoverer__versionrGr+r+r,versioncszDiscoverer.versionc Cs\i}dd|rtd|dddddntdii|d<ddtddt|jgidii|t<|S)Nv1Tapi resourcesList) ResourceGroupget_resources_for_api_versionrrDISCOVERY_PREFIX)r'request_resourcesgroupsr+r+r,default_groupsgs  zDiscoverer.default_groupsc Cs|jdr|r~|jdi|jd<|jddtj}|j|d}|D]L}i}|dD];}|d}|jditi|di|} ||dk} | rR| jni} |r`| t|d|| } t | | d ||<q-||t|d<q%|jd || |jdS) z1 Discovers all API groups present in the cluster rXGETz/{}r]versionsrSnamepreferredVersionrW) r3r8rrequestrr\r^r_rXr[rZupdater?) r'r]rfgroups_responser^rM new_group version_rawrSresource_group preferredrXr+r+r,parse_api_groupsss$  (  zDiscoverer.parse_api_groupsc Csdd}|jdsGzd|jjdd|di|jd<Wn,ttfyF}zt|tr0t|jts0|jj j dsAtd |jj j d}~ww|jd|_ dS) NcSs|SrIr+)_ serializedr+r+r, just_jsonsz/Discoverer._load_server_info..just_jsonrS kubernetesr8z/version) serializerzhttps://zGHost value %s should start with https:// when talking to HTTPS endpoint) r3r8rre ValueErrorr isinstancereasonrrr startswithrR)r'rorBr+r+r,r=s   zDiscoverer._load_server_infoc Cs:tt}i}dtd|||g}z |jd|jpg}Wn ttfy*g}Ynwttdd|} ttdd|} | D]} | d dd\} } | | sTi|| <| || | <q?| D]=} d D]}| |dqat d ||||j|| | dd | }|| d  |t|j||| d d }||j |q]|S)zT returns a dictionary of resources associated with provided (prefix, group, version)/Nr`cSs d|dvSNrvrcr+resourcer+r+r, z:Discoverer.get_resources_for_api_version..cSs d|dvSrwr+rxr+r+r,rzr{rcr )rLrMrNrrk)rLrMrNrrk subresourcesrO)rMrN base_kindr+)rlistr"filterrrerXrrsplitr8poprappendrrO)r'rLrMrSrkrXr|r!resources_response resources_rawsubresources_raw subresourceryrckey resourceobj resource_listr+r+r,r[s@   z(Discoverer.get_resources_for_api_versionc s|jd i}t|dkrdrfdd|D}t|dkr2tdd|Ds2dd|D}t|dkr<|dS|sEtdtd |) z Same as search, but will throw an error if there are multiple or no results. If there are multiple results and only one is an exact match on api_version, that resource will be returned. r rNcsg|] }|jdkr|qS)rN) group_version.0resultrPr+r, sz"Discoverer.get..cSsg|]}t|tqSr+rsr)rxr+r+r,rscSsg|] }t|ts|qSr+rrr+r+r,rsrzNo matches found for {}z!Multiple matches found for {}: {}Nr+)rQlenr8allr rr r'rPresultsr+rr,r8s  zDiscoverer.get)FNNNN)FF)__name__ __module__ __qualname____doc__r-r&r?r9rrJrrQr>propertyrSr_rlr=r[r8r+r+r+r,r$s&        )rc@sVeZdZdZddZddZddZedd Zd d Z d d Z dddZ ddZ dS)LazyDiscovererz A convenient container for storing discovered API resources. Allows easy searching and retrieval of specific resources. Resources for the cluster are loaded lazily. cCst|||d|_dSNF)rr-_LazyDiscoverer__update_cacher'rr(r+r+r,r-s zLazyDiscoverer.__init__cC|jdd|_dS)NFra)rl_LazyDiscoverer__resourcesrGr+r+r,r>zLazyDiscoverer.discovercCs|jr |d|_dSdSr)rr?rGr+r+r,__maybe_write_caches z"LazyDiscoverer.__maybe_write_cachecCs|jddddS)NFTr]rfrrlkeysrGr+r+r,rJszLazyDiscoverer.api_groupscKslz||jdi||jg}Wn tyg}Ynw|s0|||jdi||jg}||S)Nr+)_LazyDiscoverer__search_LazyDiscoverer__build_searchrr r9"_LazyDiscoverer__maybe_write_cacherr+r+r,rQs  zLazyDiscoverer.searchc Cs~|d}|dkr||}|sgSt|trat|dkr"td||jsU|d|d|}}}z |||||j|_Wn tyFt w||j d|||<d|_ | |dd|j|St|t rt| |dd|||gS|ddkrt|dt r|D]} |dD]\} } t| | | kr| gSqqgS|Sg} |D]} | | | g|dd||q| S)Nr*z2prefix and group params should be present, have %sr rXT)r8rsrZrrrrXr[rkr r r3rrdictitemsgetattrrextend)r'partsrX reqParamspart resourcePartrLrMrS _resourcetermvaluematchesrr+r+r,__searchsF        $zLazyDiscoverer.__searchNcK>|s|rd|vr|d\}}|||||g}ttdd|S)NrvcS|pdSNrr+rr+r+r,rz0z/LazyDiscoverer.__build_search..rr~mapr'rLrMrNrOrPrr+r+r,__build_search+zLazyDiscoverer.__build_searchc cs|jD]?\}}|D]6\}}|D]-\}}|js5|||||j|_||jd|||<d|_t|jD]\}}|Vq;qqq| dS)NrXT) rrrXr[rkr3rr iteritemsr) r'rLr^rMrbrSrgrmryr+r+r,__iter__2s   zLazyDiscoverer.__iter__r) rrrrr-r>rrrJrQrrrr+r+r+r,rs  * rc@sVeZdZdZddZddZddZedd Zd d Z dd dZ ddZ ddZ d S)EagerDiscovererz A convenient container for storing discovered API resources. Allows easy searching and retrieval of specific resources. All resources are discovered for the cluster upon object instantiation. cCs ||_dSrI)_EagerDiscoverer__resources)r'rXr+r+r,rfHs zEagerDiscoverer.updatecCst|||dSrI)rr-rr+r+r,r-KrzEagerDiscoverer.__init__cCr)NTra)rlrrGr+r+r,r>NrzEagerDiscoverer.discovercCs|jddddS)z list available api groups TrrrrGr+r+r,rJQszEagerDiscoverer.api_groupscKsD||jdi||j}|s |||jdi||j}|S)aW Takes keyword arguments and returns matching resources. The search will happen in the following order: prefix: The api prefix for a resource, ie, /api, /oapi, /apis. Can usually be ignored group: The api group of a resource. Will also be extracted from api_version if it is present there api_version: The api version of a resource kind: The kind of the resource arbitrary arguments (see below), in random order The arbitrary arguments can be any valid attribute for an Resource object Nr+)_EagerDiscoverer__search_EagerDiscoverer__build_searchrr9rr+r+r,rQWs  zEagerDiscoverer.searchNcKr)NrvcSrrr+rr+r+r,rzmrz0EagerDiscoverer.__build_search..rrr+r+r,rhrzEagerDiscoverer.__build_searchc Cs|d}||}|dkr]|r]t|tr||dd|jSt|tr.||dd|S|ddkr[t|dtr[|D]}|dD]\}}t|||krW|gSqEq=gS|S|dkr|g}|D]} | || g|dd|qg|SgS)Nrrr ) r8rsrZrrXrrrrr) r'rrXrrrrrrrr+r+r,ros,     "zEagerDiscoverer.__searchccsV|jD]"\}}|D]\}}|D]\}}|D]\}}|VqqqqdSrI)rr)r'rmr^rbrXryr+r+r,rszEagerDiscoverer.__iter__r) rrrrrfr-r>rrJrQrrrr+r+r+r,rAs   rc@s"eZdZdZdddZddZdS)rZz%Helper class for Discoverer containerNcCs||_|pi|_dSrIrkrX)r'rkrXr+r+r,r-szResourceGroup.__init__cCsd|j|jdS)NrZ)_typerkrXrrGr+r+r,to_dictszResourceGroup.to_dictrI)rrrrr-rr+r+r+r,rZs  rZc@seZdZddZdS)rFcCs|SrI)r)r'or+r+r,defaultszCacheEncoder.defaultN)rrrrr+r+r+r,rFs rFc@seZdZddZddZdS)r7cOs*||_tjj|g|Rd|ji|dS)N object_hook)rr5 JSONDecoderr-r)r'rargsrPr+r+r,r-s$zCacheDecoder.__init__cCsrd|vr|S|d}|dkrtd d|ji|S|dkr&t|jfi|S|dkr7t|d||ddS|S) NrrrrrZrkrXrWr+)rrrrrZr)r'objrr+r+r,rs zCacheDecoder.object_hookN)rrrr-rr+r+r+r,r7s r7)'r rr5r;rr# functoolsr collectionsrabcrr json.decoderrurllib3.exceptionsrrrpr exceptionsr r r rrryrrr\objectrrrrZ JSONEncoderrFrr7r+r+r+r,s,    ;cO