o ;iV@sjddlmZddlmZddlZddlmZmZmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd lmZddlmZeddddZedddddZej ddde j!edde"dee"fddZ#ej ddde j!edd d!deed"de"d#e$d$e$dee"fd%dZ%ej d&dde j!ded'd(e$dee"fd)d*Z&ej d+dde j!eedd,d-d.defedde"d/ee"d0e$d$e$dee"f d1d+Z'ej d2d3de j!ed4efedde"d5e(d/ee"dee"fd6d2Z)ej d7d3de j!eedd8d9d:dfedde"d/ee"d;e$dee"fd)datetime)OptionalN)ArgumentOptionTyper) LoadContext)Resolver) TaskContext synchronizer)timestamp_to_localized_str) ENV_OPTION YES_OPTION display_table)_Client) ensure_env) OutputManager)_Queue)api_pb2queueTz8Manage `modal.Queue` objects and inspect their contents.)nameno_args_is_helphelpz-pz --partitionzNName of the partition to use, otherwise use the default (anonymous) partition.)rcreate Management)rrich_help_panel)envrrc stj||dd}tIdH}t}t4IdH}t|||d}|||IdHWdIdHdS1IdHs.retrieve_pageTcs"g|] }tj|j|jddqS)T)is_another_app)r _new_hydratedqueue_idmetadata).0item)rr'r( bs"zlist_..>)Namez Created atz Created by Partitionsz Total size)rrr!floatboolrnow timestamprM creation_info created_atzipinfoappendrr created_bystrnum_partitions total_sizer) r4rrHfinishedr@rowsobj resp_datar[r'rEr(list_Ds2 reclearz-az--allz%Clear the contents of all partitions. partitionallcsDtj||d}|stjd|dddd|j||dIdHdS) z:Clear the contents of a queue by removing all of its data.rzIAre you sure you want to irrevocably delete the contents of modal.Queue 'r,FTr-)rgrhN)rr r0r1rf)rrgrhr+rr#r'r'r(rfss  peek Inspectionncs\tj||d}t}d}|j|d2z3dHW}|||d7}||kr*dSq6dS)zIPrint the next N items in the queue or queue partition (without removal).rir)rgNrl)rr rgetiterateprint)rrmrgrr#outputirOr'r'r(rjs rAz-tz--totalz:Compute the sum of the queue lengths across all partitionstotalcs2tj||d}t|j||dIdHdS)zLPrint the length of a queue partition or the total length of all partitions.ri)rgrsN)rr rrnrprA)rrgrsrr#r'r'r(len_s "rt)+rtypingrr0rrrmodal._load_contextrmodal._resolverrmodal._utils.async_utilsr r modal._utils.time_utilsr modal.cli.utilsr r r modal.clientrmodal.environmentsr modal.outputr modal.queuer modal_protor queue_cliPARTITION_OPTIONcommandcreate_blockingr^rrUr)rerfintrjrtr'r'r'r(s               " -