o d?Z.e'*e+d5ej$ed"e d5e%d dfd@dAZ/e'*e+d5ej$ed"e d5e%d dfdBdCZ0e'*e+d5ej$ed"e d5e%d dfdDdEZ1e'*e+d5ej$ed"e d5e%d dfdFdGZ2e'*ej$ed"e d dfdHdIZ3e' dqdJdKZ4e4*e+d5ej$ed"e d5e%d dfdLdMZ5e' dqdNdOZ6e6*e+d5ej$ed"e d5e%d dfdPdQZ7e6*e+d5ej$ed"e d5e%d dfdRdMZ5e'*e+dSe+dTej$ed"e dSe%dTe%d dfdUdVZ8e' dqdWdXZ9e9 dqdYdZZ:e:*dMe+d5ej+d[dd/ej$ed"e d5e%d[e%d dfd\d]Z;e:*dQe+d5ej+d[dd/ej$ed"e d5e%d[e%d dfd^d_Ze>*e+d5e+d[ej$ed"e d5e%d[e%d dfdgdMZ5e>*e+d5e+d[ej$ed"e d5e%d[e%d dfdhdQZ7e'*e+d5ej!didd/ej!dej"dej$ed"e d5e%dje%d$e&d df dkdlZ?dmedne@e%e%fd e@e%e%ffdodpZAdS)rNCallable)wraps)AnyLiteral)Context) __version__)URICatalog load_catalog) ConsoleOutput JsonOutputOutput)NoSuchNamespaceErrorNoSuchPropertyExceptionNoSuchTableError)TableProperties) SnapshotRefSnapshotRefType)property_as_intreturncCsdtdtfdd}|S)Nfuncrcs"tdtdtffdd }|S)Nargskwargsc sdz|i|WSty1}ztjdd}t|\}}|||dWYd}~dSd}~ww)NT)silent) Exceptionclickget_current_context_catalog_and_output exceptionexit)rrectx_outputrR/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/pyiceberg/cli/console.pywrapper(s   z3catch_exception..decorator..wrapper)rr)rr)r'r&r( decorator's z"catch_exception..decoratorr)r*r'r'r(catch_exception&s r+z --catalogz --verbose)typez--outputtextjson)r,defaultz --log-level)DEBUGINFOWARNINGERRORCRITICALF)case_sensitiver2PYICEBERG_LOG_LEVELzSet the logging level)r,r/envvarhelpz--ugiz--uriz --credentialr#catalogverboser% log_levelugiuri credentialc Cstjtt|ddi}|r||d<|r||t<|r ||d<|t|dkr2t|d|jd<nt |d|jd<z t |fi||jd<Wnt yf} z|jd | | d WYd} ~ nd} ~ wwt|jdts|jd td | d dSdS) Nz.%(asctime)s:%(levelname)s:%(name)s:%(message)s)levelformatr<r>r-)r:r%r9rz[Could not determine catalog type from uri. REST (http/https) and Hive (thrift) is supported)logging basicConfiggetattrupperr ensure_objectdictr objr r rr r! isinstancer ValueError) r#r9r:r%r;r<r=r> propertiesr"r'r'r(run7s8   rKcCs|jd|jdfS)zSmall helper to set the types.r9r%)rGr#r'r'r(rpsrparent)requiredcCs>t|\}}g}|r||}|s||pd}||dS)zList tables or namespaces.r'N)r list_tableslist_namespaces identifiers)r#rMr9r%rQr'r'r(listus  rRz--entity)any namespacetablerS identifierentity)namerTrUc Cst|\}}t|}d}|dvrDt|dkrDz||}||d}WntyC}z|dks7t|dkr9|WYd}~nd}~wwd} |dvrwt|dkrwz||} || d} Wnt yv}z |dkrl|WYd}~nd}~ww|dur| durt d |dSdS) z Describe a namespace or a table.F>rSrTrTrSrN>rSrUz#Table or namespace does not exist: ) rr identifier_to_tuplelenload_namespace_propertiesdescribe_propertiesr load_tabledescribe_tabler) r#rWrVr9r%identifier_tuple is_namespacenamespace_propertiesexcis_table catalog_tabler'r'r(describes:        rez --historyT)is_flaghistorycCs&t|\}}||}|||dS)z List all the files of the table.N)rr]files)r#rVrgr9r%rdr'r'r(rhs  rhcC(t|\}}||}||dS)zGet the schema of the table.N)rr]schemar#rVr9r%rUr'r'r(rj  rjcCri)z'Return the partition spec of the table.N)rr]specrkr'r'r(rmrlrmcCs(t|\}}||j}||jdS)zReturn the UUID of the table.N)rr]metadatauuid table_uuid)r#rVr9r%rnr'r'r(ros  rocCs(t|\}}||}||dS)z!Return the location of the table.N)rr]r-locationrkr'r'r(rqrlrqcCs|jdtdS)zPrint pyiceberg version.r%N)rGversionrrLr'r'r(rrsrrcCdS)z Operation to create a namespace.Nr'r'r'r'r(creatertcC*t|\}}|||d|dS)zCreate a namespace.zCreated namespace: N)rcreate_namespacer-r#rVr9r%r'r'r(rT  rTcCrs)z(Operations to drop a namespace or table.Nr'r'r'r'r(droprurzcCrv)z Drop a table.zDropped table: N)r drop_tabler-rxr'r'r(rUryrUcCrv)zDrop a namespace.zDropped namespace: N)rdrop_namespacer-rxr'r'r(rT ryfrom_identifier to_identifiercCs2t|\}}||||d|d|dS)zRename a table.zRenamed table from z to N)r rename_tabler-)r#r}r~r9r%r'r'r(renames  rcCrs)z Properties on tables/namespaces.Nr'r'r'r'r(rJ%rurJcCrs)z&Fetch properties on tables/namespaces.Nr'r'r'r'r(get*rur property_namecCsbt|\}}t|}||}|r*||}r ||dStd|d|||dS)z Fetch properties on a namespace.Could not find property z on namespace N)rr rYr[rr-rr\)r#rVrr9r%r_raproperty_valuer'r'r( get_namespace/s   rcCsht|\}}t|}||j}|r,|j|}r"||dStd|d|| |jdS)zFetch properties on a table.rz on table N) rr rYr]rnrJrr-rr\)r#rVrr9r%r_rnrr'r'r( get_tableDs   rcCrs)z$Set a property on tables/namespaces.Nr'r'r'r'r(setYrurrcCs8t|\}}|j|||id|d|d|dS)zSet a property on a namespace.)updateszUpdated  on N)rupdate_namespace_propertiesr-)r#rVrrr9r%r'r'r(rT^s c Csvt|\}}t|}||}|}|||iWdn1s&wY|d|d|d|dS)zSet a property on a table.NzSet =r)rr rYr] transactionset_propertiesr-) r#rVrrr9r%r_rUtxr'r'r(rUls     cCrs)z)Remove a property from tables/namespaces.Nr'r'r'r'r(remove}rurcCsVt|\}}|j||hd}|j|gkr!|d|d|dStd|d|)z#Remove a property from a namespace.)removals Property  removed from  does not exist on N)rrremovedr-r)r#rVrr9r%resultr'r'r(rTs  cCst|\}}||}||jjvr7| }||Wdn1s%wY|d|d|dStd|d|)zRemove a property from a table.Nrrr)rr]rnrJrremove_propertiesr-r)r#rVrr9r%rUrr'r'r(rUs     z--typer,csnt|\}}||}r$tjtjhvr$tdfdd|D}| |dS)z(List all the refs in the provided table.z(Type must be either branch or tag, got: cs2g|]\}}r |jkr||jt|jfqSr')snapshot_ref_type_retention_propertiesrJ).0ref_namerefrUr,r'r( s zlist_refs..N) rr]refslowerrBRANCHTAGrIitems describe_refs)r#rVr,r:r9r%r relevant_refsr'rr( list_refss   rrtable_propertiescCsi}|jtjkr5t|tjtj}t|tjtj}|j r t |j nt ||d<|j r.t |j nt ||d<nd|d<d|d<|j rEt |j nd|d<|S)Nmin_snapshots_to_keepmax_snapshot_age_mszN/Aforevermax_ref_age_ms) rrrrrMIN_SNAPSHOTS_TO_KEEPMIN_SNAPSHOTS_TO_KEEP_DEFAULTMAX_SNAPSHOT_AGE_MSMAX_SNAPSHOT_AGE_MS_DEFAULTrstrrr)rrretention_propertiesdefault_min_snapshots_to_keepdefault_max_snapshot_age_msr'r'r(rs( r)rN)BrAcollections.abcr functoolsrtypingrrrr pyicebergrpyiceberg.catalogr r r pyiceberg.cli.outputr r rpyiceberg.exceptionsrrrpyiceberg.tablerpyiceberg.table.refsrrpyiceberg.utils.propertiesrr+groupoptionBOOLChoice pass_contextrboolrKtuplercommandargumentrRrerhrjrmrorqrrrtrTrzrUrrJrrrrrrrFrr'r'r'r(sR       * "( $  $   $ $ ** $ $  **