dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z d d d Z ejejj Gd d ej"Ze e_y)z'trace sinks update' command.)absolute_import)division)unicode_literals)util)base) exceptions)logz/ Changes the *[destination]* associated with a sink. The new destination must already exist and Stackdriver Trace must have permission to write to it. Trace spans are exported to the new destination in a few minutes. zg/ $ {command} my-sink bigquery.googleapis.com/projects/my-project/datasets/my_new_dataset ) DESCRIPTIONEXAMPLESc,eZdZdZedZdZdZy)UpdatezUpdates a sink.c|jdd|jdd|jdd|jjd|jjd y ) z Register flags for this command. sink_namezThe name of the sink to update.)help destinationzThe new destination for the sink. The destination must be a fully qualified BigQuery resource name. The destination can be for the same Google Cloud project or for a different Google Cloud project in the same organization.z --projectzSUpdate a sink associated with this project. This will override the default project.yamlN) add_argument display_info AddFormatAddCacheUpdater)parsers !lib/surface/trace/sinks/update.pyArgsz Update.Args/sz  *KL   34 !!&) ''-c tj}tjjj |j ||j di|dj|S)z*Patches a sink specified by the arguments.,)name traceSink updateMask)r GetMessages GetClientprojects_traceSinksPatch(CloudtraceProjectsTraceSinksPatchRequest TraceSinkjoin)selfr sink_data update_maskmessagess r PatchSinkzUpdate.PatchSink@se!H >>  / / 5 599(h((595xx , : . //rctj|j|j}|j }d|i}g}|j dr"d|j i|d<|jd|stjd|j|||}tj|tj|S)zThis is what gets called when the user runs this command. Args: args: an argparse namespace. All the arguments that were provided to this command invocation. Returns: The updated sink with its new destination. rr outputConfigzoutput_config.destinationz(Please specify the destination to update)rGetTraceSinkResourcerproject RelativeName IsSpecifiedrappendcalliope_exceptionsMinimumArgumentExceptionr,r UpdatedResourceFormatTraceSink)r(argssink_refsink_resource_namer)r*results rRunz Update.RunIs((FH!..0+,IK  �$2B2B"Ci45   8 8 4 66^^. ; GF!    ''rN)__name__ __module__ __qualname____doc__ staticmethodrr,r<r rrr r +s!.. /(rr N)r@ __future__rrrgooglecloudsdk.api_lib.tracergooglecloudsdk.callioperrr4googlecloudsdk.corer DETAILED_HELP ReleaseTracks ReleaseTrackALPHA UpdateCommandr detailed_helpr rrrLsu$&'-(E#   "D%%++,6(T  6(-6(r%r