.dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z dd l mZdd lmZdd lmZddlZd ZGd deZGddeZdZdZdZdZdZdZdZdZy)z+Utility wrappers around apitools generator.)absolute_import)division)unicode_literalsN) gen_client)api_def)resource_generator)files)runtime)templatean# -*- coding: utf-8 -*- # # Copyright 2016 Google LLC. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. ceZdZdZy)NoDefaultApiErrorz;Multiple apis versions are specified but no default is set.N__name__ __module__ __qualname____doc__,lib/googlecloudsdk/api_lib/regen/generate.pyr r 5sCrr ceZdZdZy)WrongDiscoveryDocErrorzUnexpected discovery doc.Nrrrrrr9s!rrcptjg}|jd}|r2|jdj dj |d|j dddj |dj tjj |||dd d d j |d d j |||dg tjd|tj|tjj|\}} | ||fD]} tjj || }tjj |d} tjj| rctjd|tj | t"y)z)Invokes apitools generator for given api.unelidable_request_methodsz --unelidable_request_methods={0},z--init-file=emptyz--nogenerate_cliz --infile={0}z --outdir={0}z --overwritez--apitools_version=CloudSDKz--user_agent=google-cloud-sdkz--version-identifier={0}z--root_packagez {0}.{1}.{2}clientzApitools gen %sz __init__.pyz1%s does not have __init__.py file, generating ...N)r__file__getappendformatjoinextendospathloggingdebugmainsplitisfilewarningr WriteFileContents_INIT_FILE_CONTENT) discovery_doc output_dir root_packageapi_name api_version api_configargsr package_dirdir_namesubdir init_files rGenerateApitoolsApir7=sf    $)~~.JKKK299 89:<=++M*BGGLLX{KL#% '' 4<;?   --!4( //$''-- 3+x8[1f'',,{F3K [-8I 77>>) $ ooI!# i);< 2rcPdjd|jdDS)Nc3<K|]}|jyw)N) capitalize).0xs r z_CamelCase..bs?)>A)>s_)r r') snake_cases r _CamelCaserAas" ?)9)9#)>? ??rcxdj|||g}|dk(r|dk(rd}n:djdj||dgt|t|zg}dj||dg}d}|dz|z} t|}|j}t j |||| } | S#t$rY&wxYw) zMakes an ApitoolsClientDef..adminv1z admin_v1_client.AdminDirectoryV1r?rmessagesr9) class_pathclient_classpathmessages_modulepathbase_url)r rA_GetClientClassFromDefBASE_URL ExceptionrApitoolsClientDef) r.r/r0rGrHrIrJclient_full_classpathclient_classpath_def apitools_defs r_MakeApitoolsClientDefrResxxx=>* [D09xx (K238z+66!89(K!DE ($s*-== 12GH#,,H **'- ,    s;B-- B98B9c`|jdd\}}t||g}t||S)zFReturns the client class for the API definition specified in the args.rC)fromlist)rsplit __import__getattr)rO module_pathclient_class_name module_objs rrKrKs8#8#?#?Q#G + +1B0CD* . //rcdj|jddd}dj|d||g}tj|S)zMakes a GapicClientDef.rCNgapic_wrappers)rG)r r'rGapicClientDef)r.r/r0gapic_root_packagerGs r_MakeGapicClientDefrasUxx 2 23 7 <=xx+X{CE*    rc i}|jD]`\}}|jD]H\}}|jdst|dk(s&||vrtdj ||||<Jbt |j j|j }|r2tdj djt|tjt}|jD]\}}|jD]\}}|j|ij|d)|||} |jdrt| ||} nd} |jdrt| ||} nd} |||k(} |jd d } |jd d }tj | | | | ||||<|S) aCombines package_map and api_config maps into ApiDef map. Args: package_map: {api_name->api_version->root_package}, package where each generated api resides. apis_config: {api_name->api_version->{discovery,default,version,...}}, description of each api. Returns: {api_name->api_version->ApiDef()}. Raises: NoDefaultApiError: if for some api with multiple versions default was not specified. defaultrTz0Multiple default client versions found for [{}]!z+No default client versions found for [{0}]!z, Nr,gcloud_gapic_library enable_mtlsTmtls_endpoint_overrider9)itemsrlenr rsetkeys differencer sorted collections defaultdictdictrRrarAPIDef) package_map apis_configdefault_versions_mapr/api_version_configr0r1apis_without_defaultapis_mapr.apitools_client gapic_clientrcrerfs r _MakeApiMaprys &1&7&7&9"h"#5#;#;#= Z  "c*<&=&B + +!@vh! !*5X& $>': +   (()=)B)B)DE I#VDIIf5I.J$KL NN $ $T *(&1&7&7&9"h"#5#;#;#= Z 2 & * *; = E *;7l  (0 (K  . /*<;O   4X >>gNN=$7k)~~.FK(/   ; 6)8hx%#$>':, /rcTtjjtj\}}t j |dz}tjtjjtjjtd}tjd|t||}tjd|t j|5}tj |||} |j#| dddy#1swYyxYw)a"Create an apis_map.py file for the given packages and api_config. Args: output_file: Path of the output apis map file. package_map: {api_name->api_version->root_package}, packages where the generated APIs reside. api_config: regeneration config for all apis. z.pyz template.tplfilenamezGenerating api map at %szCreating following api map %s)api_def_sourcervN)r"r#splitextrrr ReadFileContentsr Templater dirnamer$r%ry FileWriterr Contextrender_context) output_filerqr1api_def_filenamer?r}tplapi_map apis_map_filectxs rGenerateApiMaprs(()9)9:A))*:U*BC.ww||BGGOOH5~F # --*K8  Z 0' --/9  $ //nw Cs %$$s +*DD'ctjj|}|j|k7r!t j d|j||j |k7r%tdj|j ||j||}|rKtg}|D]}|j|vs||jjdd}|s4|j|j||jd} t| tr|j j#| t| t$j&s| |j tj(<t|j+|z D]X} || } | d} | jdd}|s"| jdd} |j-| | | |}|j/|Zt0j2j5|||}t0j2j7|st1j8|t0j2j5|d}|rt j:d|t=j>t0j2j5t0j2jAtBd  }tEjF|5}tIjJ|tM||d jN|jP }|jS|d d d y t0j2jU|r,t j:d|t1jV|y y #1swYy xYw)aCreate resource.py file for given api and its discovery doc. Args: discovery_doc: str, Path to the discovery doc. output_dir: str, Path to the base output directory (module will be generated underneath here in api_name/api_version subdir). api_name: str, name of the api. api_version: str, the version for the api. custom_resources: dict, dictionary of custom resource collections. Raises: WrongDiscoveryDocError: if discovery doc api name/version does not match. zYDiscovery api version %s does not match %s, this client will be accessible via new alias.zapi name {0}, expected {1}apitools_compatibleTr#enable_uri_parsingz resources.pyz Generating resource module at %sz resources.tplr{r)rmrJdocs_urlNz'Removing existing resource module at %s),r DiscoveryDocFromJsonr0r$r)r/rrGetResourceCollectionsrinameradd isinstancero flat_pathsupdatesix string_typesDEFAULT_PATH_NAMErjMakeResourceCollectionrr"r#r existsmakedirsr%r rrrr rr rrlrJrrr(remove)r,r-r/r0custom_resourcesresource_collectionsmatched_resources collectionr custom_pathcollection_namecollection_defcollection_pathrcollection_infoapi_dirresource_file_namerrrs rGenerateApitoolsResourceModulers(%11::=I-+- OO 8!!;0x' !=!D!D"* ++&== %B* , ,.z?CC !4 # joo.&z7? k4 (    & &{ 3  S%5%5 66A    22 4+/4467:KK'8n&v.o*.. !)--.BDI%<< ?,> Mo!!/2L GGLLX{ ;'  KKww||G^< MM46HI   RWW\\"''//(2K2A&C DC   , - OOK(./C(D%9!%<%E%E%2%;%; =c  . -  ww~~() MM;=OPII !* . -s A M;;N)r __future__rrrrmr$r" apitools.genrgooglecloudsdk.api_lib.regenrrgooglecloudsdk.core.utilr makor r rr+rMr rr7rArRrKraryrrrrrrs2&' #0;* &D D"Y"!=H@:06r6O"r