o ;i@sDddlmZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZmZdd lmZdd lmZdd lmZdd lmZed dddZejddde jeddedeefddZejddde jdedde deefddZ!ejddde jeed ded!e deefd"dZ"ejd#dde jedd$d%d&eed'ded(e d!e deefd)d#Z#ejd*d+de jedded,edeefd-d*Z$d.ed/e d0efd1d2Z%ejd3d+de jed4d5d6fedd7d8d9d&edd:d;de&d?e d/e de deef d@d3Z'dS)A)OptionalN)ArgumentOptionTyper) LoadContext)Resolver) TaskContext synchronizer)timestamp_to_localized_str) ENV_OPTION YES_OPTION display_table)_Client)_Dict) ensure_env) OutputManagerdictTz7Manage `modal.Dict` objects and inspect their contents.)nameno_args_is_helphelpcreate Management)rrich_help_panel)envrrc stj||dd}tIdH}t}t4IdH}t|||d}|||IdHWdIdHdS1IdHstj||d}|stjd|dddd|IdHdS)z?Clear the contents of a named Dict by deleting all of its data.r)zHAre you sure you want to irrevocably delete the contents of modal.Dict ''?FTdefaultabortN)rrtyperconfirmr5)rr6rr!r%r%r&r56s deletez--allow-missingz&Don't error if the Dict doesn't exist.)r) allow_missingr6rr>cs:|stjd|ddddtjj|||dIdHdS)z(Delete a named Dict and all of its data.zN)r;r<rr+r=)rr>r6rr%r%r&r=Ds  get Inspectionkeycs2tj||d}||IdH}t|dS)z|Print the value for a specific key. Note: When using the CLI, keys are always interpreted as having a string type. r)N)rrr?rprint)rrArr!valr%r%r&r?Wsinputuse_reprreturncCs4|rt|nt|}t|dkr|dddS|S)NP...)reprstrlen)rDrErCr%r%r&_displaycs rLitemsz!Limit the number of entries shown)r9rz-az--allz8Ignore N and print all entries in the Dict (may be slow)z-rz--reprz0Display items using `repr()` to see more details)allrEr(rnrOc stj||d}dg}}|2z13dHW\} } |d7}|s,|s,||kr,|dn|r3| | f} n t| |t| |f} || q6tddg||dS)zPrint the contents of a Dict. Note: By default, this command truncates the contents. Use the `N` argument to control the amount of data shown or the `--all` option to retrieve the entire Dict, which may be slow. r)rN)rHrHKeyValue)rrrMr-rLr ) rrPrOrEr(rr!irMrArC display_itemr%r%r&rMhs     )(typingrr;rrrmodal._load_contextrmodal._resolverrmodal._utils.async_utilsrr modal._utils.time_utilsr modal.cli.utilsr r r modal.clientr modal.dictrmodal.environmentsr modal.outputrdict_clicommandcreate_blockingrJrboolr4r5r=r?rLintrMr%r%r%r&s           " &   $