-DdZddlmZddlmZddlmZddlmZddlmZddlZddl Z ddl m Z dd l mZddlZd Zd j#e Ze j&eZd ZGddeZGddeZGddeZGddeZGddeZdZdZdZy)zResource definition generator.)absolute_import)division)print_function)unicode_literals) OrderedDictN)resource)filesz)[a-zA-Z][a-zA-Z0-9_]+(?:\.[a-zA-Z0-9_]+)+z!(?P{collection})\.get) collectionceZdZdZy)ErrorzErrors raised by this module.N__name__ __module__ __qualname____doc__6lib/googlecloudsdk/api_lib/regen/resource_generator.pyr r &s%rr ceZdZdZy)NoMatchingMethodErrorz,Raised when no matching method can be found.Nrrrrrr*s4rrceZdZdZy)UnsupportedDiscoveryDocz1Raised when some unsupported feature is detected.Nrrrrrr.s9rrceZdZdZy)ConflictingCollectionz>Raised when collection names conflict and need to be resolved.Nrrrrrr2sFrrceZdZdZdZedZedZedZ edZ edZ dZ d Z d Z dd Zd ZdZy ) DiscoveryDocz%Encapsulates access to discovery doc.c||_yN_discovery_doc_dict)selfdiscovery_doc_dicts r__init__zDiscoveryDoc.__init__9s 1Drctj|5}|tj|tcdddS#1swYyxYw)N)object_pairs_hook)r FileReaderjsonloadr)clspathfs rFromJsonzDiscoveryDoc.FromJson<s1   $ 1 1 < =  s !AA c |jdS)Nnamer r"s rapi_namezDiscoveryDoc.api_nameAs  # #F ++rc |jdS)Nversionr r0s r api_versionzDiscoveryDoc.api_versionE  # #I ..rc |jdS)NbaseUrlr r0s rbase_urlzDiscoveryDoc.base_urlIr5rc |jdS)NdocumentationLinkr r0s rdocs_urlzDiscoveryDoc.docs_urlMs  # #$7 88rc |j||j}|rtj|djd}|D]~}|tj|jdk7s)t dj |j||dj|dj|j|j|j|j||||S)aReturns all resources collections found in this discovery doc. Args: custom_resources: {str, str}, A mapping of collection name to path that have been registered manually in the yaml file. api_version: Override api_version for each found resource collection. Returns: list(resource_util.CollectionInfo). Raises: UnsupportedDiscoveryDoc: if collections have different base URLs. rzWIn client {0}/{1}, collection {2} is using url {3}, but collection {4} is using url {5}) _ExtractResourcesr! resource_utilSplitEndpointUrlr8rformatr1r/extend!_GenerateMissingParentCollections)r"custom_resourcesr4 collectionsurl_api_versioncs rGetResourceCollectionsz#DiscoveryDoc.GetResourceCollectionsQs((T--/K%66 a. ! !##$&o! m<rB) r"r4infosrErJ get_methodcollection_info_infosubresource_collectionss rr>zDiscoveryDoc._ExtractResourcestsK yy )$g;;u%j 77 %?+ yyYY{+113'!T"&"8"8d"K234 rct|d}|jddd}|jd}|jd}|j||||S)z>Created collection_info object given discovery doc get_method.id.r=flatPathr+)_ExtractCollectionNamesplitrK_MakeResourceCollection)r"r4rQcollection_name flat_pathr+s rrMz%DiscoveryDoc._GetCollectionFromMethodsb,Z-=>O&++C3A6Oz*I >>& !D  ' '_dI rNc T||k(rd}|j|z}tj|\}}}|r&tj|j|z\}}}|dt| }tj||||j |||rt |initj|S)z8Make resource collection object given its name and path.N)r8r?r@lenCollectionInfor;DEFAULT_PATH_NAMEGetParamsFromPath)r"r4r]r+r^url url_api_namerSs rr\z$DiscoveryDoc._MakeResourceCollectionsDi --$ C)::3?L!T%66 --) #oaI kD z C  ' '   *3 I&''-  rc l|Dcic]}|j|}}|Dchc]}|jt}}g}t|}g} i} |r|D]}t |\} } | s| |vr| |vr*| j | t j| G| |vrJtdj|j||j| || jt| |j| | d|} | j| | || <|j| |j| | }g} |rtj| D]V\}}t!|dkDr|j#}|||dk(s1t%dj||j|X|Scc}wcc}w)aGenerates parent collections for any existing collection missing one. Args: collections: [resource.CollectionInfo], The existing collections from the discovery doc. custom_resources: {str, str}, A mapping of collection name to path that have been registered manually in the yaml file. api_version: Override api_version for each found resource collection. Raises: ConflictingCollection: If multiple parent collections have the same name but different paths, and a custom resource has not been declared to resolve the conflict. Returns: [resource.CollectionInfo], Additional collections to include in the resource module. zIn API [{api}/{version}], the parent of collection [{c}] is not registered, but a collection with [{parent_name}] and path [{existing_path}] already exists. Update the api config file to manually add the parent collection with a path of [{parent_path}].)apir3rG parent_name existing_path parent_pathTr=r+z:WARNING: Custom resource [{}] in API [{}/{}] is redundant.)r/GetPathrblist_GetParentCollection setdefaultsetaddrrAr1MakeResourceCollectionrNrBsix iteritemsr`popprint)r"rErDr4rG all_names all_paths generated in_progress to_processignoredrhrjparent_collectionr/pathsr+s rrCz.DiscoveryDoc._GenerateMissingParentCollectionss*%00KqKI07BC{!,-{ICI{#KJG !#7#: [  ) #  * *   [#% 0 4 4[ A  ) #&!"(jj+)K(001BC) "("+ , ,!77 dK9+,!2 + k"7:z"kjE L}}W- e Ua  YY[d !$'/ / KtT]]K8 ;. o1Cs F,F1c &tj|j\}}}|r |j}ndj|j|}tj|j |||j ||itj|| S)Nz{}{}/)r?r@r8rArar1r;rc)r"r]r+enable_uri_parsingr4rSrFr8s rrqz#DiscoveryDoc.MakeResourceCollectionsx)::4==IAh {;h  ' ' {Hdmmr=#B#B4#H rr)rrrrr$ classmethodr-propertyr1r4r8r;rHr>rMr\rCrqrrrrr6s-2>> , , / / / / 9 9!F" ;?8L\ rrctj|}|r|jdStdj |t )z4Extract the name of the collection from a method ID.r z%Method {0} does not match regexp {1}.) _METHOD_ID_REmatchgrouprrA_METHOD_ID_RE_RAW) method_idrs rrZrZsA   i (% ;;| $$ /  , - //rc|jt}t|dkry|jt}|j d}t |dt |d|sydj |}t |d|syd|jvr#|jjdd\}}||fS|d}||fS) aGenerates the name and path for a parent collection. Args: collection_info: resource.CollectionInfo, The collection to calculate the parent of. Returns: (str, str), A tuple of parent name and path or (None, None) if there is no parent. )NN/TFrXr=) GetParamsrbr`rkr[ _PopSegmentsjoinr/rsplit)rRparamsr+partsrjrhrSs rrmrms  $ $%6 7&[1_   !2 3$ **S/%udue  +ud  O  %))00a8NK k !!)K k !!rc|rt|djd|k(r\|djd|k(rD|j|sy|djd|k(r|djd|k(rByyyyy)Nr{}) startswithendswithrt)r is_paramss rrrAs 9   $ 1 9  c "i / iik  9   $ 1 9  c "i / 2 / 2 r) r __future__rrrrrErr(regooglecloudsdk.api_lib.utilrr?googlecloudsdk.core.utilr rr_COLLECTION_SUB_RErArcompilerrb Exceptionr rrrobjectrrZrmrrrrrs%&%'# A* B8??!@# ,- &I&5E5:e:GEGO6Od/+"\r