j<. ddlZddlmZddlZddlZddlZddlmZ ddededed ed ed df d Z dd ed ed ed eefdZ dd ededed dfdZ dd eded eefdZ dZ dZdZdZdZdZdZdZdZdZd dZdS)!N)Optional)datetimejsonutf-8in_pathout_pathtextfourccencodingreturnc jt|d}t|dks|dddvs|dddkrtdt|dkst d |Dstd ||}d}t|}g} |dz|kr~|||dz} t jd ||dzd } |dz|dz| z} } | |krtd | | || | f| | dzz}|dz|k~d}g}| D]>\} }| |kr|s|| |fd}'|| |f?|s|||fdg}|D]\} }|| |t j d t|||t|dzr|dd |}dt j d t|z|z}t|d5}| |ddddS#1swxYwYdS)a Insert (or replace) a custom RIFF chunk in a WAV file to hold an arbitrary string. - in_path: source WAV path - out_path: destination WAV path (can be the same as in_path for in-place write) - text: the string to store (e.g., JSON) - fourcc: 4-byte chunk ID; default b'json' - encoding: encoding for the string payload; default 'utf-8' Notes: * Keeps all original chunks as-is; if a chunk with the same fourcc exists, its payload is replaced; otherwise a new chunk is appended at the end. * Pads the chunk to even length per RIFF rules. * Supports standard little-endian RIFF/WAVE (not RF64 or RIFX). rb NRIFFWAVEFNot a standard little-endian RIFF/WAVE file (RF64/RIFX not supported).c36K|]}d|cxkodkncVdS) ~N).0bs CC:\Users\johnn\OneDrive\Documents\AI\shared\utils\audio_metadata.py z'write_wav_text_chunk..s6"B"Ba2>>>>c>>>>"B"B"B"B"B"Bz7fourcc must be 4 printable ASCII bytes (e.g., b'json'). &   W % % % 88I  D V[s4yy11 1D 8D h    sJ((J,/J,pathcFt|d}t|dks|dddvs|dddkrtdt|dkrtd d}t|}|dz|kr|||dz}t jd ||dzd }|dz|dz|z} }| |krtd ||kr!||| } | |d S| |dzz}|dz|kdS)a" Read and return the string stored in a custom RIFF chunk from a WAV file. Returns None if the chunk isn't present. - path: WAV file path - fourcc: 4-byte chunk ID to look for (default b'json') - encoding: decoding used for the stored bytes (default 'utf-8') rrNrrrrrzfourcc must be 4 bytes.rrr strict)errorsr!)r$r%r&r'r*r+decode) rAr r r0r2r3r5r6r7r8raws rread_wav_text_chunkrGIsI d   " "D 4yy2~~bqb33tAbDzW7L7Labbb 6{{a2333 C D A 'Q,,3s1u9o!$c!e44Q71WcAgns 77KLL L &==uSy/C::hx:88 8TAX 'Q,, 4rWanGPtag_keyc ddlm}m}m}n"#t$r}t d|d}~wwxYw ||}n#|$r |}YnwxYwt |D]5}||} t| |r| j |kr||=6| |d||g| |dS)Nr)ID3ID3NoHeaderErrorTXXXz$mutagen is required for mp3 metadata)r descr ) mutagen.id3rKrLrM Exception RuntimeErrorlistkeysget isinstancerOaddsave) rAr rIrKrLrMexctagkeyframes r_write_mp3_text_tagr]gs'L;;;;;;;;;;; LLLABBKLc$ii ceeCHHJJ  eT " " uzW'<'<CGGDD!' 7 7 7888HHTNNNNNs ,', <A Ac ddlm}m}m}m}n#t $rYdSwxYw ||}n #|$rYdSwxYw|dD]3}t||r!|j|kr|j r|j dcS4|dD]5}t||r#|j|kr|j r |j dndcS6dS)Nr)rKrLrMCOMMrMr_) rPrKrLrMr_rQgetallrVrOr )rArIrKrLrMr_rZr\s r_read_mp3_text_tagraxs8AAAAAAAAAAAAA ttc$ii ttF##%% eT " " %uzW'<'<z %z!}$$$F##99 eT " " 9uzW'<'<$)J85:a==D 8 8 8 4s   -66ctj|d}t j|}|dkrt ||dS|dkrt|||dS|dkr,ddlm }||}||d<| dStd|) Nr!.mp3.wav.flacrFLACrHz#Unsupported audio metadata format: ) osrAsplitextlowerjsondumpsr]r@ mutagen.flacrgrXr')rAconfigsextr1rgaudios rsave_audio_metadatarqs '  4  # ) ) + +Cj!!G f}}D'***** T411111 %%%%%%T  g DsDDEEErcltj|d}|dkrt |}nP|dkrt |}n:|dkr2ddlm}||dg}|r|dnd}ndS|sdStj |S)Nr!rcrdrerrfrH) rhrArirjrarGrmrgrUrkloads)rArorFrgvaluess rread_audio_metadatarus '  4  # ) ) + +C f}} && !$'' %%%%%%d,,!+fQiitt t :c??r) creation_datecreation_datetime created_at created_oncreation_timestampcreated_timestamp)datetimecreated timestamp)generation_time pause_secondsduration_seconds video_lengthc|dSt|tr|St|ttfr{d|cxkrdkr3nn0 tt|ddS#t$rYdSwxYw|dkrdS tjt|S#t$rYdSwxYwt |}|sdStj d|stj d|r? t|}|dkr|dz}tj|S#t$rYnwxYw| d r |dd d z}tj d |r| d dd}dD])} tj ||cS#t$rY&wxYw tj|S#t$rYdSwxYw)Nili r!rz\d{10}(\.\d+)?z\d{13}lJ)g@@Zz+00:00z^\d{4}:\d{2}:\d{2}\s:-)z%Y-%m-%d %H:%M:%Sz%Y-%m-%d %H:%Mz%Y-%m-%dz%Y/%m/%d %H:%M:%Sz%Y/%m/%dz%Y-%m-%d-%Hh%Mm%Ssz%Y%m%dz%Y)rVrintfloatrQ fromtimestampstrstripre fullmatchendswithmatchreplacestrptime fromisoformat)valuer tsfmts r_parse_datetime_valuersa }t%"" %#u&&  5 D  E Aq111   tt  A::4 )%,,77 7   44  u::    D t |%t,, Y0M0M tB%%%f )"-- -    D  }}S$CRCy8# x'..)||Ca((    $T3// / / /    D %d+++ ttsZ A'' A54A5 B"" B0/B0-D11 D>=D>F++ F87F8<G GGc#K|dSt|dr3|j}t|ttfr |D]}|Vn|VdSt|ttt fr|D]}t |Ed{VdS|VdS)Nr )hasattrr rVrStupleset_iter_tag_values)rtxtitems rrrs }ufj cD%= ) )     III%$s+,, . .D'-- - - - - - - - - KKKKKrct|tsdStD]*}t||}||cS+|d}t|tr2tD]*}t||}||cS+|t|tr|nifD]}|D]\}}t |tfdtDrYtfdtDszt|}||ccSdS)N extra_infoc3 K|]}|vV dSNrrpartlkeys rrz:extract_creation_datetime_from_metadata..s'>>D44<>>>>>>rc3 K|]}|vV dSrrrs rrz:extract_creation_datetime_from_metadata..'@@tt|@@@@@@r) rVdict_CREATION_KEYSrrUitemsrrrjany_DATE_KEY_EXCLUDE_DATE_KEY_PARTS)metadatar[dtrsourcerrs @r'extract_creation_datetime_from_metadatars h % %t "8<<#4#4 5 5 >III l++J*d##!  C&z~~c':':;;B~ :j$+G+GOZZRP   ,,..  JCs88>>##))++D>>>>,=>>>>> @@@@@@@@@ &u--B~   4rcV  ddlm}n#t$rYdSwxYw ||d}n#t$rYdSwxYw|dnt|dd}|dSt |drdD][} ||}n#t$rg}YnwxYw|D]-}t |D]}t|}||cccS.\ |d} n#t$rg} YnwxYw| D]|}tt|d d  t fd tDsSt |D]}t|}||ccS}t |d r| ng} | D]\} } t|  tfd tDrGtfdtdzDskt | D]}t|}||ccSdS)Nr)FileF)easytagsr`)TDRCTDENTORYTYERTDATTIMErMrOc3 K|]}|vV dSrr)rrrOs rrz1_extract_native_audio_datetime..:rrrc3 K|]}|vV dSrrrs rrz1_extract_native_audio_datetime..Ds'::tt|::::::rc3 K|]}|vV dSrrrs rrz1_extract_native_audio_datetime..Fs'ZZD44<ZZZZZZr)icrdu©dayyear)mutagenrrQgetattrrr`rrrrjrrrr)rArrpr frame_nameframesr\rr txxx_framesrr[rrOrs @@r_extract_native_audio_datetimers       ttT&&& tt=44geVT&B&BD |ttXJ " "J Z00     " ",U33""D.t44B~! &" "  ++f--KK   KKK    Ewufb112288::D@@@@@@@@@ (//  *400>IIIII"  $D'22 :DJJLLLE   U3xx~~ ::::(9::: : :  ZZZZO>Y,YZZZZZ  $U++  D&t,,B~   4s<  + 99+B BBC C*)C*cdtjtj|Sr)rrrhrAgetmtime)rAs r_get_file_creation_datetimerOs#  !"'"2"24"8"8 9 99rc|}|# t|}n#t$rd}YnwxYwt|}||St|}||St |Sr)rurQrrr)rAwangp_metadatarrs rresolve_audio_creation_datetimerTsH *400HH   HHH  1 : :B ~ ' - -B ~ &t , ,,s  %%)rr)rHr)r*typingrrkrhrrrbytesr@rGr]rarqrurrrrrrrrrrrrrs  CJ==#==C=!&=