GdZddlZddlZddlZddlmZmZddlmZddlm Z ddlm Z eje je jZded efd Zded efd Z dd ed edeeefdeeefdeeded efdZy)z?An adapter that takes a bq command and executes it with gcloud.N)DictOptional)bq_to_gcloud_config_classes) gcloud_runner)supported_gcloud_commands)global_flag_mappingscommand_mappings gcloud_outputreturncf|jddjddjddjddjddjddjddjd djd djd d jd dS)Nu┌+u┐u└u┘u├u┤u┬u┼u┴u│|─-)replace)r s 2platform/bq/gcloud_wrapper/bq_to_gcloud_adapter.py_swap_gcloud_box_to_bq_prettyrsE3'wucwucwucwucwucwucwucwucwucwuc ctjdd|}tjdd|}tjdd|}tjdd|}tjdd|}tjd d |S) z1Converts gcloud table output to bq sparse output.u┌.*┐u└.*┘u │(.*)│z \1 u[├┤] u[│┼]rr)resub)r stripped_upper_borderstripped_lower_bordermostly_stripped_side_bordersstripped_side_bordersno_vertical_barss r_swap_gcloud_box_to_bq_sparser%s~&&b-@&&b2GH!#W3"&&d4PQVVK/DE - ..rresource bq_commandbq_global_flagsbq_command_flags identifierdry_runctj|||||}|r2tdjdgt j |zyt j|tj}|jdd}tj||} |js |jSd } t|jjd D]} d } t!| j#} | j%| }|r| } n]| j&s#| j)| ||jd } n.d |vr| | z } n$|d k(r t+| } n|dk(r t-| } n| } | st| | r t/j0| }t3|t4r+g}|D]#}|j7| j9||%n| j9||}d |k(r,tt/j:|d |jSd|k(r!tt/j:|dd|jS|jS#t.j<$rt| Y|jSwxYw)azTakes a bq command and executes it with gcloud returning the exit code. Args: resource: The resource the command is being run on, named to align with `gcloud` commands. For example, 'jobs' or 'datasets'. bq_command: The bq command to run. For example, 'ls' or 'show'. bq_global_flags: The BQ CLI global flags to use when running the command. bq_command_flags: The BQ CLI command flags to use when running the command. identifier: The identifier of the resource to act on. dry_run: If true, the gcloud command will be printed instead of executed. Returns: The exit code of the gcloud command. )r r!r"r#r$rgcloudr)stderrformatsparse)r r!r project_idjsonpretty),:) separators prettyjsonT)indent sort_keys)GCLOUD_COMMAND_GENERATORget_gcloud_commandprintjoinrquote_flag_valuesrrun_gcloud_command subprocessSTDOUTgetget_command_mappingstdout returncodeiterreadlinestrstrip$synchronous_progress_message_matcherprint_resourcestatus_mappingrrr,loads isinstancelistappend json_mappingdumpsJSONDecodeError)r r!r"r#r$r%gcloud_commandproc bq_formatcommand_mapping json_outputraw_line line_to_printoutputis_progress_message parsed_json json_object item_dicts rrun_bq_command_using_gcloudr[1so,,>>%' ?.    J);;NK L   ) )    $ !!(H5),@@JA/  ??+t{{++R0hM ] "F)NNm  + +%44 *o11,?m 9 Vk h 3F;m h 3F;mm M/10JJ{+k K & $I   _99)YO P%&22; J 9  djj<=  9 $ djjQ$?@     K  s;BI &I I87I8)NF)__doc__r,rr;typingrrgcloud_wrapperrrrGcloudCommandGeneratorSUPPORTED_GLOBAL_FLAGSSUPPORTED_COMMANDSr5rCrrboolintr[rrresE !6(4M6MM2II.AA $ / / /"!% ]]]#s(^]38n ]  ]  ] ]r