o iT @sUddlmZddlmZddlmZmZmZmZddl m Z ddl m Z e gdZded<ed ZGd d d eeZd"ddZd#ddZd$ddZd%ddZd%dd Zd!S)&) annotations)Path)AnyFinalGenericTypeVar)is_path_or_str_sequence) PartitionBy)Zfile_cache_ttl max_retriesZretry_timeout_msZretry_init_backoff_msZretry_max_backoff_msZretry_base_multiplierzFinal[frozenset[str]]POLARS_STORAGE_CONFIG_KEYSTc@sDeZdZdZddddZdd d Zdd d ZdddZdddZdS)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 | NonereturnNonecC ||_dSNZ _opt_value)selfrrM/home/app/Keep/.python/lib/python3.10/site-packages/polars/io/cloud/_utils.py__init__  zNoPickleOption.__init__cCs|jSrrrrrrget#szNoPickleOption.getvaluecCrrr)rrrrrset&rzNoPickleOption.set tuple[()]cCsdS)Nrrrrrr __getstate__)szNoPickleOption.__getstate___statecCst|dSr)r r)rr rrr __setstate__-szNoPickleOption.__setstate__r)rrrr)rr)rrrr)rr)r rrr) __name__ __module__ __qualname____doc__rrrrr!rrrrr s    r sourcerrstr | Path | NonecCs<t|ttfr |St|r|r|dSt|tr|jjSdS)Nr) isinstancestrrrr Z_pl_partition_by base_path)r&rrr_first_scan_path1s  r+path str | Path str | NonecCs*t|}|d}|dkr|d|SdS)Nz://r)r)find)r,Zpath_strirrr_get_path_scheme>s r1schemer)first_scan_pathboolcstfdddDr dSdksdkrA|d}|d|d }|d td |d ks3d |vr5dSd |ko>|kSSdS)Nc3|]}|kVqdSrr.0xr2rr Fz _is_aws_cloud..)Zs3Zs3aThttphttpsz.s3.z.amazonaws.com//r?F)anyr/len)r2r3Z bucket_endZ region_endrr9r _is_aws_cloudEs rDctfdddDS)Nc3r5rrr6r9rrr:Yr;z"_is_azure_cloud..)azZazureadlZabfsZabfssrBr9rr9r_is_azure_cloudXrIcrE)Nc3r5rrr6r9rrr:]r;z _is_gcp_cloud..)gsZgcpZgcsrHr9rr9r _is_gcp_cloud\rJrLN)r&rrr')r,r-rr.)r2r)r3r)rr4)r2r)rr4) __future__rpathlibrtypingrrrrZpolars._utils.variousrZpolars.io.partitionr frozensetr __annotations__r r r+r1rDrIrLrrrrs