?HAMA

Edit File Size: 3.09 KB

/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/contracts/sentry.py

Text Base64

from pathlib import Path
from subprocess import DEVNULL, CalledProcessError, check_output
from typing import Any

from defence360agent.utils import stub_unexpected_error

def _run_cmd(cmd):
    try:
        out = check_output(cmd, stderr=DEVNULL)
    except (FileNotFoundError, CalledProcessError):
        return None

return out.decode("utf-8", errors="ignore").strip()

@stub_unexpected_error
def _get_virtualization_type():
    systemd_virt = _run_cmd(["systemd-detect-virt"])
    if systemd_virt:
        return systemd_virt

virt_what = _run_cmd(["virt-what"])
    if virt_what:
        return virt_what

demicode = _run_cmd(["dmidecode", "-s", "system-manufacturer"])
    if demicode:
        return demicode

return "fail to detect"

@stub_unexpected_error
def _get_total_ram():
    import psutil

return psutil.virtual_memory().total // 2**20

_TAGS = None

def _tags():
    global _TAGS
    if _TAGS is None:
        from defence360agent.utils import OsReleaseInfo

_TAGS = {
            "av_version": None,
            "core_version": None,
            "version": None,
            "os_details": stub_unexpected_error(OsReleaseInfo.pretty_name)(),
            "ip": None,
            "hosting_panel": None,
            "total_ram": _get_total_ram(),
            "firewall": None,
            "strategy": None,
            "virtualization": _get_virtualization_type(),
            "server_id": None,
            "iaid": None,
            "name": None,
            "test_build_id": None,
            "test_build_job_id": None,
            "test_parent_build_id": None,
        }
    return _TAGS

def set_firewall_type(firewall: str) -> None:
    _tags()["firewall"] = firewall

def set_hosting_panel(panel: str) -> None:
    _tags()["hosting_panel"] = panel

def set_strategy(strategy: str) -> None:
    _tags()["strategy"] = strategy

def set_ip(ip: str) -> None:
    _tags()["ip"] = ip

def set_product_name(product: str) -> None:
    _tags()["name"] = product

def set_server_id(id: str | None) -> None:
    _tags()["server_id"] = id

def set_iaid(iaid: str | None) -> None:
    _tags()["iaid"] = iaid

def set_version(version: str) -> None:
    _tags()["version"] = version

def set_av_version(version: str) -> None:
    _tags()["av_version"] = version

def set_core_version(version: str) -> None:
    _tags()["core_version"] = version

def tags() -> dict:
    return _tags().copy()

def tag(name: str) -> Any:
    return _tags()[name]

def set_test_env() -> None:
    """Set tags for sentry events about test environment."""
    for file_name, tag in [
        (
            Path("/var/imunify360/TEST_BUILD_ID"),
            "test_build_id",
        ),
        (
            Path("/var/imunify360/TEST_BUILD_JOB_ID"),
            "test_build_job_id",
        ),
        (
            Path("/var/imunify360/TEST_PARENT_BUILD_ID"),
            "test_parent_build_id",
        ),
    ]:
        if file_name.exists():
            try:
                _tags()[tag] = file_name.read_text().strip()
            except Exception:
                # Ignore errors on loading test env tags
                pass

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-06-08 20:24:30	Edit Download recursive recursive
config.py	46.66 KB	lrw-r--r--	2026-05-26 21:25:11	Edit Download recursive recursive
config_provider.py	13.74 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
eula.py	1.47 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
hooks.py	6.13 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
hook_events.py	1.51 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
license.py	24.26 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
messages.py	12.65 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
myimunify_id.py	5.79 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
permissions.py	6.46 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
plugins.py	8.13 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
sentry.py	3.09 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive

Directory Contents