jT UddlmZddlmZddlmZmZmZmZddl m Z ddl m Z e gdZded<ed ZGd d eeZddZddZddZd dZd dZdS)!) annotations)Path)AnyFinalGenericTypeVar)is_path_or_str_sequence) PartitionBy)file_cache_ttl max_retriesretry_timeout_msretry_init_backoff_msretry_max_backoff_msretry_base_multiplierzFinal[frozenset[str]]POLARS_STORAGE_CONFIG_KEYSTc<eZdZdZdddZddZdd Zdd ZddZdS)NoPickleOptionz Wrapper that does not pickle the wrapped value. This wrapper will unpickle to contain a None. Useful for cached or sensitive values. N opt_valueT | NonereturnNonec||_dSN _opt_value)selfrs [/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/io/cloud/_utils.py__init__zNoPickleOption.__init__ s #c|jSrrrs rgetzNoPickleOption.get#s r valuec||_dSrr)rr$s rsetzNoPickleOption.set&s r tuple[()]cdS)Nr)r"s r __getstate__zNoPickleOption.__getstate__)srr _statec:t|dSr)rr)rr+s r __setstate__zNoPickleOption.__setstate__-s%%%%%r r)rrrr)rr)r$rrr)rr')r+r'rr) __name__ __module__ __qualname____doc__rr#r&r*r-r)r rrrs$$$$$    &&&&&&r rsourcerrstr | Path | Nonect|ttfr|St|r |r|dSt|tr |jjSdS)Nr) isinstancestrrr r _pl_partition_by base_path)r2s r_first_scan_pathr91sc&3+&&1  ( (1V1ay FK ( (1&00 4r path str | Path str | Nonecnt|}|d}|dkr |d|ndS)Nz://r)r6find)r:path_stris r_get_path_schemerA>s84yyH eA668BQB<z _is_aws_cloud..Fs' . .16Q; . . . . . .r )s3s3aThttphttpsz.s3.z.amazonaws.com//r?F)anyr>len)rBrC bucket_end region_ends` r _is_aws_cloudrWEs . . . . . . ...t 6W,,$))&11 $))*;Z!^LL   c&kkAoz B BQ F Fo%%5:**** ***** 5r c:tfddDS)Nc3$K|] }|kV dSrr)rGs rrJz"_is_azure_cloud..Ys'LLqv{LLLLLLr )azazureadlabfsabfssrSrBs`r_is_azure_cloudraXs' LLLL$KLLL L LLr c:tfddDS)Nc3$K|] }|kV dSrr)rGs rrJz _is_gcp_cloud..]s'99qv{999999r )gsgcpgcsr_r`s`r _is_gcp_cloudrg\s' 9999$8999 9 99r N)r2rrr3)r:r;rr<)rBr6rCr6rrD)rBr6rrD) __future__rpathlibrtypingrrrrpolars._utils.variousr polars.io.partitionr frozensetr__annotations__rrr9rArWrargr)r rrosZ"""""""////////////999999++++++5>I 5 5     GCLL&&&&&WQZ&&&2    ,,,,&MMMM::::::r