dZddlmZddlmZddlmZddlZddlmZddlm Z ddl Z ejZ ejZ ejZ ejZejZej Zej"Zej$Zej&ZdZej*Zej,Zej.Zej0Zej2ZGd d ej4ZGd d eZGd deZy)zA basic command line parser. This command line parser does the bare minimum required to understand the commands and flags being used as well as perform completion. This is not a replacement for argparse (yet). )absolute_import)division)unicode_literalsN)cli_tree)lexerzinteractive.is_specialc,eZdZdZdZdZdZdZdZdZ dZ y ) ArgTokenTyperN) __name__ __module__ __qualname__UNKNOWNPREFIXGROUPCOMMANDFLAGFLAG_ARG POSITIONALSPECIAL4lib/googlecloudsdk/command_lib/interactive/parser.pyr r 5s* ' & % ' $ (* 'rr c@eZdZdZej dddfdZdZdZy)ArgTokena#Shell token info. Attributes: value: A string associated with the token. token_type: Instance of ArgTokenType tree: A subtree of CLI root. start: The index of the first char in the original string. end: The index directly after the last char in the original string. NcJ||_||_||_||_||_yN)value token_typetreestartend)selfr#r$r%r&r's r__init__zArgToken.__init__Ks%DJ DODIDJDHrcbt||jr|j|jk(Sy)zEquality based on properties.F) isinstance __class____dict__)r(others r__eq__zArgToken.__eq__Ss&%( ]]enn ,, rczdj|j|j|j|jS)zImprove debugging during tests.zArgToken({}, {}, {}, {}))formatr#r$r&r')r(s r__repr__zArgToken.__repr__Ys0 % , ,TZZ-1ZZ CCr) rrr__doc__r rr)r/r2rrrr r @s)(4';';$t Crr c,eZdZdZddZddZdZdZy) ParserzShell command line parser. Attributes: args: context: cmd: hidden: positionals_seen: root: statement: tokens: Nc||_||_g|_|j|_d|_d|_d|_d|_|j|y)Nr) roothiddenargscmdpositionals_seen previous_line statementtokens SetContext)r(r7contextr8s rr)zParser.__init__msLDIDKDIyyDHDDDNDKOOGrc@tj|xsd|_y)z(Sets the default command prompt context.N)six text_typer@)r(r@s rr?zParser.SetContextzs==B/DLrc  tj||_|j|_d|_g|_d}|jr1|jjd}|j}|jtjjk(rid}|j|_|j jt|tj|j|j |j"nh|jtjj$k(r|j'||n-|jtjj(k(r|s||jt*vr|jt*||_|jt,rtj.}nDt0|jvr"tj}|j|_ntj2}|j jt|||j|j |j"n"|j|jk(rmd|vritj}|j|_|j jt|||j|j |j"n|j t5|jt6kr|jt6|j }|j jt|tj8||j |j"|t:dvr|xj dz c_n|s |j Sd}|j|jk(rtj<}ntj>}|j jt|||j|j |j"nVd}|j jt|tj>|j|j |j"|jr1|j S)aParses the next command from line and returns a list of ArgTokens. The parse stops at the first token that is not an ARG or FLAG. That token is not consumed. The caller can examine the return value to determine the parts of the line that were ignored and the remainder of the line that was not lexed/parsed yet. Args: line: a string containing the current command line Returns: A list of ArgTokens. rF=)*+r T) rGetShellTokensr>r7r:r;r9pop UnquotedValuelexShellTokenType TERMINATORappendr r rr&r'r ParseFlagARGLOOKUP_COMMANDSLOOKUP_IS_GROUPrLOOKUP_IS_SPECIALrlenLOOKUP_POSITIONALSr LOOKUP_NARGSrr)r(lineunknowntokenr#r$ positionals r ParseCommandzParser.ParseCommand~sB&&t,DKyyDHDDIG ++kkooa e!!#e e**55 599 %)=)=txx"'++uyy: ; 99,,11 1 ue$ 99,,00 0 DHH_- -XXo.u5$( XXo &%++J DHH,%--JyyDH%--J ))  8E:txx$)KK< =XX "se|#++*YY$( ))  8E:txx$)KK< = " "S2D)E%F Fxx 23D4I4IJ* ))  8E<+B+B$. UYYH I  %Z 7  ! !Q & !  99' XX "%,,J%--J ))  8E:txx$)KK< = %)=)=txx"'++uyy: ;k ++p 99rc h|j}|j}d}d}d}d|vr6|jdd\}}|t|z}|dz}|t|z}|jt j |}|r|js^|trU|jjt|tj|j|j|jy|tdk7rQ|O|jrC|jj!d}|j#}|j}|j}|jjt|tj$||||7|jjt|tj&d||yy)z5Parses the flag token and appends it to the arg list.NrFr boolr)r&r'splitrUr: LOOKUP_FLAGSgetr8LOOKUP_IS_HIDDENr9rOr r r LOOKUP_TYPEr>rJrKrr) r(rZname name_startname_endr# value_start value_endflags rrPzParser.ParseFlagspJyyH EKI d{JJsA&kdEc$i'hqLkE *i 88L ! % %d +D t{{t,<'= iixl&:&:DHH % UYY89  KF"u}kkooa e!!#eKKk))iIIXdL$5$5t((45  iix|'<'rws '' ,8 "" **(( 66,,$$ && **,,,"" $$ 00"" 00499CvC>O9VO9r