o 2¡µiúã@sÌddlmZddlZddlZddlmZmZmZmZddl m Z e e¡Z dZdZdZdZd=d>dd„Zd?dd„Zd@dd „ZdAdBd%d&„ZdCdDd)d*„ZdEd,d-„ZdFdGd2d3„ZdHdId5d6„ZdJd8d9„ZdKd;d<„ZdS)Lé)ÚannotationsN)ÚAnyÚDictÚListÚOptional)Úget_connectionÚpendingÚ processingÚdoneÚfailedÚbidÚstrÚcall_idÚ recording_urlÚmetadataúOptional[Dict]Úreturnú Optional[int]cCsÖtƒ^}| ¡I}| d|||tt |pi¡f¡|jdkr/|jWdƒWdƒS| d||f¡| ¡}|rA|dndWdƒWdƒS1sTwYWdƒdS1sdwYdS)NzINSERT IGNORE INTO stt_jobs (bid, call_id, recording_url, status, metadata, created_at, updated_at) VALUES (%s, %s, %s, %s, %s, NOW(), NOW())rz3SELECT id FROM stt_jobs WHERE bid=%s AND call_id=%sÚid) rÚcursorÚexecuteÚSTATUS_PENDINGÚjsonÚdumpsÚrowcountÚ lastrowidÚfetchone)rrrrÚconnÚcurÚrow©r úB/home/aiteam/pcaa-dev/call-proccessing/stt_pipeline/db/stt_jobs.pyÚ insert_jobs" ü øÿóÿ"ÿr"Újob_idÚintÚNonecCst|tƒdS©N)Ú_update_statusÚSTATUS_PROCESSING)r#r r r!Úmark_processingsr)Ú transcriptÚspeaker_segmentsú List[Dict]Ú speaker_countÚdurationÚfloatÚstt_providercCs„tƒ5}| ¡}| dt|t |¡||||f¡Wdƒn1s#wYWdƒdSWdƒdS1s;wYdS)Nz«UPDATE stt_jobs SET status=%s, transcript=%s, speaker_segments=%s, speaker_count=%s, duration_seconds=%s, stt_provider=%s, error_message=NULL, updated_at=NOW() WHERE id=%s)rrrÚSTATUS_DONErr)r#r*r+r-r.r0rrr r r!Ú mark_done"s ÿüÿÿ"ÿr2TÚerrorÚincrement_retryÚboolc Cs’tƒ<}| ¡}|r dnd}| d|›dt|dd…|f¡Wdƒn1s*wYWdƒdSWdƒdS1sBwYdS)Nz, retry_count = retry_count + 1Úz/UPDATE stt_jobs SET status=%s, error_message=%sz, updated_at=NOW() WHERE id=%siÐ)rrrÚ STATUS_FAILED)r#r3r4rrÚ retry_sqlr r r!Úmark_failed.s þþÿ"ÿr9éÚstale_after_minutesc Csztƒ0}| ¡}| dtt|f¡|jWdƒWdƒS1s&wYWdƒdS1s6wYdS)NztUPDATE stt_jobs SET status=%s, updated_at=NOW() WHERE status=%s AND updated_at < DATE_SUB(NOW(), INTERVAL %s MINUTE))rrrrr(r)r;rrr r r!Úreset_stale_processing_jobs7s ýúÿ"ÿr<ú List[str]c Csrtƒ'}| ¡}| d|f¡| ¡}Wdƒn1swYWdƒn1s-wYdd„|DƒS)Nz)SELECT call_id FROM stt_jobs WHERE bid=%scSsg|]}|d‘qS)rr ©Ú.0Úrr r r!Ú Fsz)get_all_seen_call_ids..©rrrÚfetchall)rrrÚrowsr r r!Úget_all_seen_call_idsAs þ€ÿrEé ú Optional[str]ÚlimitúList[Dict[str, Any]]c Cs’tƒ<}| ¡'}|r| dt||f¡n| dt|f¡| ¡WdƒWdƒS1s2wYWdƒdS1sBwYdS)NzRSELECT * FROM stt_jobs WHERE status=%s AND bid=%s ORDER BY created_at ASC LIMIT %szGSELECT * FROM stt_jobs WHERE status=%s ORDER BY created_at ASC LIMIT %s)rrrrrC)rrHrrr r r!Úget_pending_jobsHs þþõÿ"ÿrJÚmax_retriesc Cs|tƒ1}| ¡}| dt||f¡| ¡WdƒWdƒS1s'wYWdƒdS1s7wYdS)NzZSELECT * FROM stt_jobs WHERE status=%s AND retry_count<%s ORDER BY updated_at ASC LIMIT %s)rrrr7rC)rKrHrrr r r!Úget_retryable_failed_jobsWs ýúÿ"ÿrLúDict[str, int]c Csntƒ%}| ¡}| d¡| ¡}Wdƒn1swYWdƒn1s+wYdd„|DƒS)Nz