Source code for htr2hpc.context_processors

from os import cpu_count, getloadavg
from socket import gethostname

import psutil
from django.contrib.sites.shortcuts import get_current_site
from django.template.defaultfilters import filesizeformat

# Get host and CPU count once on load, since changing these requires a reboot

#: number of logical CPUs in the system
CPU_COUNT = cpu_count()
#: hostname for the local VM
HOSTNAME = gethostname()

# os.uname potentially useful information (release, version, hardware)




def vm_status(request):
    """Custom context processor to return information about VM
    configuration and resources."""

    # load average returns average load over last 1, 5, and 15 minutes
    load_average = getloadavg()
    # system memory usage: total, available, percent, etc
    system_memory = psutil.virtual_memory()

    return {
        "vm_status": {
            # use to track if we're switching between VMs or if nginx+ is sticky
            "hostname": HOSTNAME,
            "cpu_count": CPU_COUNT,
            "load_average": {
                "1": load_average[0],
                "5": load_average[1],
                "15": load_average[2],
            },
            "total_memory": filesizeformat(system_memory.total),
            "available_memory": filesizeformat(system_memory.available),
            "used_memory": system_memory.percent,
        },
        "system_memory": system_memory,
    }





def htr2hpc_version(request):
    """Custom context processor to expose the htr2hpc package version."""
    from htr2hpc import __version__

    return {"HTR2HPC_VERSION": __version__}





def site_domain(request):
    """Custom context processor to expose the current site from the
    Django sites framework. Used to display environment-specific values
    (e.g. SSH key labels) without hardcoding hostnames in templates."""
    return {"current_site": get_current_site(request)}