dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd l mZddlZGd d e j ZGd deZdZGdde j ZdZdZGdde j.Zy)zDCommon command line processing utilities for access context manager.)absolute_import)division)unicode_literalsN)encoding)waiter)base) exceptions)yamlc"eZdZdZfdZxZS)ParseFileErrorz-Error raised when a file could not be parsed.cLtt| dj||y)zInitializes a ParseFileError. Args: path: The path of the file that could not be parsed. reason: The reason the file could not be parsed. zIssue parsing file [{}]: {}N)superr __init__format)selfpathreason __class__s =lib/googlecloudsdk/command_lib/accesscontextmanager/common.pyrzParseFileError.__init__"s$ .$(%,,T6:__name__ __module__ __qualname____doc__r __classcell__rs@rr r s5  rr c"eZdZdZfdZxZS)InvalidMessageParseErrorzAError raised when a message could not be parsed from a YAML file.c |jDcgc]}|j}}tt||dj ||rdnddj |ycc}w)a,Initializes an InvalidMessageParseError. Args: path: The path of the file that could not be parsed. reason: The reason the file could not be parsed. message_class: The message class that could not be parsed. pluralize_error: Whether the error meessage is pluralized. zfThe YAML-compliant file provided contains errors: {} The {} in this file can contain the fields [{}].objectsobjectz, N) all_fieldsnamerrrrjoin)rrr message_classpluralize_errorf valid_fieldsrs rrz!InvalidMessageParseError.__init__1sh%2$<$<$>?$>qAFF$>L? "D2   & (Ih IIl #  @sA"rrs@rrr.sIrrc:tj|}|s t|d |r%|Dcgc]}tj||}}|Stj||g} |Scc}w#t $r&}t |tj|||d}~wwxYw)aParse a YAML representation of a message(s). Args: path: str, path to YAML file containing data to parse message_class: obj, message type to parse the contents of the yaml file to is_list: bool, whether the file contains a list of messages or a single message Returns: list of message object(s). Raises: ParseFileError: if the file could not be read into the proper object(s) z File is emptyN) r load_pathr r DictToMessage Exceptionrsix text_type)rr&is_listdatacmessageserrs r)ParseAccessContextManagerMessagesFromYamlr5Ls  $  //DHIDq(((M:DhI / ((}=>h /J  " cmmC - s.A+A&A+ A+&A++ B4!BBceZdZfdZxZS)ParseResponseErrorcJtt| dj|y)NzIssue parsing response: {})rr7rr)rrrs rrzParseResponseError.__init__ns"  5<s r GetTitleArgrBzs$  2 9 9$ ? rc(eZdZdZfdZdZxZS)BulkAPIOperationPollerzA Poller used by the Bulk API. Polls ACM Operations endpoint then calls LIST instead of GET. Attributes: policy_number: The Access Policy ID that the Poller needs in order to call LIST. ctt| ||tjd|j }|r|j |_ytd)Nz^accessPolicies/\d+z=Could not determine Access Policy ID from operation response.) rrDrresearchNamegroup policy_numberr7)rresult_serviceoperation_service operation_ref policy_idrs rrzBulkAPIOperationPoller.__init__sY $01BD 0-2D2D2FGI$??,d  5 66rc~|jjd}|jj||jS)NList)parent)rKGetRequestTyperPrJ)r operation request_types r GetResultz BulkAPIOperationPoller.GetResults=&&55f=L    # #L8J8J$K LLr)rrrrrrUrrs@rrDrDs 6MrrD)r __future__rrrrFapitools.base.pyrgooglecloudsdk.api_lib.utilrgooglecloudsdk.calliopergooglecloudsdk.corer r r.Errorr rr5r7r@rBCloudOperationPollerrDrrr^sK&' %.(*$  Z%% ~<@F))FMV88Mr