o GY)j#@sdZddlmZddlmZmZddlmZmZmZd4ddZ d5d d Z d6ddZ d7ddZ d8ddZ d9ddZd6ddZ d:ddd;d!d"Zd#d$dd.d/Z d:ddd?d2d3ZdS)@zOShared minimum call duration checks for ingest, transcribe, and backlog repair.) annotations)datetime timedelta)AnyDictOptionalreturnOptional[datetime]cCsr|durdSt|tr|jr|jddS|Sztt|dd}|jr,|jddWS|WSty8YdSw)N)tzinfoZz+00:00) isinstancerr replace fromisoformatstr Exception)valueparsedr#sz(parse_duration_value..i<) r rmaxint total_secondsrrstripsplitlen TypeError ValueError)rtextpartsnumsrrrparse_duration_values4    ,  r,callDict[str, Any]cCs|dp|dp|d}|dp|dp|d}|rD|rDt|}t|}|rD|rDz tdt||WStyCYnwdD]}||}|d ura|d krat|}|d ura|SqFd D]}t||}|d ury|dkry|Sqdd S) zBest-effort duration from a source row or raw_calls row. Wall-clock (call_end - call_start) is preferred over billing fields (answeredtime/billsec) so short calls are not inflated at ingest. call_starttime starttime call_start call_endtimeendtimecall_endrduration_secondscall_duration_s duration_sdurationNr) answeredtime answered_timetalktime talk_timebillsec)getrr!r"r#rr,)r-startendstart_dtend_dtkeyrawrrrrcall_duration_seconds0s0  rFmin_duration_sr"boolcCst|pddkS)Nr)r")rGrrrmin_duration_configuredPsrIcCs"t|dp|dp|dS)Nr/r0r1)rr?)r-rrrcall_start_datetimeTsrJcCs<|durdSt|}|durdSt|}|durdS||kS)z True when the min-duration policy applies to this call. Calls that started before ``effective_at`` are grandfathered. NFT)rrJ)r- effective_ateffr@rrrmin_duration_applies_to_callZsrMcCsg}|dp|dp|d}|dp|dp|d}|rJ|rJt|}t|}|rJ|rJz|tdt||Wn tyIYnwdD]}t||}|d ur^||qLd D]}t||}|d urw|dkrw||qa|s|d St|S) z>Shortest reliable metadata estimate (talk time vs wall clock).r/r0r1r2r3r4rr5N)r:r;r<r=r>pulse) r?rappendr!r"r#rr,min)r- candidatesr@rArBrCrDrrrrmin_duration_gate_secondsks2    rRNaudio_duration_srTOptional[float]cCsZt|pd}|dkr dSt||sdS|durt|t|kSt|}|dur)dS||kS)z True when a call should not be ingested or processed. When ``audio_duration_s`` is provided (WAV probe), it is the authority. Otherwise uses the shortest metadata estimate from ``min_duration_gate_seconds``. rFNT)r"rMrrR)r-rGrKrTmin_sr9rrrshould_skip_at_ingests  rWT) probe_audio recording_url Optional[str]rX!tuple[bool, str, Optional[float]]c Cst|pd}|dkst||sdSddlm}m}t|p"|dp"d}|rN||s.dS||} | dur8dS| |krId d | d d |d | fSdd| fSt|||r\d t ||dfSdS)z Decide whether a call may be ingested/queued. Returns ``(should_skip, reason, probed_audio_seconds)``. When min duration applies and probing is enabled, the recording URL must be probed; unreachable audio blocks ingest (no DB row). r)FrN)probe_wav_duration_from_urlrecording_url_probe_readyfileurlr)Tz*recording URL not ready for duration probeNN)Tz-recording not reachable or not a readable WAVNT actual audio .1fs < min sF) r"rMwav_probe_utilr\r]rr?r$rW skip_reason) r-rGrKrYrXrVr\r]urlprobedrrr evaluate_min_duration_for_ingests"   rgcCs t|||S)z'Alias for ingest/STT min-duration gate.)rW)r-rGrKrrris_below_min_durations rhrcCsZ|durdt|ddt|dSt|}|dur"dt|dSd|dt|dS)Nr_r`rarbzduration unknown (min z s required)z duration )rr"rR)r-rGrTr9rrrrds rdbidcallidc Cst||||ds dSt|p|dpd}|sdS|d}|d}|d} |d|d |ft|} |d | d |ft|} | sP| r]|d |d |fdS|d|d|fdS)aD Remove or terminal-mark calls below minimum duration. Only affects calls on or after ``effective_at``. Unprocessed rows are deleted. Rows that already have transcript or analytics are marked ``skipped_short`` so they stay out of the dashboard and STT queue. Returns True when an action was taken. rSFrjr _raw_calls_sarvamresponse_callanalyticsz SELECT 1 FROM `z~` WHERE callid = %s AND transcript IS NOT NULL AND TRIM(transcript) != '' LIMIT 1 zSELECT 1 FROM `z` WHERE callid = %s LIMIT 1z UPDATE `z` SET transcription_status = 'skipped_short', status = -2 WHERE callid = %s Tz DELETE FROM `z` WHERE callid = %s)rWrr?r$executerHfetchone) cursorrirjr-rGrKrT raw_table sarvam_tableanalytics_tablehas_transcript has_analyticsrrrpurge_unprocessed_if_below_minsF       rv)rr )rrrr)r-r.rr)rGr"rrH)r-r.rr )r-r.rrH)N)r-r.rGr"rTrUrrH) r-r.rGr"rYrZrXrHrr[)r-r.rGr"rrH)r-r.rGr"rTrUrr) rirrjrr-r.rGr"rTrUrrH)__doc__ __future__rrrtypingrrrrr,rFrIrJrMrRrWrgrhrdrvrrrrs2      ! (