rbidZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z dZd eezd efd Zd d dddddddddddddZdeDZd efdZd efdZdZdZd eezfdZ dS)zUtility functions.N)Version)Path)urlparse)MULTI_EXTENSIONS)vsimem_rmtree_toplevelcnt|trt|St|tr|St |drO|}t |dr)|r|d|Stt|S)aGet VSI-prefixed path or bytes buffer depending on type of path_or_buffer. If path_or_buffer is a bytes object, it will be returned directly and will be read into an in-memory dataset when passed to one of the Cython functions. If path_or_buffer is a file-like object with a read method, bytes will be read from the file-like object and returned. Otherwise, it will be converted to a string, and parsed to prefix with appropriate GDAL /vsi*/ prefixes. Parameters ---------- path_or_buffer : str, pathlib.Path, bytes, or file-like A dataset path or URI, raw buffer, or file-like object with a read method. Returns ------- str or bytes readseekabler) isinstancervsi_pathbyteshasattrr r seekstr)path_or_buffer bytes_buffers \/var/www/html/mdtn/previsions/meteo_cartes/venv/lib/python3.11/site-packages/pyogrio/util.pyget_vsi_path_or_bufferr s0.$''('''.%((~v&&%**,,  >: . . #>3J3J3L3L #    " " " C'' ( ((pathreturnct|tr[tjdkr<|dr|}nt |}|dr|Stjdkrtjd|rj| dd ds|S| dd tr|Sd|}t|\}}}|s1|s/| dr+| tst|||S|S)z:Ensure path is a local path or a GDAL-compatible VSI path.win32z/vsiz ^[a-zA-Z]\:!r.zipzzip://)r rsysplatformas_posix startswithrrematchsplitendswithr _parse_uri_construct_vsi_path)rarchiveschemes rr r 7sc$ <7 " "t}}'A'A&'I'I "==??DDt99D v  |w28ND#A#Azz#q!**622 K ::c??1  & &'7 8 8 K&t,,D'6 : : MM& ! !:+/--8H*I*I: #4&999 Krfileziptargzipcurls3gsazadlshdfswebhdfs)r(r)r*r+httphttpsftpr-r.r/r0adlr1r2c$h|] \}}|dk |S)r,).0kvs r r<qs!<<z_parse_uri..s&NNW NNNNNNr+?r)rr'allr"rquerynetlocpop)rpartsr'r&s rr$r$ts T5 1 1 1E |CNNel6H6H6M6MNNNNNR| :D \ RF {" ek!! |# #|d" JJsOOE '599;;;RD"*eiikkkG '6 ""rcd}d}|d}d|vrZ|ds/|dr0|ts|dd|r9dd|D}|dt vr |dd }|r7|r+d ||||dSd|d||S|S) z)Convert a parsed path to a GDAL VSI path.rFrEr)rr/c3FK|]}||dk dt|VdS)r(vsiNr@rBs rrDz&_construct_vsi_path..s8SSSqF{{, ,,{{{{SSrz://z /{}/{}{}/{})r"r#rinsertjoin CURLSCHEMESformatlstrip)rr&r'prefixsuffixschemess rr%r%s F Fll3G G   MM& ! !*.--8H*I*I q%   )SSgSSSSS 2;+ % % (((F /  / ''SAQAQRR R.v..... . Krc4t|tstdt|i}|D]L\}}||}t|t r|rdnd}nt|}|||<M|S)zhPreprocess options. For example, `spatial_index=True` gets converted to `SPATIAL_INDEX="YES"`. z#Expected options to be a dict, got NONOFF)r dict TypeErrortypeitemsupperboolr)optionsresultr:r;s r_preprocess_options_key_valueres gt $ $OMd7mmMMNNN F 1 9  GGII a   $uAAAAq Mrc||S ddl}t|jtdkrd}n#t$rd}YnwxYw|st dt ||jst d||S)aHConvert a Shapely mask geometry to WKB. Parameters ---------- mask : Shapely geometry The geometry to convert to WKB. Returns ------- WKB bytes or None Raises ------ ValueError raised if Shapely >= 2.0 is not available or mask is not a Shapely Geometry object Nrz2.0.0z('mask' parameter requires Shapely >= 2.0z+'mask' parameter must be a Shapely geometry)shapelyr __version__ ImportError ValueErrorr Geometryto_wkb)maskrgs r _mask_to_wkbrns& |  7& ' '''*:*: : :G  ECDDD dG, - -HFGGG >>$  s+2 AAcvt|tr|}t|dS)aRemove the parent directory of the file path recursively. This is used for final cleanup of an in-memory dataset, which may have been created within a directory to contain sibling files. Additional VSI handlers may be chained to the left of /vsimem/ in path and will be ignored. Remark: function is defined here to be able to run tests on it. Parameters ---------- path : str or pathlib.Path path to in-memory file N)r rr_vsimem_rmtree_toplevel)rs rrrs7"$}}D!!!!!r)__doc__r rpackaging.versionrpathlibr urllib.parser pyogrio._ogrr pyogrio._vsirrprrr rAr`rTr$r%rernr8rrrws %%%%%%!!!!!!))))))JJJJJJ')')')T#3:######P               &=