iddlZddlZddlZddlZddlZddlmZddlmZeZ eja GddZ dejdede d dfd Zd ed efd Zd Zdeded efdZy)N)Console)internet_managerc"eZdZdZdZdZdZy) ProgressDataz%Class to store the last progress datacDd|_tj|_yN) last_data threadingLocklockselfs _/home/nidran/Documenti/script/streaming-community/StreamingCommunity/core/processors/capture.py__init__zProgressData.__init__sNN$ cT|j5||_dddy#1swYyxYwrr r )rdatas rupdatezProgressData.updates% YY "!DN " " "s'c^|j5|jcdddS#1swYyxYwrrr s rgetzProgressData.get!s% YY ">> " " "s #,N)__name__ __module__ __qualname____doc__rrrrrrrs/%""rrprocess description progress_datareturnc d}tj}t|jjdD]} |j }|st j rnd|vr tj|z }t|}d|vrZt|jdddk(}|rdnd} ttjd|j| d dd z} n5ttjd|jdd dd z} |jd d } |jd d } |jdd } |jdd }tj|d}| || | d}|j||d| d|dtj | d| d| d|d}t#|t%|}t&j)|j+|d t3|y#t,$r%}t/j0d|d|Yd}~9d}~wwxYw#t,$r#}t/j0d|Yd}~.d}~wwxYw#t,$r"}t/j0d |Yd}~d}~wwxYw#t,$r"}t/j0d!|Yd}~yd}~wwxYw# t3|w#t,$r"}t/j0d!|Yd}~wd}~wwxYwxYw)"a9 Function to capture and print output from a subprocess. Parameters: - process (subprocess.Popen): The subprocess whose output is captured. - description (str): Description of the command being executed. - progress_data (ProgressData): Object to store the last progress data. rzsize=qgLsizesizez\d+0ifpszN/AtimebitratespeedT) add_hours)r'r*r(r)z [white]: ([green]'fps': [yellow]z![white], [green]'speed': [yellow]z [white], [green]'size': [yellow]z [white], [green]'time': [yellow]z#[white], [green]'bitrate': [yellow]z#[white], [green]'elapsed': [yellow]z[white]) )endzError parsing output line:  - Nz'Error processing line from subprocess: zError in capture_output: zError terminating process: )r(iterstdoutreadlinestripterminate_flagis_setparse_output_linefloatrintrefindallr format_timerformat_file_sizemaxlenconsoleprintljust Exceptionloggingerrorterminate_process)rrr max_length start_timeline elapsed_timeris_endsize_key byte_sizer'time_processedr)r*elapsed_formatted json_dataprogress_stringes rcapture_outputrQ&sD= YY[ 00"57 MD6 Mzz|"((*d?(R'+yy{Z'? 06$;&+DHHS$,?&@D&HF28wfH(+BJJvtxxRU?V,WXY,Z([^b(bI(+BJJvtxxPS?T,UVW,X(Y\`(`I#hhue4)-&%)@"&((9e"< $% 8-=,H,Hae,f)-0bi$j %,,Y7 +m,669U;7766F6W6WXa6b5cd66D5EF99@ B99J8K8 U(&)S5I%J  o&;&;J> Rc7 Mz = g &%R (CD6QC&PQQR M GsKLL M 7 1!5667  = MM7s; < < = = g & = MM7s; < < =s:I.H?I.H?$I.'H?-FH>I. J H<H72H?7H<<H?? I+I& I.&I++I.. J7JK JK  K%KK L KL L!K>9L>LLrGc i}|jddjddj}|D]^}|jd}t|dk(s#|d}|d}|dk(r(t|trd |vr|jd d}|||<`|S#t $r'}t jd |d |icYd }~Sd }~wwxYw) z Function to parse the output line and extract relevant information. Parameters: - line (str): The output line to parse. Returns: dict: A dictionary containing parsed information. z r"z= =rr(.zError parsing line: r.N)replacesplitr= isinstancestrrArBrC)rGrpartspart key_valuekeyvaluerPs rr5r5vs T2&..tS9??A "D 3I9~"l!! &=Zs%;u !KK,Q/E!S  "  ,TF#aS9: s$AB?B C "C>C C c |j|jyy#t$r"}tjd|Yd}~yd}~wwxYw)z Function to terminate a subprocess if it's still running. Parameters: - process (subprocess.Popen): The subprocess to terminate. NzFailed to terminate process: )pollkillrArBrC)rrPs rrDrDsH; <<> ! LLN " ; 5aS9::;s $ AA  Affmpeg_commandctjt} tj|tj tj d}tjt|||f}|j |jtj!|j# |j%S#t$rtjdYUt$r"}tjd|Yd}~zd}~wwxYw#tj!|j#wxYw#t$r1}tjd|Yd}~|j%Sd}~wwxYw)a/ Function to capture real-time output from ffmpeg process. Parameters: - ffmpeg_command (list): The command to execute ffmpeg. - description (str): Description of the command being executed. Returns: dict: JSON dictionary with the last progress data containing T)r0stderruniversal_newlines)targetargszTerminating ffmpeg process...zError in ffmpeg process: Nz Failed to start ffmpeg process: )r3clearr subprocessPopenPIPESTDOUTr ThreadrQstartwaitKeyboardInterruptrBrCrAsetjoinr)rcrrr output_threadrPs rcapture_ffmpeg_real_timerus3!NM>"">*//R\RcRcx|}"((g{\iEjk  ! LLN          ! ; MM9 : ; MM5aS9 : : ;       > 8<==     >sZA#D?C $D? D'D)D1D DDD&D<<D?? E9E44E9)r8r(rBr rj rich.consolerStreamingCommunity.utils.osrr>Eventr3rrkrZrQdictr5rDlistrurrrr{s !9 ) " " "M=J,,M=3M=|M=`dM=`CDB ;.T...r