- dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z dZdZdZd Zd Zd Zd Zd ZdZdZdZdZGddej4ZddZedk(rej<eyy)z%Command-line interface to gen_client.N) exceptions)gen_client_lib)utilctjtj|d5}t j d|}|t jd|z|j|dddy#1swYyxYw)Nwapitools.base.pyzCould not find file %s) contextlibclosingioopenpkgutilget_datarGeneratedClientErrorwrite)filenameoutsrc_datas *lib/third_party/apitools/gen/gen_client.py_CopyLocalFiler!sk   BGGHc2 3s## *  11(835 5 ( 4 3 3s AA55A>c|jr tj|jSt jj|jxsd}tj|d5}tjtj|jcdddS#tj$rtj dwxYw#1swYyxYw)z!Get the discovery doc from flags.zCould not fetch discovery docz /dev/stdinutf8encodingN) discovery_urlrFetchDiscoveryDocrCommunicationErrorrospath expanduserinfiler r jsonloadsReplaceHomoglyphsread)argsr fs r_GetDiscoveryDocFromFlagsr'+s  1))$*<*<= = WW   , < F & )Qzz$00:; * ) ,, 111/1 1 1 * )sB116C1)CC&c t|}tj|j|j|j }|j r tj|j d5}tjtj|j}|jd|jdi}|jd}|jd}dddn|j"}|j$}st'j(d d }st'j(d d }tj*j-||j.|||j0||j2|j4}t6j8j;|j<xs |j>} t6j8jA| r!|jBstjDd t6j8jA| st7jF| tIjJ||||jL| |jN|jP|jRd k(|jT|jV|jX S#1swYxYw#t$r#tj d|j zwxYw)z#Create a codegen object from flags.rr installedweb client_id client_secretNz#Failed to open client json file: %szNo client ID suppliedzNo client secret suppliedzHOutput directory exists, pass --overwrite to replace the existing files. wildcards) base_packageprotorpc_packageinit_wildcards_file use_proto2unelidable_request_methodsapitools_version)-r'rNames strip_prefixexperimental_name_conventionexperimental_capitalize_enums client_jsonr r r!r"r#r$getIOErrorr NotFoundErrorr+r,loggingwarning ClientInfoCreatescope user_agentapi_keyversion_identifierrrroutdirdefault_directoryexists overwriteConfigurationValueErrormakedirsrDescriptorGenerator root_packager/r0 init_fileexperimental_proto2_outputr3r4) r% discovery_docnamesr9r&r*r+r, client_inforEs r_GetCodegenFromFlagsrR9s@-d3M JJ  )) ** ,E   J))F;{JJt55k6F6F6HIJeeKub)9:GGK0 # 8 <NN ** /0 34 //((tzz9m  d.E.EGKWW   , M 0M0MF ww~~fdnn00 "# # 77>>& ! F  - -{E4+<+!C 2B; 7C >C C  Cc|jr t|tj|j5t j |jjd5}|j|dddt j |jjd5}|j|ddddddy#1swYVxYw#1swYxYw#1swYyxYwr^) r2rcrrTrEr r rQmessages_file_nameWriteMessagesFileclient_file_nameWriteClientLibrary)r%rUrs r_WriteGeneratedFilesris! GNN # WWW((;;S AS  % %c *B WWW((993 ?3  & &s +@ $ # A A ? ? $ #s;+C!"C 42C!&C8C! C C!C C!!C*ctj|j5tjdd5}|j |ddddddy#1swYxYw#1swYyxYwrX)rrTrEr r WriteInitrZs r _WriteInitrlsM GNN # WW]C (C   c ") $ # ( ( $ #s"A&A A&A# A&&A/c|tjdd5}|j|dddy#1swYyxYw)Nzsetup.pyr)r r WriteSetupPyrZs r _WriteSetupPyros, S !SS! " ! !r\ct|}|tjdyt|||jdk7r t |yy)z"Driver for client code generation.N"Failed to create codegen, exiting.none)rRr=errorrirMrlr%rUs rGenerateClientrvsF#4(G :;w' ~~7 ct|}|d}tjj|j}tjj |jd|z|_d|z|_t|}|tjdyt||t|tj|5t|tjd5t|tjd5t|dddddddddy#1swYxYw#1swYxYw#1swYyxYw) z/Generate a client as a pip-installable tarball.namezapitools/clients/%szapitools.clients.%sNrqapitoolsclients)r'rrrrEjoinrLrRr=rtrirlrrTror[)r%rOpackageoriginal_outdirrUs rGeneratePipPackagers.d3MF#Ggg((5O'',, *W46DK-7D"4(G :;w'w O $g ZZ # "7 +I&&w/'$ % $'&$ # % $s<!E&!D8 D,D8E,D5 1D88E =EE c0t|}t|y)z2Generate just the two proto files for a given API.N)rRrcrus r GenerateProtors#4(GWrwceZdZddZy)_SplitCommaSeparatedListNcPt||j|jdy)N,)setattrdestsplit)selfparser namespacevalues option_strings r__call__z!_SplitCommaSeparatedList.__call__s 499fll3&78rwN)__name__ __module__ __qualname__rrwrrrs9rwrc|tj}tjd}|j }|j dd|j dd|j dd d |j d d d |j dd|j ddd |j dddd|j ddd |j ddgd|j dd |j d!d"|j d#d$d% |j d&d'd( |j d)dgd*|j d+d,d- |j d.d/dd01|j d2d/d3d01|j d4gd5d6d7d89|j d:t gd;<|j d=dd>d?@|j dAdddB|j dCtjjtjjdDE|j dFdddG|jdH}|jdIdJ}|jtK|jdLdM}|jtK|jdNdO}|jt K|j#|dPd}|j%|xsdQS)RNzApitools Client Code Generator) descriptionz--infilezLFilename for the discovery document. Mutually exclusive with --discovery_url)helpz--discovery_urlz[URL (or "name.version") of the discovery document to use. Mutually exclusive with --infile.z--base_packagerz;Base package path of apitools (defaults to apitools.base.py)defaultrz--protorpc_packagezapitools.base.protorpclitezEBase package path of protorpc (defaults to apitools.base.protorpclitez--version-identifierzVersion identifier to use for the generated client (defaults to "version" value in discovery doc). This must be a valid identifier when used in a Python module name.z--outdirr-zOnly overwrite the output directory if this flag is specified.)ractionrz--root_packagezCPython import path for where these modules should be imported from.z--strip_prefix*z]Prefix to strip from type names in the discovery document. (May be specified multiple times.))nargsrrz --api_keyzAPI key to use for API access.z --client_jsonzTUse the given file downloaded from the dev. console for client_id and client_secret.z --client_id CLIENT_IDz*Client ID to use for the generated client.z--client_secret CLIENT_SECRETz'Client secret for the generated client.z--scopezKScopes to request in the generated client. May be specified more than once.z --user_agentx_Tw5K8nnjoRAqULM9PFAC2bzEUser agent for the generated client. Defaults to -generated/0.1.z--generate_cli generate_clizIgnored.)rrrz--nogenerate_cli store_falsez --init-file)rsemptyr.c"|jSr)lower)ss rzmain..)s qwwyrwr.z>Controls whether and how to generate package __init__.py file.)choicestyperrz--unelidable_request_methodsztFull method IDs of methods for which we should NOT try to elide the request type. (Should be a comma-separated list.)rrrz--apitools_versionr4zzApitools version used as a requirement in generated clients. Defaults to version of apitools used to generate the clients.)rrrz--experimental_capitalize_enumsz:Dangerous: attempt to rewrite enum values to be uppercase.z--experimental_name_conventionz6Dangerous: use a particular style for generated names.)rrrz--experimental_proto2_outputz-Dangerous: also output a proto2 message file.zType of generated codeclientz.Generate apitools client in destination folder)func pip_packagez$Generate apitools client pip packageprotozGenerate apitools client protosrzr)sysargvargparseArgumentParseradd_mutually_exclusive_group add_argumentrrr5NAME_CONVENTIONSDEFAULT_NAME_CONVENTIONadd_subparsers add_parser set_defaultsrvrr parse_argsr)rrdiscovery_group subparsers client_parserpip_package_parser proto_parserr%s rmainrs |xx  $ $46F99;O   !"   2!4  " JL  ,8:  >@   KM  l MO  )+  35  .1 -/   9;   68  13  *13  ~l    .  M O &'K M +NP  )l IK  ( ++ 22 E G  &l <> &&,D&EJ))G*IMN3#..B/D##);#<((7)9L=1   T!"X &D 99T? arw__main__r)__doc__rr r r!r=rr rapitools.base.pyr apitools.genrrrr'rRrVr[rcrirlrorvrrActionrrrexitrrwrrs",   '' <00h(+ 0,# " 009x9 Q h z CHHTVrw