o ;i2@sddlmZmZddlmZddlmZmZmZddlm Z ddl m Z ddl m Z Gdd d eZd d d d d d edeededeeedee deeeeff ddZd S))datetimetimezone)Decimal)AnyOptional TypedDict)api_pb2)_Client) InvalidErrorc@sFeZdZUeed<eed<eed<eed<eed<eeefed<dS)WorkspaceBillingReportItem object_id descriptionenvironment_nameinterval_startcosttagsN)__name__ __module__ __qualname__str__annotations__rrdictrrB/home/ubuntu/.local/lib/python3.10/site-packages/modal/_billing.pyr s r Nd)end resolution tag_namesclientstartrrrrreturnc s|dur tIdH}|pg}|durttj}|jdur'|jtjd}n |jtjkr1td|jdur>|jtjd}n |jtjkrHtdt j ||d}|j ||j |g}|jj|2z%3dHW}|j|j|j|jjtjdt|jt|jd}||qd6|S)aGenerate a tabular report of workspace usage by object and time. The result will be a list of dictionaries for each interval (determined by `resolution`) between the `start` and `end` limits. The dictionary represents a single Modal object that billing can be attributed to (e.g., an App) along with metadata (including user-defined tags) for identifying that object. The `start` and `end` parameters are required to either have a UTC timezone or to be timezone-naive (which will be interpreted as UTC times). The timestamps in the result will be in UTC. Cost will be reported for full intervals, even if the provided `start` or `end` parameters are partial: `start` will be rounded to the beginning of its interval, while partial `end` intervals will be excluded. Additional user-provided metadata can be included in the report if the objects have tags and `tag_names` (i.e., keys) are specified in the request. Alternatively, pass `tag_names=["*"]` to include all tags in the report. Note that tags will be attributed to the entire interval even if they were added or removed at some point within it. If the tag name was not in use during an interval, it will be absent from the tags dictionary in that output row. In most cases, billing data will be available in the database that this API queries within minutes, although there may be collection delays. If completeness is important for your use case, we recommend leaving a buffer after the end of the query interval. N)tzinfoz0Timezone-aware 'start' parameter must be in UTC.z.Timezone-aware 'end' parameter must be in UTC.)rr)r rrrrr)r from_envrnowrutcr"replacer rWorkspaceBillingReportRequeststart_timestamp FromDatetime end_timestampstubWorkspaceBillingReport unary_streamr rrinterval ToDatetimerrrrappend) r rrrrrequestrowspb_itemitemrrr_workspace_billing_reports@          r5)rrdecimalrtypingrrr modal_protorrr exceptionr r rlistrr5rrrrs0