U Cv f@sddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z mZe rddlZddlmZddl mZdd d gZd d Zd dZddZddZddZddddZejddddddZdddd Zd!d"d#d$d%ZdS)&) annotationsN) entry_points) TYPE_CHECKINGAny)BACKEND_ENTRYPOINTSBackendEntrypoint)BufferedIOBase)AbstractDataStoreZnetcdf4Zh5netcdfZscipyc Cst|ddd}tj|ddd}g}|D]n\}}tt|}||dt|}|dkr*dd|D}|d}td |d |d |d |d t q*|S)NcSs|jSNnameepr;/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/plugins.pyz#remove_duplicates..)keycSs|jSr r r rrrrrrcSsg|]}|jddqS):r)valuesplit).0errr "sz%remove_duplicates..zFound z! entrypoints for the engine name z: z. The entrypoint z will be used.) sorted itertoolsgroupbylistsetappendlenwarningswarnRuntimeWarning) entrypointsZentrypoints_groupedZunique_entrypointsr matchesZ matches_lenZall_module_namesZselected_module_namerrrremove_duplicatess  r'cCsft|}|j}g}|D]@\}}|jtjjtjjfkrJtd|d|dkr| |qt |S)NzAll the parameters in zB signature should be explicit. *args and **kwargs is not supportedself) inspect signature parametersitemskind Parameter VAR_KEYWORDVAR_POSITIONAL TypeErrorr tuple) open_datasetr*r+Zparameters_listr paramrrrdetect_parameters-s   r5c Cshi}|D]Z}|j}z|}|||<Wqtk r`}ztd|d|tW5d}~XYqXq|S)NzEngine z loading failed: )r load Exceptionr"r#r$)r%backend_entrypointsZ entrypointr backendexrrrbackends_dict_from_pkg?s ,r;cCs0|D]"\}}|jdkr|j}t||_qdSr )r,Zopen_dataset_parametersr3r5)r8r r9r3rrrset_missing_parametersKs r<csLi}tD]"}|ddk r|||<q|fddtD|S)Ncsi|]}||qSrr)rr r8rr Xsz!sort_backends..)STANDARD_BACKENDS_ORDERgetpopupdater)r8Zordered_backends_entrypointsZbe_namerr=r sort_backendsRsrCzdict[str, BackendEntrypoint])returncCs`i}tD]\}}|jr |||<q t|}t|}||t|}t|dd|DS)NcSsi|]\}}||qSrr)rr r9rrrr>gsz!build_engines..)rr, availabler'r;rBrCr<)r%r8 backend_namer9Zexternal_backend_entrypointsrrr build_engines]s  rGr)maxsizecCs,tjdkrtdd}ntdd}t|S)a Return a dictionary of available engines and their BackendEntrypoint objects. Returns ------- dictionary Notes ----- This function lives in the backends namespace (``engs=xr.backends.list_engines()``). If available, more information is available about each backend via ``engs["eng_name"]``. # New selection mechanism introduced with Python 3.10. See GH6514. ) zxarray.backends)groupr)sys version_inforr@rG)r%rrr list_enginesjs  rNz;str | os.PathLike[Any] | BufferedIOBase | AbstractDataStore) store_specc Cst}|D]H\}}z||r,|WSWqtk rTt|dtYqXqg}tD]N\}}z|}||r||Wqdtk rt|dtYqdXqddd|D}|s|rd|d}qd}n d|d}t |dS) Nz fails while guessingcSsg|]}|dkr|qS)storer)rkrrrrsz guess_engine..zHdid not find a match in any of xarray's currently installed IO backends a. Consider explicitly selecting one of the installed engines via the ``engine`` parameter, or installing additional IO dependencies, see: https://docs.xarray.dev/en/stable/getting-started-guide/installing.html https://docs.xarray.dev/en/stable/user-guide/io.htmlaxarray is unable to open this file because it has no currently installed IO backends. Xarray's read/write support requires installing optional IO dependencies, see: https://docs.xarray.dev/en/stable/getting-started-guide/installing.html https://docs.xarray.dev/en/stable/user-guide/iozIfound the following matches with the input file in xarray's IO backends: z. But their dependencies may not be installed, see: https://docs.xarray.dev/en/stable/user-guide/io.html https://docs.xarray.dev/en/stable/getting-started-guide/installing.html) rNr,Zguess_can_openr7r"r#r$rr ValueError)rOenginesenginer9Zcompatible_enginesZ backend_clsZinstalled_engines error_msgrrr guess_engines2     rVzstr | type[BackendEntrypoint]r)rTrDcCsht|tr:t}||kr0td|dt|||}n*t|trVt|trV|}ntd||S)z3Select open_dataset method based on current engine.zunrecognized engine z must be one of: zLengine must be a string or a subclass of xarray.backends.BackendEntrypoint: ) isinstancestrrNrRrtype issubclassrr1)rTrSr9rrr get_backends  r[) __future__r functoolsr)rrLr"importlib.metadatartypingrrZxarray.backends.commonrrosiorr r?r'r5r;r<rCrG lru_cacherNrVr[rrrrs.        3