iT UddlmZddlmZddlmZmZmZmZddl m Z ddl m Z e gdZded<ed ZGd d eeZ dd Zdd ZddZddZddZy)) 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:eZdZdZdd dZd dZd dZd dZd dZy)NoPickleOptionz Wrapper that does not pickle the wrapped value. This wrapper will unpickle to contain a None. Useful for cached or sensitive values. Nc||_yN _opt_value)self opt_values X/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/io/cloud/_utils.py__init__zNoPickleOption.__init__ s #c|jSrrrs rgetzNoPickleOption.get#s rc||_yrr)rvalues rsetzNoPickleOption.set&s rcy)Nr%rs r __getstate__zNoPickleOption.__getstate__)src.tj|yr)rr)r_states r __setstate__zNoPickleOption.__setstate__-s%rr)rT | NonereturnNone)r+r*)r"r*r+r,)r+ tuple[()])r(r-r+r,) __name__ __module__ __qualname____doc__rr r#r&r)r%rrrrs $ &rrct|ttfr|St|r|r|dSt|tr|j j Sy)Nr) isinstancestrrr r _pl_partition_by base_path)sources r_first_scan_pathr81sJ&3+&  (Vay FK (&&000 rcRt|}|jd}|dk\r|d|SdS)Nz://r)r4find)pathpath_stris r_get_path_schemer>>s04yH eA68BQ<+t+rctfddDrydk(sdk(r]|jd}|jd|dz}|jd td z|d kDsd |vry d |cxkxr|kScSy )Nc3(K|] }|k( ywrr%.0xschemes r z _is_aws_cloud..Fs .16Q; .)s3s3aThttphttpsz.s3.z.amazonaws.com//r?F)anyr:len)rDfirst_scan_path bucket_end region_ends` r _is_aws_cloudrTEs . .. 6W,$))&1 $))*;Z!^L   c&kAoz BQ Fo%:* **** rc,tfddDS)Nc3(K|] }|k( ywrr%rAs rrEz"_is_azure_cloud..YsLqv{LrF)azazureadlabfsabfssrOrDs`r_is_azure_cloudr^Xs L$KL LLrc,tfddDS)Nc3(K|] }|k( ywrr%rAs rrEz _is_gcp_cloud..]s9qv{9rF)gsgcpgcsr\r]s`r _is_gcp_cloudrd\s 9$89 99rN)r7rr+zstr | Path | None)r;z str | Pathr+z str | None)rDr4rQr4r+bool)rDr4r+re) __future__rpathlibrtypingrrrrpolars._utils.variousr polars.io.partitionr frozensetr__annotations__rrr8r>rTr^rdr%rrrmsv"//9+5> 51  CL&WQZ&2    ,&M:r