o i,@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZe r:ddlmZd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z Gd d d e Z!dIddZ"e rddl#Z#ddl$Z$ddl%Z%ddl&Z&ddl'Z'ddl(Z(ddl)Z)ddl*Z*ddl+Z+ddl,Z,ddl-Z-ddl.Z.ddl/Z/ddl0Z0ddl1Z1ddl2Z2ddl3Z3ddl4Z4ddl5Z4ddl6Z6ddl7Z7nxe"d\Z#Z8e"d\Z$Z8e"d\Z%Z8e"d\Z&Z8e"d\Z'Z8e"d\Z(Ze"d\Z)Z9e"d\Z*Ze"d\Z+Ze"d\Z,Ze"d\Z-Ze"d\Z.Ze"d\Z/Ze"d\Z0Ze"d \Z1Ze"d!\Z2Ze"d"\Z3Ze"d#\Z4Ze"d$\Z7Ze"d%\Z6Z edJd*d+Z:d d,dKd0d1Z;d d,dKd2d3Zd d,dKd8d9Z?d d,dKd:d;Z@ < =  > dLdMdFdGZAgdHZBdS)N) annotationsN)cache) import_module) find_spec) ModuleType) TYPE_CHECKINGAnyClassVarcast)HashableTcsReZdZUdZdZdddddZded <dfdd ZdddZdddZ Z S) _LazyModulea/ Module that can act both as a lazy-loader and as a proxy. Notes ----- We do NOT register this module with `sys.modules` so as not to cause confusion in the global environment. This way we have a valid proxy module for our own use, but it lives *exclusively* within polars. Tznp.zpd.zpa.zpc.)numpypandaspyarrow polars_cloudzClassVar[dict[str, str]]_mod_pfx module_namestrmodule_availableboolreturnNonecs$||_||_t|_t|dS)a Initialise lazy-loading proxy module. Parameters ---------- module_name : str the name of the module to lazy-load (if available). module_available : bool indicate if the referenced module is actually available (we will proxy it in both cases, but raise a helpful error when invoked if it doesn't exist). N)_module_available _module_nameglobals_globalssuper__init__)selfrr __class__K/home/app/Keep/.python/lib/python3.10/site-packages/polars/_dependencies.pyr2sz_LazyModule.__init__rcCs(t|j}||j|j<|j|j|SN)r__name__rr__dict__update)rmoduler!r!r"_importIs  z_LazyModule._importnamercCs|dkr|jd|}t||jr|}t||S|dkr#|jStd|r/|dkr/dS|j|jd}||d|jd}t |d) N __wrapped__z object has no attribute r$z ^__\w+__$ __version__z requires z module to be installed) rAttributeErrorrr(getattrrematchrgetModuleNotFoundError)rr)msgr'pfxr!r!r" __getattr__Ps  z_LazyModule.__getattr__)rrrrrr)rr)r)rrr) r$ __module__ __qualname____doc__Z__lazy__r__annotations__rr(r5 __classcell__r!r!rr"r s   r rrrtuple[ModuleType, bool]cCs`|tjvr tj|dfSzt|}|dup|jdu }Wn ty'd}Ynwt||d|fS)a Lazy import the given module; avoids up-front import costs. Parameters ---------- module_name : str name of the module to import, eg: "pyarrow". Notes ----- If the requested module is not available (eg: has not been installed), a proxy module is created in its place, which raises an exception on any attribute access. This allows for import and use as normal, without requiring explicit guard conditions - if the module is never used, no exception occurs; if it is, then a helpful exception is raised. Returns ------- tuple of (Module, bool) A lazy-loading module and a boolean indicating if the requested/underlying module exists (if not, the returned module is a proxy). TNF)rr)sysmodulesrloaderr2r )rZ module_specrr!r!r" _lazy_importks  r? dataclasseshtmljsonpickle subprocessaltairboto3 deltalakefsspecgevent great_tables hypothesisr rrrpydantic pyicebergtorchpytzclstypetype_rcs2ztfdd|DWStyYdSw)Nc3s"|] }dt|vVqdS).Nr).0orRr!r" s z_might_be..F)anymro TypeError)rPrRr!rWr" _might_bes  r\) check_typeobjrr]cC tottd|r t|n|dS)Nr r )_NUMPY_AVAILABLEr\r rQr^r]r!r!r"_check_for_numpyrbcCr_)Nr r)_PANDAS_AVAILABLEr\r rQrar!r!r"_check_for_pandasrcrecCr_)Nr r)_PYARROW_AVAILABLEr\r rQrar!r!r"_check_for_pyarrowrcrgcCr_)Nr rL)_PYDANTIC_AVAILABLEr\r rQrar!r!r"_check_for_pydanticrcricCr_)Nr rN)_TORCH_AVAILABLEr\r rQrar!r!r"_check_for_torchrcrkcCr_)Nr rO)_PYTZ_AVAILABLEr\r rQrar!r!r"_check_for_pytzrcrmrequired package not foundrequires err_prefix err_suffix min_versionstr | tuple[int, ...] | Nonemin_err_prefixinstall_message str | Nonec Cs ddlm}ddlm}|ddd}zt|} Wn7tyQ|r+|ddnd} |r7d|dnd} | d|d| d |pJd |d } t| d w|r||}|| j } | |kr|d|dd d d|Ddd dd| Dd}||| S)aj Import an optional dependency, returning the module. Parameters ---------- module_name : str Name of the dependency to import. err_prefix : str, optional Error prefix to use in the raised exception (appears before the module name). err_suffix: str, optional Error suffix to use in the raised exception (follows the module name). min_version : {str, tuple[int]}, optional If a minimum module version is required, specify it here. min_err_prefix : str, optional Override the standard "requires" prefix for the minimum version error message. install_message : str, optional Override the standard "Please install it using..." exception message fragment. Examples -------- >>> from polars._dependencies import import_optional >>> import_optional( ... "definitely_a_real_module", ... err_prefix="super-important package", ... ) # doctest: +SKIP ImportError: super-important package 'definitely_a_real_module' not installed. Please install it using the command `pip install definitely_a_real_module`. r) parse_version)ModuleUpgradeRequiredErrorrS r,'z. z.Please install using the command `pip install z`.Ncs|]}t|VqdSr#rTrUvr!r!r"rX2z"import_optional..z or higher (found csr}r#rTr~r!r!r"rX3r)) Zpolars._utils.variousrxZpolars.exceptionsrysplitr ImportErrorstripr2r+join)rrqrrrsrurvrxryZ module_rootr'prefixsuffixZ err_messageZ mod_versionr3r!r!r"import_optionals4 $        r)&r@rArBrCrDrErFrGrHrIrJr rrrLrMrrNrOrbrergrirkrm_ALTAIR_AVAILABLE_DELTALAKE_AVAILABLE_FSSPEC_AVAILABLE_GEVENT_AVAILABLE_GREAT_TABLES_AVAILABLE_HYPOTHESIS_AVAILABLEr`rd_POLARS_CLOUD_AVAILABLErfrh_PYICEBERG_AVAILABLErj)rrrr;)rPrQrRrrr)r^rr]rrr)rnroNrpN)rrrqrrrrrsrtrurrvrwrr)C __future__rr/r< functoolsr importlibrimportlib.utilrtypesrtypingrrr r collections.abcr rrrrrrr`rdrrfrhrrjrlr r?r@rArBrCrDrErFrGrHrIrJrKr rrrrLrMZpyiceberg.schemarOrN_Z_BOTO3_AVAILABLEr\rbrergrirkrmr__all__r!r!r!r"s       M-                        A