polars-runtime-compat 1.34.0b2__cp39-abi3-macosx_11_0_arm64.whl

polars/_plr.py

@@ -0,0 +1,99 @@
+# This module represents the Rust API functions exposed to Python through PyO3. We do a
+# bit of trickery here to allow overwriting it with other function pointers.
+
+import builtins
+import os
+import sys
+
+from polars._cpu_check import check_cpu_flags
+
+PKG_VERSION = "1.34.0-beta.2"
+
+# Replaced during the build process with our list of required feature flags
+# enabled at compile time.
+RT_COMPAT_FEATURE_FLAGS = ""
+RT_NONCOMPAT_FEATURE_FLAGS = ""
+
+
+def rt_compat() -> None:
+    check_cpu_flags(RT_COMPAT_FEATURE_FLAGS)
+
+    import _polars_runtime_compat._polars_runtime_compat as plr
+
+    sys.modules[__name__] = plr
+
+
+def rt_64() -> None:
+    check_cpu_flags(RT_NONCOMPAT_FEATURE_FLAGS)
+
+    import _polars_runtime_64._polars_runtime_64 as plr
+
+    sys.modules[__name__] = plr
+
+
+def rt_32() -> None:
+    check_cpu_flags(RT_NONCOMPAT_FEATURE_FLAGS)
+
+    import _polars_runtime_32._polars_runtime_32 as plr
+
+    sys.modules[__name__] = plr
+
+
+if hasattr(builtins, "__POLARS_PLR"):
+    sys.modules[__name__] = builtins.__POLARS_PLR
+else:
+    # Each of the Polars variants registers a `_polars...` package that we can import
+    # the PLR from.
+
+    _force = os.environ.get("POLARS_FORCE_PKG")
+    _prefer = os.environ.get("POLARS_PREFER_PKG")
+
+    pkgs = {"compat": rt_compat, "64": rt_64, "32": rt_32}
+
+    if _force is not None:
+        try:
+            pkgs[_force]()
+
+            if sys.modules[__name__].__version__ != PKG_VERSION:
+                msg = f"Polars Rust module for '{_force}' ({sys.modules[__name__].__version__}) did not match version of Python package '{PKG_VERSION}'"
+                raise ImportError(msg)
+        except KeyError:
+            msg = f"Invalid value for `POLARS_FORCE_PKG` variable: '{_force}'"
+            raise ValueError(msg) from None
+    else:
+        preference = list(pkgs.values())
+        if _prefer is not None:
+            try:
+                preference.insert(0, pkgs[_prefer])
+            except KeyError:
+                msg = f"Invalid value for `POLARS_PREFER_PKG` variable: '{_prefer}'"
+                raise ValueError(msg) from None
+
+        version_warnings = []
+        for pkg in preference:
+            try:
+                pkg()
+
+                if sys.modules[__name__].__version__ != PKG_VERSION:
+                    import warnings
+
+                    version_warnings += [sys.modules[__name__].__version__]
+                    warnings.warn(
+                        f"Skipping Polars' Rust module version '{sys.modules[__name__].__version__}' did not match version of Python package '{PKG_VERSION}'.",
+                        ImportWarning,
+                        stacklevel=2,
+                    )
+                    continue
+
+                break
+            except ImportError:
+                pass
+        else:
+            msg = "could not find Polars' Rust module"
+            if len(version_warnings) > 0:
+                msg += f". Skipped versions {version_warnings} which don't match Python package version"
+            raise ImportError(msg)
+
+
+# The version at the top here should match the version specified by the PLR.
+assert sys.modules[__name__].__version__ == PKG_VERSION