ndZddlmZddlmZddlmZddlmZddlmZGddejZ y ) #Refine server response for display.)absolute_import)division)unicode_literals)encoding)utilcXeZdZdZfdZedZedZedZdZ xZ S) DisplayHelperrc`tt| tj|||_y)N)superr __init__r MessageToDict_job)selfjob __class__s 5lib/googlecloudsdk/api_lib/dataproc/display_helper.pyr zDisplayHelper.__init__s$ -'(>(>s(CDDIc$|jdS)NJobgetTruncatedFieldNameBySuffixrs rjobTypezDisplayHelper.jobType!s  - -e 44rc$|jdS)NBatchrrs r batchTypezDisplayHelper.batchType%s  - -g 66rc$|jdS)NSessionrrs r sessionTypezDisplayHelper.sessionType)s  - -i 88rc:|jjDcgc]}|jc}D]H}|j|s|j |\}}}|jj |sF|cSt dj|cc}w)aPGet a field name by suffix and truncate it. The one_of fields in server response have their type name as field key. One can retrieve the name of those fields by iterating through all the fields. Args: suffix: the suffix to match. Returns: The first matched truncated field name. Raises: AttributeError: Error occur when there is no match for the suffix. Usage Example: In server response: { ... "sparkJob":{ ... } ... } type = helper.getTruncatedFieldNameBySuffix('Job') z(Response has no field with {} as suffix.)r all_fieldsnameendswith rpartitionget_assigned_valueAttributeErrorformat)rsuffixfieldtoken_s rrz+DisplayHelper.getTruncatedFieldNameBySuffix-s6+/))*>*>*@A*@%***@A  &&v. q! 99 ' ' ., B C &. ** BsB) __name__ __module__ __qualname____doc__r propertyrrr r __classcell__)rs@rr r sL+ 5 5 7 7 9 9!*rr N) r0 __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.dataprocrBunchr rrr8s) *&'%04*DJJ4*r