+dY >ddlZddlmZmZmZGddeZdS)N) RemoteSourceSchema get_partitionc~eZdZdZdZdZfdZdZdZdZ dZ d Z d Z e dd Ze dd ZxZS) RemoteArrayznd-array on an Intake serverremote_ndarrayndarrayc tt|j||fi||d|_t |d|_|d|_|d|_t dt |ddD|_ d|_ t|j|j|j|j|j |_ dS) N npartitionsshapemetadatadtypec34K|]}t|VdSN)tuple).0cs 8lib/python3.11/site-packages/intake/container/ndarray.py z'RemoteArray.__init__..s(QQE!HHQQQQQQchunks))r extra_metadatarr r) superr__init__r rr rrgetrarrr_schema)selfurlheaderskwargs __class__s rrzRemoteArray.__init__s)k4  )#wAA&AAA!-06'?++ z* G_ QQeFJJx4O4O.P.PQQQQQ $*:4=`d`jrvr|FJFQRRR rcddlm}jddjzdjD}fdt j|D}||jjj _j S)Nrz remote-array-c3NK|] }tt|V!dSr)rangelen)rns rrz-RemoteArray._load_metadata.."s.99eCFFmm999999rc \i|](}f|ztjjjj|f)S)rr!r" _source_id container)rpartnamer s r z.RemoteArray._load_metadata..#sL[[[w{TGdN]DHdlDO]a]kmq$r[[[r)r/rrr ) dask.arrayarrayrr,r itertoolsproductArrayrr r)r danpartsdaskr/s` @r_load_metadatazRemoteArray._load_metadatas 8 i"T_4D99T[999F[[[[[@I@QSY@Z[[[Dxx4 4:]a]gxhhDH|rcr||jj|Sr)r9rblockscomputer is r_get_partitionzRemoteArray._get_partition(s/ xq!))+++rcT|||Sr)r9r?r=s rread_partitionzRemoteArray.read_partition,s( ""1%%%rc\||jSr)r9rr<r s rreadzRemoteArray.read0s( x!!!rc8||jSr)r9rrCs rto_daskzRemoteArray.to_dask4s xrcd|_dSr)rrCs r_closezRemoteArray._close8s rNc |}n$#t$r|}YnwxYwtj||fddd|S)aSave array to local persistent store Makes a parquet dataset out of the data using zarr. This then becomes a data entry in the persisted datasets catalog. Only works locally for the moment. Parameters ---------- source: a DataSource instance to save name: str or None Key to refer to this persisted dataset by. If not given, will attempt to get from the source's name kwargs: passed on to zarr array creation, see N componentstorage_options)rFNotImplementedErrorrDr_data_to_source)sourcepathrKrLr#rs r_persistzRemoteArray._persist;sh  ..""CC"   ++--CCC *3eVZee^deees 88c ddlm}m}ddlm}ddlm}||stt|ds ||d}|||fd|d |||||} | S) Nr) from_arrayto_zarr) is_arraylike)ZarrArraySourcer auto)rrJ) r1rSrT dask.utilsrU source.zarrrWrMhasattr) rrPrKrLr#rSrTrUrWrOs rrNzRemoteArray._data_to_sourceQs22222222++++++111111|C   &% %sM** 1*S000CTUT?UUfUUU  BB r)NN)__name__ __module__ __qualname____doc__r/r-rr9r?rArDrFrH staticmethodrQrN __classcell__)r$s@rrr s&& DIRRRRR   ,,,&&&"""fff\f*   \     rr)r3baserrrrr+rrrcso5555555555QQQQQ,QQQQQr