tPj~(`dZddlZddlmZmZddlmZmZddlZddlZ ddl m Z GddZ y)z+??`aj`kkstu&3&9&9&; F "J  " ":z9 E Fc &i}d|jvr|jjdiSd}|jdD]V} |j d}|s|j |}|j d}||vrg||<||j|X|jD cic]\}} | s |tj|  c} }S#t$r,}|jjd|d|d Yd}~d}~wwxYwcc} }w) z;Group data by date, converting to UTC and validating dates.time/usage_startzE[red]Error: Missing 'time/usage_start' column for date grouping[/red]NTnamed%Y-%m-%dz.[yellow]Warning: Could not process timestamp '': [/yellow]) columnsrr iter_rowsget_parse_and_convert_timestampstrftimeappend Exceptionrpl DataFrame) rrr! timestamp_strrowdtr"edate_keyrecordss rrz CloudZeroStreamer._group_by_dateEs!@B  T\\ 1 LL  f gI$( >>>- C  #(: ; $66}E[[4 ]202M*-j)005 *JWI\I\I^j4EHgbi",,w//jj   ""%STaSbbefgehhq#rs  ks*C?C. D 9D  D "DD r6c |jdr&tj|jdd}nhd|vs|jdrtj|}n=tj|}|j|j|j }|j tjS#t$r}td|d|d}~wwxYw) z*Parse timestamp string and convert to UTC.Z+00:00+)z-00:00z-01:00z-02:00z-03:00z-04:00z-05:00z-06:00z-07:00z-08:00z-09:00z-10:00z-11:00z-12:00z+01:00z+02:00z+03:00z+04:00z+05:00z+06:00z+07:00z+08:00z+09:00z+10:00z+11:00z+12:00N)tzinfozCould not parse timestamp 'r+) endswithr fromisoformatreplacer@r astimezonerr ValueError)rr6r8r9s rr0z.CloudZeroStreamer._parse_and_convert_timestampfs, R%%c*++M,A,A#x,PQ %)?)?*:++M:++M:99$4+=+=>B==. . R:=/QCPQ Q RsB=C C CCr"r#c 8|jryd|jdd}|jd|jd}|j |||} t j d5}|jjd |d t|d |j||| }|j|jjd |d t|ddddy#1swYyxYw#t j$r(} |jjd|d| dd} ~ wt j$rS} |jjd|d| jjd| jj dd} ~ wwxYw)z+Send a single daily batch to CloudZero API.NzBearer zapplication/json) Authorizationz Content-Typez /v2/connections/billing/anycost/z/billing_dropsg>@)timeoutz[blue]Sending batch for z (z records)[/blue])headersjsonu'[green]✓ Successfully sent batch for z records)[/green]u)[red]✗ Network error sending batch for z: z[/red]u&[red]✗ HTTP error sending batch for  )rr rr _prepare_batch_payloadhttpxClientrrrpostraise_for_status RequestErrorHTTPStatusErrorresponse status_codetext) rr"r#rrIurlpayloadclientrSr9s rr z#CloudZeroStreamer._send_daily_batchs     't||n5.   &FtGYGYFZZhi44ZYW d+ v ""%=j\CPZOK\\l#mn"(++c7+"Q))+ ""=j\CPZOK\\mn    !!  LL  !J:,VXYZX[[ab c $$  LL  8 BqzzG]G]F^^_`a`j`j`o`o_ppvw    sDC:*A;C.%C:.C73C:7C::F #D00FAFFcB tj|d}|jd}g}|j dD]'}|j |}|s|j|)|||d} | S#t$r&tjjd}YtwxYw)z@Prepare batch payload according to CloudZero AnyCost API format.r*z%Y-%mTr()monthrr)rstrptimer1rEnowr._convert_cbf_to_api_formatr2) rr"r#rdate_obj month_str data_recordsr7recordrWs rrLz(CloudZeroStreamer._prepare_batch_payloads 9&// JGH ))'2I ! ''d'3 ,C44S9F##F+ , $-9l[ 9 //8I 9s'A//,BBr7c i}|jD]h\}}| t|ttfrEt|tr&|dj dj d||<Ut |||<d|||<jd|vr|j |d|d<|S#t$r)}|jjd|dYd}~yd}~wwxYw)z\Convert CBF row to CloudZero API format - keeping CBF field names as CloudZero expects them.Nz.10f0.r'z9[yellow]Warning: Could not convert record to API format: r,) r isinstanceintfloatrstripstr_ensure_utc_timestampr3rr)rr7 api_recordkeyvaluer9s rr]z,CloudZeroStreamer._convert_cbf_to_api_formats  "J"iik 0 U$!%#u6%eU316t .D.DS.I.P.PQT.UJsO.1%jJsO*/ 3 0"Z/151K1KJWiLj1k -.   LL  !Z[\Z]]fg h sBA>B C%C  Ccl|s1tjtjj S |j |}|j j ddS#t$rDtjtjj j ddcYSwxYw)z*Ensure timestamp is in UTC format for API.r>r=)rr\rr isoformatr0rCr3)rr6r8s rrjz'CloudZeroStreamer._ensure_utc_timestamps<< -779 9 Q99-HB<<>))(C8 8 Q<< -779AA(CP P Qs0A&&A B32B3)N)replace_hourly)__name__ __module__ __qualname____doc__rirr4r5r$dictrrr0r rrLr]rjr%rr r s ^..C.d .$FF#FUYF$k2<<kDbll9J4KkB/R#/R(/Rb"C"R\\"VY"^b"H",,[^cghkmphpcq*d38nc3hRVAV< Q3 Q3 Qr%r ) rtrrrtypingrrrMpolarsr4 rich.consolerr rvr%rrzs+&C'  ZQZQr%