sf-veritas 0.10.3__cp313-cp313-manylinux_2_28_x86_64.whl

sf_veritas/import_hook.py

@@ -0,0 +1,62 @@
+import importlib
+import importlib.util
+import sys
+from importlib.abc import MetaPathFinder
+
+from .custom_excepthook import custom_excepthook
+from .custom_output_wrapper import get_custom_output_wrapper_django
+from .env_vars import PRINT_CONFIGURATION_STATUSES
+
+
+class ImportInterceptor:
+    def __init__(self, module_name, callback):
+        self.module_name = module_name
+        self.callback = callback
+
+    def find_spec(self, fullname, path, target=None):
+        if fullname == self.module_name:
+            self.callback()
+            # Remove the interceptor once the callback has been called
+            sys.meta_path.remove(self)
+            return importlib.util.find_spec(fullname, path)
+        return None
+
+
+def setup_django_import_hook():
+    interceptor = ImportInterceptor(
+        "django.core.management", get_custom_output_wrapper_django
+    )
+    sys.meta_path.insert(0, interceptor)
+
+
+def wrap_excepthook(old_excepthook):
+    def custom_wrapper_excepthook(exc_type, exc_value, exc_traceback):
+        # Call your custom exception hook
+        custom_excepthook(exc_type, exc_value, exc_traceback)
+        # Call the original exception hook (which might be Sentry's)
+        if old_excepthook is not None:
+            old_excepthook(exc_type, exc_value, exc_traceback)
+
+    return custom_wrapper_excepthook
+
+
+def set_excepthook():
+    # Wrap the current excepthook
+    current_excepthook = sys.excepthook
+    wrapped_excepthook = wrap_excepthook(current_excepthook)
+    sys.excepthook = wrapped_excepthook
+    if PRINT_CONFIGURATION_STATUSES:
+        print("Patched sys.excepthook for custom exception handling.")
+
+
+class ExcepthookImportHook(MetaPathFinder):
+    def find_spec(self, fullname, path, target=None):
+        print("fullname", fullname, log=False)
+        if fullname in sys.modules:
+            print("\t> fullname in sys.modules", log=False)
+            set_excepthook()
+        return None
+
+
+def install_import_hook():
+    sys.meta_path.insert(0, ExcepthookImportHook())

sf_veritas/infra_details/__init__.py

@@ -0,0 +1,3 @@
+from .get_infra_details import get_infra_details
+
+__all__ = ["get_infra_details"]

sf_veritas/infra_details/get_infra_details.py

@@ -0,0 +1,24 @@
+from dataclasses import dataclass
+from typing import Any, Dict
+
+from .kubernetes import get_details as get_kubernetes_details
+from .running_on import System, running_on
+
+
+@dataclass
+class InfraDetails:
+    system: System
+    details: Dict[str, Any]
+
+
+def get_running_on_details(system: System) -> Dict[str, Any]:
+    if system == System.KUBERNETES:
+        return get_kubernetes_details()
+    if system == System.UNKNOWN:
+        return {}
+
+
+def get_infra_details() -> InfraDetails:
+    system = running_on()
+    details = get_running_on_details(system)
+    return InfraDetails(system=system, details=details)

sf_veritas/infra_details/kubernetes/__init__.py

@@ -0,0 +1,3 @@
+from .get_details import get_details
+
+__all__ = ["get_details"]

sf_veritas/infra_details/kubernetes/get_cluster_name.py

@@ -0,0 +1,147 @@
+import os
+import socket
+
+import requests
+
+DEFAULT_CLUSTER_NAME = "UNKNOWN"
+
+# ─── 1. ConfigMap (optional) ─────────────────────────────────
+
+
+def get_from_config_map(path: str = "/etc/cluster-info/cluster-name"):
+    """If you’ve mounted a ConfigMap at this path, read it."""
+    try:
+        with open(path, "r") as f:
+            name = f.read().strip()
+            if name:
+                return name
+    except IOError:
+        pass
+
+
+# ─── 2. Cloud Metadata ────────────────────────────────────────
+
+
+def get_gke_cluster_name(timeout: float = 1.0):
+    """GKE nodes automatically get a 'cluster-name' instance attribute."""
+    try:
+        resp = requests.get(
+            "http://metadata.google.internal/computeMetadata/v1/instance/attributes/cluster-name",
+            headers={"Metadata-Flavor": "Google"},
+            timeout=timeout,
+        )
+        if resp.ok and resp.text:
+            return resp.text
+    except requests.RequestException:
+        pass
+
+
+def get_eks_cluster_name(timeout: float = 1.0):
+    """EKS-backed EC2 instances are tagged 'eks:cluster-name'."""
+    try:
+        # 1) fetch IMDSv2 token
+        token = requests.put(
+            "http://169.254.169.254/latest/api/token",
+            headers={"X-aws-ec2-metadata-token-ttl-seconds": "21600"},
+            timeout=timeout,
+        ).text
+        # 2) read the eks:cluster-name tag
+        resp = requests.get(
+            "http://169.254.169.254/latest/meta-data/tags/instance/eks:cluster-name",
+            headers={"X-aws-ec2-metadata-token": token},
+            timeout=timeout,
+        )
+        if resp.ok and resp.text:
+            return resp.text
+    except requests.RequestException:
+        pass
+
+
+def get_aks_cluster_name(timeout: float = 1.0):
+    """AKS nodes live in a VM RG named MC_<resourceGroup>_<clusterName>_<zone>."""
+    try:
+        resp = requests.get(
+            "http://169.254.169.254/metadata/instance",
+            params={"api-version": "2021-02-01", "format": "json"},
+            headers={"Metadata": "true"},
+            timeout=timeout,
+        )
+        if resp.ok:
+            compute = resp.json().get("compute", {})
+            rg = compute.get("resourceGroupName", "")
+            parts = rg.split("_")
+            if len(parts) >= 3:
+                return parts[2]
+    except requests.RequestException:
+        pass
+
+
+# ─── 3. Kubernetes API fallback ────────────────────────────────
+
+
+def get_via_k8s_api(timeout: float = 1.0):
+    """
+    If you’re running in K8s and have in‑cluster RBAC, try:
+      A) ClusterProperty CRD (KEP‑2149)
+      B) Node labels on your own Pod’s node
+    """
+    try:
+        from kubernetes import client, config
+    except ImportError:
+        return
+
+    try:
+        # load service account creds
+        config.load_incluster_config()
+
+        # A) ClusterProperty CRD
+        co = client.CustomObjectsApi()
+        props = co.list_cluster_custom_object(
+            group="multicluster.k8s.io",
+            version="v1alpha6",
+            plural="clusterproperties",
+        ).get("items", [])
+        if props:
+            name = props[0].get("spec", {}).get("clusterName")
+            if name:
+                return name
+
+        # B) read this Pod → its Node → cluster label
+        v1 = client.CoreV1Api()
+        # Pod name = hostname in K8s
+        pod_name = socket.gethostname()
+        ns = (
+            open("/var/run/secrets/kubernetes.io/serviceaccount/namespace")
+            .read()
+            .strip()
+        )
+        pod = v1.read_namespaced_pod(
+            name=pod_name, namespace=ns, _request_timeout=timeout
+        )
+        node = v1.read_node(pod.spec.node_name, _request_timeout=timeout)
+        labels = node.metadata.labels or {}
+        for key in ("cluster.x-k8s.io/cluster-name", "topology.kubernetes.io/cluster"):
+            if labels.get(key):
+                return labels[key]
+    except Exception:
+        pass
+
+
+# ─── 4. Aggregator ────────────────────────────────────────────
+
+
+def get_cluster_name():
+    for fn in (
+        get_from_config_map,
+        get_gke_cluster_name,
+        get_eks_cluster_name,
+        get_aks_cluster_name,
+        get_via_k8s_api,
+    ):
+        try:
+            name = fn()
+            if name:
+                return name
+        except Exception:
+            continue
+    return DEFAULT_CLUSTER_NAME

sf_veritas/infra_details/kubernetes/get_details.py

@@ -0,0 +1,7 @@
+from typing import Any, Dict
+
+from .get_cluster_name import get_cluster_name
+
+
+def get_details() -> Dict[str, Any]:
+    return {"clusterName": get_cluster_name()}

sf_veritas/infra_details/running_on/__init__.py

@@ -0,0 +1,17 @@
+from enum import Enum
+
+from .kubernetes import kubernetes
+
+
+class System(Enum):
+    KUBERNETES = "Kubernetes"
+    UNKNOWN = "Unknown"
+
+
+def running_on() -> System:
+    if kubernetes():
+        return System.KUBERNETES
+    return System.UNKNOWN
+
+
+__all__ = ["running_on"]

sf_veritas/infra_details/running_on/kubernetes.py

@@ -0,0 +1,11 @@
+import os
+
+
+def kubernetes() -> bool:
+    # 1) service‐account token (default mount in every Pod)
+    if os.path.exists("/var/run/secrets/kubernetes.io/serviceaccount/token"):
+        return True
+    # 2) built‐in K8s env var
+    if os.getenv("KUBERNETES_SERVICE_HOST"):
+        return True
+    return False