hg'ddZddlZddlmZmZddlmZmZmZddlZddl Z ddl m Z GddZ y)z+??`aj`kkstu&3&9&9&; "J  " ":z9 E'<c &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 Exceptionr pl DataFrame) rrr" timestamp_strrowdtr#edate_keyrecordss rrz CloudZeroStreamer._group_by_dateEs9;  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 r7c |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 fromisoformatreplacerAr astimezonerr ValueError)rr7r9r:s rr1z.CloudZeroStreamer._parse_and_convert_timestampfs R%%c*++M,A,A#x,PQ %)?)?Au*v ++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$rrJurlpayloadclientrTr:s rr!z#CloudZeroStreamer._send_daily_batchs     't||n5.  ?@R@R?SSab--j*iP d+ v ""%=j\CPZOK\\l#mn!;;sG';J))+ ""%LZLXZ[^_i[jZkk|#}~    !!  LL  !J:,VXYZX[[ab c $$  LL  !G |SUVWV`V`VlVlUmmnopoyoyo~o~n@F G H  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)rstrptimer2rFnowr/_convert_cbf_to_api_formatr3) rr#r$rdate_obj month_str data_recordsr8recordrXs rrMz(CloudZeroStreamer._prepare_batch_payloads 9((Z@H ))'2I  ''d'3C44S9F##F+4 "   # 9 //8I 9s'A//,BBr8c 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_timestampr4rr)rr8 api_recordkeyvaluer:s rr^z,CloudZeroStreamer._convert_cbf_to_api_formats J"iik U$!%#u6%eU316t .D.DS.I.P.PQT.UJsO.1%jJsO*/ 3*"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 isoformatr1rDr4)rr7r9s rrkz'CloudZeroStreamer._ensure_utc_timestamps<< -779 9 Q22=AB<<>))(C8 8 Q<< -779AA(CP P Qs0A&&A B32B3)N)replace_hourly)__name__ __module__ __qualname____doc__rjrrr5r6r%dictrrr1r!rrMr^rkr&rr r s ^..C.QT .$FF#FUYF$k2<<kDbll9J4KkBR#R(R4CR\\VY^b@",,[^cghkmphpcq2d38n$sTWx.AY< Q3 Q3 Qr&r )rurrrtypingrrrrNpolarsr5 rich.consolerr rwr&rr{s.&C'''  CQCQr&