pyogrio 0.10.0__cp313-cp313-win_amd64.whl

pyogrio/__init__.py

@@ -0,0 +1,68 @@
+"""Vectorized vector I/O using OGR."""
+
+
+# start delvewheel patch
+def _delvewheel_patch_1_8_2():
+    import os
+    libs_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, 'pyogrio.libs'))
+    if os.path.isdir(libs_dir):
+        os.add_dll_directory(libs_dir)
+
+
+_delvewheel_patch_1_8_2()
+del _delvewheel_patch_1_8_2
+# end delvewheel patch
+
+try:
+    # we try importing shapely, to ensure it is imported (and it can load its
+    # own GEOS copy) before we load GDAL and its linked GEOS
+    import shapely
+    import shapely.geos  # noqa: F401
+except Exception:
+    pass
+
+from pyogrio._version import get_versions
+from pyogrio.core import (
+    __gdal_geos_version__,
+    __gdal_version__,
+    __gdal_version_string__,
+    detect_write_driver,
+    get_gdal_config_option,
+    get_gdal_data_path,
+    list_drivers,
+    list_layers,
+    read_bounds,
+    read_info,
+    set_gdal_config_options,
+    vsi_listtree,
+    vsi_rmtree,
+    vsi_unlink,
+)
+from pyogrio.geopandas import read_dataframe, write_dataframe
+from pyogrio.raw import open_arrow, read_arrow, write_arrow
+
+__version__ = get_versions()["version"]
+del get_versions
+
+__all__ = [
+    "list_drivers",
+    "detect_write_driver",
+    "list_layers",
+    "read_bounds",
+    "read_info",
+    "set_gdal_config_options",
+    "get_gdal_config_option",
+    "get_gdal_data_path",
+    "open_arrow",
+    "read_arrow",
+    "read_dataframe",
+    "vsi_listtree",
+    "vsi_rmtree",
+    "vsi_unlink",
+    "write_arrow",
+    "write_dataframe",
+    "__gdal_version__",
+    "__gdal_version_string__",
+    "__gdal_geos_version__",
+    "__version__",
+]

pyogrio/_compat.py

@@ -0,0 +1,47 @@
+from packaging.version import Version
+
+from pyogrio.core import __gdal_geos_version__, __gdal_version__
+
+# detect optional dependencies
+try:
+    import pyarrow
+except ImportError:
+    pyarrow = None
+
+try:
+    import pyproj
+except ImportError:
+    pyproj = None
+
+try:
+    import shapely
+except ImportError:
+    shapely = None
+
+try:
+    import geopandas
+except ImportError:
+    geopandas = None
+
+try:
+    import pandas
+except ImportError:
+    pandas = None
+
+
+HAS_ARROW_API = __gdal_version__ >= (3, 6, 0)
+HAS_ARROW_WRITE_API = __gdal_version__ >= (3, 8, 0)
+HAS_PYARROW = pyarrow is not None
+HAS_PYPROJ = pyproj is not None
+
+HAS_GEOPANDAS = geopandas is not None
+
+PANDAS_GE_15 = pandas is not None and Version(pandas.__version__) >= Version("1.5.0")
+PANDAS_GE_20 = pandas is not None and Version(pandas.__version__) >= Version("2.0.0")
+PANDAS_GE_22 = pandas is not None and Version(pandas.__version__) >= Version("2.2.0")
+
+GDAL_GE_38 = __gdal_version__ >= (3, 8, 0)
+
+HAS_GDAL_GEOS = __gdal_geos_version__ is not None
+
+HAS_SHAPELY = shapely is not None and Version(shapely.__version__) >= Version("2.0.0")

pyogrio/_env.py

@@ -0,0 +1,59 @@
+# With Python >= 3.8 on Windows directories in PATH are not automatically
+# searched for DLL dependencies and must be added manually with
+# os.add_dll_directory.
+# adapted from Fiona: https://github.com/Toblerity/Fiona/pull/875
+
+
+import logging
+import os
+import platform
+from contextlib import contextmanager
+from pathlib import Path
+
+log = logging.getLogger(__name__)
+log.addHandler(logging.NullHandler())
+
+
+try:
+    # set GDAL_CURL_CA_BUNDLE / PROJ_CURL_CA_BUNDLE for GDAL >= 3.2
+    import certifi
+
+    ca_bundle = certifi.where()
+    os.environ.setdefault("GDAL_CURL_CA_BUNDLE", ca_bundle)
+    os.environ.setdefault("PROJ_CURL_CA_BUNDLE", ca_bundle)
+except ImportError:
+    pass
+
+
+gdal_dll_dir = None
+
+if platform.system() == "Windows":
+    # if loading of extension modules fails, search for gdal dll directory
+    try:
+        import pyogrio._io  # noqa: F401
+
+    except ImportError:
+        for path in os.getenv("PATH", "").split(os.pathsep):
+            if list(Path(path).glob("gdal*.dll")):
+                log.info(f"Found GDAL at {path}")
+                gdal_dll_dir = path
+                break
+
+        if not gdal_dll_dir:
+            raise ImportError(
+                "GDAL DLL could not be found.  It must be on the system PATH."
+            )
+
+
+@contextmanager
+def GDALEnv():
+    dll_dir = None
+
+    if gdal_dll_dir:
+        dll_dir = os.add_dll_directory(gdal_dll_dir)
+
+    try:
+        yield None
+    finally:
+        if dll_dir is not None:
+            dll_dir.close()