+d~ddlmZddlmZddlmZddlmZGddeZddddd d Z Gd d eZ d S))from_uri)Node)Catalog) DataSourcec6eZdZdZdZdfd Zd dZdZxZS) TiledCatalogaGView Tiled server as a catalog See the documentation for setting up such a server at https://blueskyproject.io/tiled/ A tiled server may contain sources of dataframe, array or xarray type. This driver exposes the full tree as exposed by the server, but you can also specify the sub-path of that tree. tiled_catNct||_t|tr@|dr|ddd}n|}t |d}n |}|j}||_|||}t|d| dddzdS) a Parameters ---------- server: str or tiled.client.node.Node Location of tiles server. Usually of the form "http[s]://address:port/" May include a path. If the protocol is "tiled", we assume HTTP connection. Alternatively, can be a Node instance, already connected to a server. path: str (optional) If given, restrict the catalog to this part of the server's catalog tree. Equivalent to extending the server URL. tiledhttpdaskNztiled::)entriesname) path isinstancestr startswithreplacerurisuper__init__split)selfserverrrclient __class__s 3lib/python3.11/site-packages/intake/source/tiled.pyrzTiledCatalog.__init__s fc " "   )) nnWfa88c6**FFF*C  "D\F h39J9J19M.MNNNNNtextc|dkrddlm}||}ntt|j||j|jS)zSFull text search Queries other than full text will be added later r!r)FullText)rr) tiled.queriesr#NotImplementedErrorr from_dict_entriessearchrr)rquerytyper#qs rr(zTiledCatalog.search2sh 6> & . . . . . .AA% %%%dm&:&:1&=&=48RVR[%\\\r c|j|}t|trt|St |j||S)N)rrinstance)r'rrr TiledSourcer)ritemnodes r __getitem__zTiledCatalog.__getitem__?sF}T" dD ! ! G%% %48$FFF Fr N)r!) __name__ __module__ __qualname____doc__rrr(r1 __classcell__rs@rrrs DOOOOOO: ] ] ] ]GGGGGGGr rndarrayxarray dataframe)DaskArrayClientDaskDataArrayClientDaskDatasetClientDaskVariableClientDaskDataFrameClientcDeZdZdZdZd fd ZdZdZdZfd Z xZ S) r.zA source on a Tiled server The container type of this source is determined at runtime. The attribute ``.instance`` gives access to the underlying Tiled API, but most users will only call ``.to_dask()``. r Ncd|(t|d|}||_t|j}|r||t |||_tt|jj |_ dS)aq Parameters ---------- uri: str (optional) Location of the server. If ``instance`` is given, this is only used for the repr pathL str (optional) Path of the data source within the server tree. If ``instance`` is given, this is only used for the repr instance: tiled.client.node.None (optional) The tiled object pointing to the data source; normally created by a ``TiledCatalog`` metadata: dict Extra metadata for this source; metadata will also be provided by the server. Nr)metadata) rreadr-dictrDupdaterrrtypesr*r3 container)rrrr-rDmdrs rrzTiledSource.__init__Zs$  :V,,T27799H  (# $ $  IIh    "%%% tDM22;<r c8|}t|ddpt|dd}t|ddp |jj}t |t|jjd|j||j S)Ndtypedtypes npartitionsshape)rLrOrNrD) to_daskgetattrdatarNrFr- structuremacrorOrD)rxdtpartss rdiscoverzTiledSource.discovervs LLNN Q & & D'!Xt*D*D=$//E163E"GDM,C,C,E,E,KWVWV]$^$^lq}A}JKKK Kr c4|jSr2)r-rErs rrPzTiledSource.to_dask|s}!!###r cX|jSr2)r-rEcomputerZs rrEzTiledSource.reads"}!!##++---r ct}t|dd}|dd|S)Nsourcesrargsr-)r_yamllistvaluespop)ryvrs rr`zTiledSource._yamlsQ GGMMOO 9$$&& ' ' * & j!!!r )rBrBNN) r3r4r5r6rrrXrPrEr`r7r8s@rr.r.Ps D======8KKK $$$...r r.N) tiled.clientrtiled.client.noderintake.catalogr intake.sourcerrrHr.r rrks!!!!!!""""""""""""$$$$$$<G<G<G<G<G7<G<G<G@!#!"&   77777*77777r