A&dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd l mZdd l mZdd l mZd ZdZdZdZdZGdde j,ZGddeZGddeZGddeZdZdZdZdZdZy)z4Utilities for gcloud ml video-intelligence commands.)absolute_import)division)unicode_literalsN) storage_util)apis) exceptions)log)files) iso_duration)timesvideointelligencev1zz[{}] is not a valid format for video input. Must be a local path or a Google Cloud Storage URI (format: gs://bucket/file).zl[{}] is not a valid format for result output. Must be a Google Cloud Storage URI (format: gs://bucket/file).zCould not get video segments from [{0}]. Please make sure you give the desired segments in the form: START1:END1,START2:END2, etc.: [{1}]ceZdZdZy)Errorz!Base error class for this module.N__name__ __module__ __qualname____doc__/lib/googlecloudsdk/command_lib/ml/video/util.pyrr2s)rrceZdZdZy) SegmentErrorz2Error for poorly formatted video segment messages.Nrrrrrr6s:rrceZdZdZy)VideoUriFormatErrorz(Error if the video input URI is invalid.Nrrrrrr:s0rrceZdZdZy)AudioTrackErrorz-Error if the audio tracks setting is invalid.Nrrrrrr>s5rrc |sytjtt}|j}g}|Dcgc]}|j d}}|D]}t |dk7r.ttjdj|d|d|d}} t|} t|} d} |j|| j| j| j| j |Scc}w#t$r3} ttjdj|| d} ~ wwxYw) ajGet VideoSegment messages from string of form START1:END1,START2:END2.... Args: given_segments: [str], the list of strings representing the segments. Raises: SegmentError: if the string is malformed. Returns: [GoogleCloudVideointelligenceXXXVideoSegment], the messages representing the segments or None if no segments are specified. N:,zMissing start/end segmentrz{}s) endTimeOffsetstartTimeOffset)rGetMessagesModule VIDEO_APIVIDEO_API_VERSION*GoogleCloudVideointelligenceV1VideoSegmentsplitlenrSEGMENT_ERROR_MESSAGEformatjoin_ParseSegmentTimestamp ValueErrorappend total_seconds) given_segmentsmessages segment_msgsegment_messagesssegmentssegmentstartendstart_duration end_durationvesec_fmts rValidateAndParseSegmentsr@BsA    # #I/@ A(CC+$2 3NqaggclN( 3g 7|q .55 ((> "$?A BBWQZ3E )-e4n+C0l GKnn\%?%?@~'C'CDFG%* -4 ) .55 ((> "B( )))sDD E.D==Ec t|}tjdj||dkrt dj|tj|S#t$r t j |}|jdkrt j|cYS#t jt jf$rt dj|wxYwwxYw)aParse duration formatted segment timestamp into a Duration object. Assumes string with no duration unit specified (e.g. 's' or 'm' etc.) is an int representing microseconds. Args: timestamp_string: str, string to convert Raises: ValueError: timestamp_string is not a properly formatted duration, not a int or int value is <0 Returns: Duration object represented by timestamp_string zTTime unit missing ('s', 'm','h') for segment timestamp [{}], parsed as microseconds.rzCould not parse timestamp string [{}]. Timestamp must be a properly formatted duration string with time amount and units (e.g. 1m3.456s, 2m, 14.4353s)zICould not parse duration string [{}]. Timestamp must begreater than >= 0)) microseconds) intr warningr-r0r ParseDurationr2DurationValueErrorDurationSyntaxErrorr Duration)timestamp_stringrBdurations rr/r/ns$D'(LKK**0&1A*BDA **0&1A*B DD   L 99'  . .$$%56h   ! #&&(( o  % %u'?'? @. HHN+I- ... .s# A&& C.09B,)C.,>C**C.c|r=tjj|sttj ||S)zValidates given output URI against validator function. Args: output_uri: str, the output URI for the analysis. Raises: VideoUriFormatError: if the URI is not valid. Returns: str, The same output_uri. )rObjectReference IsStorageUrlrOUTPUT_ERROR_MESSAGEr-) output_uris rValidateOutputUrirPs7 44AA*M 299*E FF rc|j}tjj|rt j ||_|Stjj|r ||_ |Sttj|)zEThe Python hook for yaml commands to inject content into the request.) input_pathospathisfiler ReadBinaryFileContents inputContentrrLrMinputUrirINPUT_ERROR_MESSAGEr-) unused_refargsrequestrTs rUpdateRequestWithInputr]ss $WW^^D 77=G . ##006G . 188> ??rc8t|dkDr td|S)z1Verify at most two tracks, convert to [int, int].r!z+Can not specify more than two audio tracks.)r+r)trackss rAudioTrackProcessorr`s[1_ G HH -r)r __future__rrrrSgooglecloudsdk.api_lib.storagergooglecloudsdk.api_lib.utilrgooglecloudsdk.corerr googlecloudsdk.core.utilr r r r'r(rYrNr,rrrrr@r/rPr]r`rrrrfs ;&' 7,*#*1*  56- *J  *;5;1%16e6)X':T" r