U Cv f@sddlmZddlZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZmZmZmZmZmZdd lmZGd d d e Zd d ZGdddeZGddde Zeed<dS)) annotationsN)Version)BACKEND_ENTRYPOINTSAbstractDataStore BackendArrayBackendEntrypointrobust_getitem)StoreBackendEntrypoint)indexing) integer_types)Frozen FrozenDictclose_on_error is_dict_like is_remote_urimodule_available)Variablec@sBeZdZddZeddddZeddZd d Zd d Zd S)PydapArrayWrappercCs ||_dSN)array)selfrr:/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/pydap_.py__init__szPydapArrayWrapper.__init__ztuple[int, ...])returncCs|jjSr)rshaperrrrrszPydapArrayWrapper.shapecCs|jjSr)rdtyperrrrr#szPydapArrayWrapper.dtypecCst||jtjj|jSr)r Zexplicit_indexing_adapterrZIndexingSupportZBASIC_getitem)rkeyrrr __getitem__'s zPydapArrayWrapper.__getitem__cCs\t|jd|j}t||td}tddt|D}|jt||jkrX|rXt ||}|S)Nr)Zcatchcss |]\}}t|tr|VqdSr) isinstancer ).0nkrrr 2s z-PydapArrayWrapper._getitem..) getattrrr ValueErrortuple enumeratendimlennpZsqueeze)rrrresultZaxisrrrr,s  zPydapArrayWrapper._getitemN) __name__ __module__ __qualname__rpropertyrrr rrrrrrs rcstt|}t|D]^dks.dr@||qt|r|fdd|Dq|S)NglobalZ_globalcs i|]\}}d||qS).r)r"Zk_childZv_childr$rr Ds z#_fix_attributes..)dictlistlowerendswithupdatepopritems) attributesrr4r_fix_attributes9s    r>c@sFeZdZdZddZedddZddZd d Zd d Z d dZ dS)PydapDataStorezStore for accessing OpenDAP datasets with pydap. This store provides an alternative way to access OpenDAP datasets that may be useful if the netCDF4 library is not available. cCs ||_dS)zN Parameters ---------- ds : pydap DatasetType N)ds)rr@rrrrSszPydapDataStore.__init__Nc Csddl}ddl}|dkr(ddlm} | }||||p4d|d} t|jjtdkr||dk rf| d|i|dk r|| d|i|jjf| } || S)Nr)DEFAULT_TIMEOUTT)url applicationsession output_gridtimeoutz3.3.0verify user_charset) Z pydap.clientZ pydap.librArlib __version__r:clientZopen_url) clsrBrCrDrErFrGrHpydaprAkwargsr@rrropen[s$  zPydapDataStore.opencCs"tt|}t|j|t|jSr)r ZLazilyIndexedArrayrr dimensionsr>r=)rvardatarrropen_store_variable}sz"PydapDataStore.open_store_variablecstfddjDS)Nc3s"|]}|j|fVqdSr)rSr@)r"r$rrrr%sz/PydapDataStore.get_variables..)r r@keysrrrr get_variabless zPydapDataStore.get_variablescCstt|jjSr)r r>r@r=rrrr get_attrsszPydapDataStore.get_attrscCs t|jjSr)r r@rPrrrrget_dimensionsszPydapDataStore.get_dimensions)NNNNNN) r.r/r0__doc__r classmethodrOrSrUrVrWrrrrr?Ls !r?c @s2eZdZdZedZdZdZddZd d d Z dS) PydapBackendEntrypointa^ Backend for steaming datasets over the internet using the Data Access Protocol, also known as DODS or OPeNDAP based on the pydap package. This backend is selected by default for urls. For more information about the underlying library, visit: https://www.pydap.org See Also -------- backends.PydapDataStore rMz6Open remote datasets via OPeNDAP using pydap in XarrayzWhttps://docs.xarray.dev/en/stable/generated/xarray.backends.PydapBackendEntrypoint.htmlcCst|tot|Sr)r!strr)rfilename_or_objrrrguess_can_opensz%PydapBackendEntrypoint.guess_can_openTNc Cs\tj|| | | | | |d}t}t|,|j||||||||d}|W5QRSQRXdS)N)rBrCrDrErFrGrH)mask_and_scale decode_timesconcat_characters decode_coordsdrop_variables use_cftimedecode_timedelta)r?rOr r open_dataset)rr\r^r_r`rarbrcrdrCrDrErFrGrHstoreZstore_entrypointr@rrrres,   z#PydapBackendEntrypoint.open_dataset) TTTTNNNNNNNNN) r.r/r0rXr available descriptionrBr]rerrrrrZs&rZrM) __future__rZnumpyr,Zpackaging.versionrZxarray.backends.commonrrrrrZxarray.backends.storer Z xarray.corer Zxarray.core.pycompatr Zxarray.core.utilsr r rrrrZxarray.core.variablerrr>r?rZrrrrs       AB