#dZddlmZddlmZddlmZddlmZddlZddlmZddl m Z dd l m Z d Z Gd d e jZy) z4Custom table printer for CAI team's asset query API.)absolute_import)division)print_function)unicode_literalsN) extra_types)custom_printer_base)resource_printer cloudassetc8eZdZdZedZdZdZdZdZ y)AssetQueryPrinterzBPrints the asset query response in a custom human readable format.ctjttd|jj ty)z3Register this custom printer with the given parser.T)hiddenN)r RegisterFormatterASSET_QUERY_PRINTER_FORMATr display_info AddFormat)parsers ;lib/googlecloudsdk/command_lib/asset/asset_query_printer.pyRegisterzAssetQueryPrinter.Register#s0&&"$5dD !!"<=ctjj|}d|vr"d|dvr|dd=d|dvr|dd=|ds|d=tj}t j |d||jS)N queryResultrowsschemayamlout)rencodingMessageToPyValueioStringIOr Printgetvalue)selfresp resp_message string_bufs r_FormatMetadataz!AssetQueryPrinter._FormatMetadata*s''88>L $ < . .  ' / \-0 0  ' 1 - (  'J<Z@     rct|drt|jdsy|jj}|jj}g}|jsyt j }|jdjdjd|jD|D]b}tjj|}tjj|}|j|j||ddt j } tj ||j#| t%j&| j#gS) uFormats rows in a [QueryAssetsResponse]'s queryResults into a table. Args: resp: The [QueryAssetsResponse] that contains 0 or more rows. Returns: A 'Lines' custom printer object that corresponds to the formatted table when printed out. The response.queryResult.rows in response: { "jobReference": "CiBqb2JfdDR2SFgwa3BPNFpQVDFudVJJaW5TdVNfS1N0YxIBAxjH8ZmAo6qckik", "done": true, "queryResult": { "rows": [ { "f": [ { "v": "//cloudresourcemanager.googleapis.com/folders/417243649856" } ] } ], "schema": { "fields": [ { "field": "name", "type": "STRING", "mode": "NULLABLE" } ] }, "total_rows": 1 } } Will return a custom printer Lines object holding the following string: ┌────────────────────────────────────────────────────────────┐ │ name │ ├────────────────────────────────────────────────────────────┤ │ //cloudresourcemanager.googleapis.com/folders/417243649856 │ └────────────────────────────────────────────────────────────┘ rrNztable[box]({})z, c3hK|]*}dj|j|j,yw)z {}:label={}N)formatfield).0r,s r z4AssetQueryPrinter._FormatRowTable..xs-6$"E U[[%++6"s02Fr)hasattrrrrfieldsr r!writer+joinrrrappend_ConvertFromFVr r"r#cpLines) r$r%rrrow_list schemabufrowrow_json schema_jsonraw_outs r_FormatRowTablez!AssetQueryPrinter._FormatRowTable:s3b 4 'wt7G7G7?0A     $ $F    DH ==  I OO$++DII6$]]6$-$%&%%66s;h((99&Akood))+xGH  kkmG8Y%7%7%9wG 88W%%'( ))rc |sy|jdgDcgc]}|jdd}}i}|}|s|d}t||D]\}} d|vr td|djdk(r|jdg} |jdd jd k(r8| D cgc]%} |j | | jddd 'c} ||d <|j | | d } | r | ||d <d||d <|jdd jd k(r-| D cgc]} | jdd} } | | nd||d <| r| nd||d <|Scc}wcc} wcc} w) aHConverts from FV format to values. Args: schema: The schema struct within the queryResult struct in the response. row: A single row of the response's queryResult.rows message. is_record: True if the row object is a record within an actual row. Returns: A dictionary mapping row keys to the values that may be a simple datatype, a record (struct) in the form of a dictionary, or a list of either simple data types or records (again, in the form of dictionaries). Raises: IOError: An error occurred accessing the smalltable. fvr0typez'Invalid response: missing type propertyRECORDmodeNULLABLEREPEATEDTr,)getzipIOErrorupperr4) r$rr9 is_recordentryvaluesresult new_schemar,rA subfieldssubvaluecur_vals rr4z AssetQueryPrinter._ConvertFromFVs  .1ggc2.> ?.>UeiiR .>F ? FJ (#j F+q u ?@@ v   ( *IIh+ 99VZ ( . . 0J > $ (!!)X\\#r-BDI$ &w '' 1d;' %,F5> "%'F5> " 99VZ ( . . 0J >9:;X8<<R(;,3,?RuW~'(RuW~3,4 MA@$ > ! L*\3j:rr )r_ __future__rrrrr apitools.base.pyrgooglecloudsdk.core.resourcerr5r rCustomPrinterBaser rarrrfs>;&%' (B9*k:,,k:r