o K@sddZddlmZddlmZddlmZddlZddlmZddlm Z ddl Z Gdd d e Z dS) z-Calliope argparse argument completer objects.)absolute_import)division)unicode_literalsN)resource_cache) console_attrc@sfeZdZdZdddZeddZeddZdd d Z dd d Z dddZ dddZ dddZ dS)ArgumentCompletera Argument completer wrapper to delay instantiation until first use. Attributes: _argument: The argparse argument object. _completer_class: The uninstantiated completer class. _parsed_args: argparse parsed_args, used here if not known at __call__ time. NcCs@ddlm}||_||_||_dtjvr|j|_dS|j |_dS)Nr)progress_tracker _ARGCOMPLETE) googlecloudsdk.core.consoler_completer_class _argument _parsed_argsosenvironCompletionProgressTracker_progress_trackerProgressTracker)selfcompleter_class parsed_argsargumentrrE/tmp/google-cloud-sdk/lib/googlecloudsdk/calliope/parser_completer.py__init__&s    zArgumentCompleter.__init__cCs|jSN)r )rrrrr4sz!ArgumentCompleter.completer_classcs&t}|\}fdd|DS)z@Returns a msgs list that will display 1 per line as completions.cs$g|]}|dt|dqS) )len).0msgwidthrr >s$zBArgumentCompleter._MakeCompletionErrorMessages..)rGetConsoleAttr GetTermSize)clsmsgsattr_rr r_MakeCompletionErrorMessages8s z.ArgumentCompleter._MakeCompletionErrorMessagesc CsB|r t|dr |j}n|jj}|d||d|t|gS)aHandles completer errors by crafting two "completions" from exception. Fatal completer errors return two "completions", each an error message that is displayed by the shell completers, and look more like a pair of error messages than completions. This is much better than the default that falls back to the file completer with no indication of errors, typically yielding the list of all files in the current directory. NOTICE: Each message must start with different characters, otherwise they will be taken as valid completions. Also, the messages are sorted in the display, so the messages here are displayed with ERROR first and REASON second. Args: exception: The completer exception. prefix: The current prefix string to be matched by the completer. completer: The instantiated completer object or None. Returns: Two "completions" crafted from the completer exception. collectionz&{}ERROR: {} resource completer failed.z {}REASON: {})hasattrr*r __name__r)formatsix text_type)r exceptionprefix completercompleter_namerrr_HandleCompleterException@s z+ArgumentCompleter._HandleCompleterExceptionc Kst|jts |j|dS|s|j}|,t}|j|||dWdWdS1s4wYWddS1sDwYdS)z+A completer function suitable for argparse.r1)r1cacherN) isinstancer type_CompleteFromFunctionr rr ResourceCache_CompleteFromCompleterClass)rr1rkwargsr7rrr__call___s    "zArgumentCompleter.__call__c Cs@z||WSty}z |j||dWYd}~Sd}~ww)z-Helper to complete from a function completer.r6Nr BaseExceptionr4)rr1errrr:ls  z'ArgumentCompleter._CompleteFromFunctionc CsNd}z |}||dWSty&}z|j|||dWYd}~Sd}~ww)z=Helper to complete from a class that isn't a cache completer.Nr6r1r2r?)rr1r2rArrr"_CompleteFromGenericCompleterClassss z4ArgumentCompleter._CompleteFromGenericCompleterClassc Cs|rt|jjdkrdh}nt}d}z|j||d}|||j}|||WSt yW}z!t |t rE|sE|j |dWYd}~S|j |||dWYd}~Sd}~ww)z Helper to complete from a class.r*N)r7qualified_parameter_namesr6rB)r _GetCommandaipositional_completerssetr ParameterInfor Completer@r8 TypeErrorrCr4)rr1r7rrEr2parameter_inforArrrr<~s(z-ArgumentCompleter._CompleteFromCompleterClass)NNr)r5N)r5)r5NN)r, __module__ __qualname____doc__rpropertyr classmethodr)r4r>r:rCr<rrrrrs       r) rP __future__rrrrgooglecloudsdk.core.cacherr rr.objectrrrrrs