o \hLD@s ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZddlmZmZmZmZmZdd lmZmZd ZGd d d eZGd ddeZGdddeZGdddeZ Gdddej!Z"Gdddej#Z$dS)N)partial) defaultdict)abstractmethodabstractproperty) 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|pKs& zDiscoverer._write_cachecCs|jdddS)NT)r?)r%r&r*r*r+r8SszDiscoverer.invalidate_cachecCdSNr*rFr*r*r+ api_groupsVzDiscoverer.api_groupsNcKrGrHr*)r&prefixgroup api_versionkindkwargsr*r*r+searchZrJzDiscoverer.searchcCrGrHr*rFr*r*r+r=^rJzDiscoverer.discovercCs|jSrH)_Discoverer__versionrFr*r*r+versionbszDiscoverer.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_groupsfs  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 rWGETz/{}r\versionsrRnamepreferredVersionrV) r2r7rrequestrr[r]r^rWrZrYupdater>) r&r\regroups_responser]rL new_group version_rawrRresource_group preferredrWr*r*r+parse_api_groupsrs$  (  zDiscoverer.parse_api_groupsc Csdd}|jdsGzd|jjdd|di|jd<Wn,ttfyF}zt|tr0t|jts0|jj j dsAtd |jj j d}~ww|jd|_ dS) NcSs|SrHr*)_ serializedr*r*r+ just_jsonsz/Discoverer._load_server_info..just_jsonrR kubernetesr7z/version) serializerzhttps://zGHost value %s should start with https:// when talking to HTTPS endpoint) r2r7rrd ValueErrorr isinstancereasonrrr startswithrQ)r&rnrAr*r*r+r<s   zDiscoverer._load_server_infoc Cs6tt}i}dtd|||g}z |jd|jpg}Wn ty(g}Ynwttdd|} ttdd|} | D]} | ddd\} } | | sRi|| <| || | <q=| D]=} d D]}| |dq_t 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|dvSNrurbr*resourcer*r*r+ z:Discoverer.get_resources_for_api_version..cSs d|dvSrvr*rwr*r*r+ryrzrbr )rKrLrMrrj)rKrLrMrrj subresourcesrN)rLrM base_kindr*)rlistr!filterrrdrWrsplitr7poprappendrrN)r&rKrLrRrjrWr{r resources_response resources_rawsubresources_raw subresourcerxrbkey resourceobj resource_listr*r*r+rZs@    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 rMcsg|] }|jdkr|qS)rM) group_version.0resultrOr*r+ sz"Discoverer.get..cSsg|]}t|tqSr*rrr)rxr*r*r+rscSsg|] }t|ts|qSr*rrr*r*r+rsrzNo matches found for {}z!Multiple matches found for {}: {}Nr*)rPlenr7allr rr r&rOresultsr*rr+r7s  zDiscoverer.get)FNNNN)FF)__name__ __module__ __qualname____doc__r,r%r>r8rrIrrPr=propertyrRr^rkr<rZr7r*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)NFr`)rk_LazyDiscoverer__resourcesrFr*r*r+r=zLazyDiscoverer.discovercCs|jr |d|_dSdSr)rr>rFr*r*r+__maybe_write_caches z"LazyDiscoverer.__maybe_write_cachecCs|jddddS)NFTr\rerrkkeysrFr*r*r+rIszLazyDiscoverer.api_groupscKslz||jdi||jg}Wn tyg}Ynw|s0|||jdi||jg}||S)Nr*)_LazyDiscoverer__search_LazyDiscoverer__build_searchrr r8"_LazyDiscoverer__maybe_write_cacherr*r*r+rPs  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 rWT)r7rrrYrrqrWrZrjr r r2rrdictitemsgetattrrextend)r&partsrW reqParamspart resourcePartrKrLrR _resourcetermvaluematchesrr*r*r+__searchsF        $zLazyDiscoverer.__searchNcK>|s|rd|vr|d\}}|||||g}ttdd|S)NrucS|pdSNrr*rr*r*r+ry.z/LazyDiscoverer.__build_search..rr}mapr&rKrLrMrNrOrr*r*r+__build_search)zLazyDiscoverer.__build_searchc cs|jD]?\}}|D]6\}}|D]-\}}|js5|||||j|_||jd|||<d|_t|jD]\}}|Vq;qqq| dS)NrWT) rrrWrZrjr2rr iteritemsr) r&rKr]rLrarRrgrlrxr*r*r+__iter__0s   zLazyDiscoverer.__iter__r) rrrrr,r=rrrIrPrrrr*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 ||_dSrH)_EagerDiscoverer__resources)r&rWr*r*r+reFs zEagerDiscoverer.updatecCst|||dSrH)rr,rr*r*r+r,IrzEagerDiscoverer.__init__cCr)NTr`)rkrrFr*r*r+r=LrzEagerDiscoverer.discovercCs|jddddS)z list available api groups TrrrrFr*r*r+rIOszEagerDiscoverer.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_searchrr8rr*r*r+rPUs  zEagerDiscoverer.searchNcKr)NrucSrrr*rr*r*r+rykrz0EagerDiscoverer.__build_search..rrr*r*r+rfrzEagerDiscoverer.__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 ) r7rrrYrrWrrrrr) r&rrWrrrrrrrr*r*r+rms,     "zEagerDiscoverer.__searchccsV|jD]"\}}|D]\}}|D]\}}|D]\}}|VqqqqdSrH)rr)r&rlr]rarWrxr*r*r+rszEagerDiscoverer.__iter__r) rrrrrer,r=rrIrPrrrr*r*r*r+r?s   rc@s"eZdZdZdddZddZdS)rYz%Helper class for Discoverer containerNcCs||_|pi|_dSrHrjrW)r&rjrWr*r*r+r,szResourceGroup.__init__cCsd|j|jdS)NrY)_typerjrWrrFr*r*r+to_dictszResourceGroup.to_dictrH)rrrrr,rr*r*r*r+rYs  rYc@seZdZddZdS)rEcCs|SrH)r)r&or*r*r+defaultszCacheEncoder.defaultN)rrrrr*r*r*r+rEs rEc@seZdZddZddZdS)r6cOs*||_tjj|g|Rd|ji|dS)N object_hook)rr4 JSONDecoderr,r)r&rargsrOr*r*r+r,s$zCacheDecoder.__init__cCsrd|vr|S|d}|dkrtd d|ji|S|dkr&t|jfi|S|dkr7t|d||ddS|S) NrrrrrYrjrWrVr*)rrrrrYr)r&objrr*r*r+rs zCacheDecoder.object_hookN)rrrr,rr*r*r*r+r6s r6)%rrr4r:rr" functoolsr collectionsrabcrrurllib3.exceptionsrrror exceptionsr r r r rrxrrr[objectrrrrY JSONEncoderrErr6r*r*r*r+s*   :cO