o ;i@sZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z m Z mZdd lmZejd d d d ZdZejdd de jejdddedejdddeddejddddejdddddejddddejddd d!dejd"d#d$dejd"d%d&dfd'eed(eed)eed*ed+eed,eed-ed.efd/dZdS)0)datetime)dumps)OptionalN)_workspace_billing_report) synchronizer) parse_dateparse_date_rangeresolve_timezone) display_tablebillingz#View workspace billing information.T)namehelpno_args_is_helpz\Date (in UTC by default): ISO format (2025-01-01) or relative (yesterday, 3 days ago, etc.).report)rz--startz Start date. )rz--endz End date. z Defaults to now.z--forzRConvenience range: today, yesterday, this week, last week, this month, last month.dz --resolutionz-rz-Time resolution: 'd' (daily) or 'h' (hourly).z--tzzlTimezone for date interpretation: 'local', offset (5, -4, +05:30), or IANA name. Requires hourly resolution.z --tag-namesz-tz-Comma-separated list of tag names to include.Fz--jsonzOutput as JSON.z--csvzOutput as CSV.startendfor_ resolutiontz tag_namesjsoncsvc sr r tdd|dur*zt|Wnty)}ztt|d}~ww|durY|dus6|dur;tdz t|d\} } WnDtyX}ztt|d}~ww|durzt|d} |rlt|dnd} Wnttfy}ztt|d}~wwtd|rdd|dDnd} d vrtd durd krtd t | | | d IdH} gd} | r| ddt dtffdd }g}| D]'}|d|d|d||dt|dg}| r| t |d| |qt | |ddS)anGenerate a billing report for the workspace. The report range can be provided by setting `--start` / `--end` dates (`--end` defaults to 'now') or by requesting a date range using `--for` (e.g., `--for today`, `--for 'last month'`). Note that, as with the Billing API, the start date is inclusive and the end date is exclusive. Data will be reported for full intervals only. Using `--for` is a convenient way to define a complete interval. Examples: modal billing report --start 2025-12-01 --end 2026-01-01 modal billing report --for "last month" --tag-names team,project modal billing report --for today --resolution h modal billing report --for yesterday -r h --tz local modal billing report --for "last month" --csv > report.csv modal billing report --start 2025-12-01 --json > report.json z'--json and --csv are mutually exclusiveNz2--for is mutually exclusive with --start and --end)rz#Either --for or --start is requiredcSsg|]}|qS)strip).0trrE/home/ubuntu/.local/lib/python3.10/site-packages/modal/cli/billing.py bszreport..,)rhz.Resolution must be 'd' (daily) or 'h' (hourly)r!z--tz requires hourly resolution (--resolution h / -r h). Daily intervals are UTC-aligned and cannot be shifted to a custom timezone.)rrrr)z Object ID Description EnvironmentzInterval StartCostTagsdtreturncsLdur |}n|jdd}sr|Sdkr!|dS|dS)N)tzinforz%Y-%m-%dz%Y-%m-%dT%H:%M) astimezonereplace isoformatstrftime)r&rrr resolved_tzrrformat_interval|s    zreport..format_interval object_id descriptionenvironment_nameinterval_startcosttags)rr)click UsageErrorr ValueErrorstrrr OverflowErrorsplitrappendrrr )rrrrrrrrexcstart_dtend_dtr5 rows_datacolumnsr/rowsitemrowrr-rrst0           )rrrtypingrr6typermodal._billingrmodal._utils.async_utilsrmodal._utils.time_utilsrrr utilsr Typer billing_cli DATE_HELPcommandcreate_blockingOptionr9boolrrrrrs\