h2dZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z dZd Zd eeefd efd Zd ddddddddddddddZej'Dchc] \}}|dk(s |c}}Zd efdZd efdZdZdZd eeeffdZ ycc}}w)zUtility functions.N)Version)Path)Union)urlparse)vsimem_rmtree_toplevel)z .gpkg.zipz.shp.zipct|tr t|St|tr|St |dr?|j }t |dr!|j r|jd|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/immo/lib/python3.12/site-packages/pyogrio/util.pyget_vsi_path_or_bufferrs|0.$'''.%(~v&%**,  >: .>3J3J3L    " C' ((pathreturnclt|trNtjdk(r0|j j dr|j }n t |}|j dr|Stjdk(ritjd|rS|jddjds|S|jddjtr|Sd|}t|\}}}|s(|s&|jdr"|jts t|||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 startswithrrematchsplitendswithMULTI_EXTENSIONS _parse_uri_construct_vsi_path)rarchiveschemes rr r 9s$ <<7 "t}}'A'A&'I==?Dt9D v  ||w288ND#Azz#q!**62K ::c?1  & &'7 8Kv&t,D'6   MM& !$--8H*I"4&99 Krfileziptargzipcurls3gsazadlshdfswebhdfs)r)r*r+r,httphttpsftpr.r/r0r1adlr2r3ct|d}|jr0td|jjdDs|ddfS|j}|jxsd}|j r|d|j zz }|jr|j r|j |z}|jd}|r|jnd}|r|jnd}|||fS)aParse a URI. Returns a tuples of (path, archive, scheme) path : str Parsed path. Includes the hostname and query string in the case of a URI. archive : str Parsed archive path. scheme : str URI scheme such as "https" or "zip+s3". F)allow_fragmentsc3,K|] }|tvyw)NSCHEMES.0ps r z_parse_uri..sNW Ns+?r)rr(allr"rquerynetlocpop)rpartsr(r's rr%r%vs T5 1E ||CNell6H6H6MNNR| ::D \\ RF {{ ekk!! || ||d" JJsOE599;RD"eiikG '6 ""rcd}d}|jd}d|vrI|jds&|jdr'|jts|jdd|r+dj d|D}|dt vr|dd }|r/|r#d j ||||jdSd|d||S|S) z)Convert a parsed path to a GDAL VSI path.rBrAr*rr/c3FK|]}|s|dk7s dt|yw)r)vsiNr;r=s rr@z&_construct_vsi_path..s#SqF{C |,Ss !!!z://z /{}/{}{}/{})r"r#r$insertjoin CURLSCHEMESformatlstrip)rr'r(prefixsuffixschemess rr&r&s F Fll3G G MM& !$--8H*Iq% SgSS 2;+ % }C(F  ''SAQR Rvhaxv. . Krct|tstdt|i}|j D]?\}}| |j }t|t r|rdnd}n t|}|||<A|S)zhPreprocess options. For example, `spatial_index=True` gets converted to `SPATIAL_INDEX="YES"`. z#Expected options to be a dict, got ONOFF)r dict TypeErrortypeitemsupperboolr)optionsresultkvs r_preprocess_options_key_valuercs gt $=d7m_MNN F 1 9  GGI a uAAAq  Mrc||S ddl}t|jtdkrd}|s t dt ||j s t d|j|S#t$rd}YLwxYw)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)maskres r _mask_to_wkbrls& |  7&& '''*: :G CDD dG,, -FGG >>$  s'A,, A:9A:cZt|tr|j}t|y)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 rrrs!"$}}D!r)__doc__r rpackaging.versionrpathlibrtypingr urllib.parser pyogrio._vsirrnr$rrr r<r\rPr%r&rcrl)rarbs00rrus %!J,')T#5d###P              &%]]_