o ưig'@s^dZddlZddlmZmZddlmZmZmZddlZddl Z ddl m Z GdddZ dS)zds z4CloudZeroStreamer._group_by_date..) columnsr r iter_rowsget_parse_and_convert_timestampstrftimeappend Exceptionr)rrr timestamp_strrowdtrerrrrEs*      z CloudZeroStreamer._group_by_dater4c Csz6|drt|dd}n d|vs|drt|}nt|}|jdur0|j|jd}|tjWSt yL}z t d|d|d}~ww) 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)rr4r6r7rrrr0fs    z.CloudZeroStreamer._parse_and_convert_timestamprr c CsD|rdSd|jdd}|jd|jd}||||}zEtjdd4}|jd |d t |d |j |||d }| |jd |d t |dWdWdS1s^wYWdStj y} z|jd|d| dd} ~ wtj y} z|jd|d| jjd| jjdd} ~ ww)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 r r _prepare_batch_payloadhttpxZClientr rrpostraise_for_status RequestErrorZHTTPStatusErrorresponse status_codetext) rrr rrCurlpayloadclientrKr7rrrrs.  &(z#CloudZeroStreamer._send_daily_batchc Cszz t|d}|d}Wntytd}Ynwg}|jddD]}||}|r4||q&|||d} | S)z@Prepare batch payload according to CloudZero AnyCost API format.r%z%Y-%mTr#)monthrr)rstrptimer1r@nowr._convert_cbf_to_api_formatr2) rrr rZdate_obj month_strZ data_recordsr5recordrOrrrrFs"    z(CloudZeroStreamer._prepare_batch_payloadr5c CszCi}|D],\}}|dur3t|ttfr/t|tr(|ddd||<qt|||<q|||<qd|vrA||d|d<|WSty`}z|j d|dWYd}~dSd}~ww)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_timestampr3r r)rr5Z api_recordkeyvaluer7rrrrTs$ z,CloudZeroStreamer._convert_cbf_to_api_formatcCs\|s ttjSz||}|ddWSty-ttjddYSw)z*Ensure timestamp is in UTC format for API.r9r8)rrSrr isoformatr0r>r3)rr4r6rrrr^s  z'CloudZeroStreamer._ensure_utc_timestamp)N)r)__name__ __module__ __qualname____doc__r]rrr(r)r!dictrrr0rrrFrTr^rrrrrs!$ &r)rerrrtypingrrrrGZpolarsr(Z rich.consolerrrrrrs