Add DPM (Dynamic Power Management) feature for Proxmox cluster nodes

Fixes: #141

.changelogs/1.1.10/335-prevalidate-affinity-matrix.yml

@@ -1,2 +0,0 @@
-added:
-  - Prevent redundant rebalancing by validating existing affinity enforcement before taking actions (@gyptazy). [#335]

.changelogs/1.1.10/359-add-pve8-user-protections-for-conntrack-aware-migrations.yml

@@ -1,2 +0,0 @@
-added:
-  - Add safety-guard for PVE 8 users when activating conntrack-aware migrations mistakenly (@gyptazy). [#359]

.changelogs/1.1.10/361_fix_proxmox_api_connection_validation.yml

@@ -1,3 +0,0 @@
-fixed:
-  - Fixed the Proxmox API connection validation which returned a false-positive logging message of timeouts (@gyptazy). [#361]
-  - Refactored Proxmox API connection functions

.changelogs/1.1.10/368_fix_crash_enumerating_pools_with_storage_members.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fixed a crash during PVE resource pool enumeration by skipping members not having a 'name' property (@stefanoettl). [#368]

.changelogs/1.1.10/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-11-25

.changelogs/1.1.2/222_fix_force_type_cast_cpu_metrics_to_int.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Force type cast cpu count of guests to int for some corner cases where a str got returned (by @gyptazy). [#222]

.changelogs/1.1.2/release_meta.yml

@@ -1 +1 @@
-date: 2025-05-13
+date: TBD

.changelogs/1.1.3/189_add_reload_function.yml

@@ -1,2 +0,0 @@
-added:
-  - Add relaod (SIGHUP) function to ProxLB to reload the configuration (by @gyptazy). [#189]

.changelogs/1.1.3/232_align_maintenance_mode_proxmox_ha.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Align maintenance mode with Proxmox HA maintenance mode (by @gyptazy). [#232]

.changelogs/1.1.3/239_add_optional_delay_time_until_service_starts.yml

@@ -1,2 +0,0 @@
-added:
-  - Add optional wait time parameter to delay execution until the service takes action (by @gyptazy). #239

.changelogs/1.1.3/241_make_amount_of_parallel_migrations_configurable.yml

@@ -1,2 +0,0 @@
-added:
-  - Make the amount of parallel migrations configurable (by @gyptazy). [#241]

.changelogs/1.1.3/94_cpu_balancing_use_avg_instead_current_consumption.yml

@@ -1,2 +0,0 @@
-changed:
-  - Use the average CPU consumption of a guest within the last 60 minutes instead of the current CPU usage (by @philslab-ninja & @gyptazy). [#94]

.changelogs/1.1.3/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-06-19

.changelogs/1.1.4/245_add_guest_pinning_to_groups_of_nodes.yml

@@ -1,2 +0,0 @@
-added:
-  - Allow pinning of guests to a group of nodes (@gyptazy). [#245]

.changelogs/1.1.4/248_fix_dry_run_combination_balancing_disabled.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fixed an issue where balancing was performed in combination of deactivated balancing and dry-run mode (@gyptazy). [#248]

.changelogs/1.1.4/255_fix_loglevels.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Modified log levels to make output lighter at INFO level (@pmarasse) [#255]

.changelogs/1.1.4/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-06-27

.changelogs/1.1.5/260_allow_custom_api_ports.yml

@@ -1,2 +0,0 @@
-added:
-  - Allow custom API ports instead of fixed tcp/8006 (@gyptazy). [#260]

.changelogs/1.1.5/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-07-14

.changelogs/1.1.6/268_fix_balancing_type_eval.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fix balancing evaluation of guest types (e.g., VM or CT) (@gyptazy). [#268]

.changelogs/1.1.6/290_validate_user_token_syntax.yml

@@ -1,2 +0,0 @@
-added:
-  - Add validation for provided API user token id to avoid confusions (@gyptazy). [#291]

.changelogs/1.1.6/291_fix_stack_trace_no_user_permissions_testing.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fix stacktrace output when validating permissions on non existing users in Proxmox (@gyptazy). [#291]

.changelogs/1.1.6/295_fix_overprovisioning_first_node_if_anti_affinity_group_has_only_one-member.yml

@@ -1,3 +0,0 @@
-fixed:
-  - Fix Overprovisioning first node if anti_affinity_group has only one member (@MiBUl-eu). [#295]
-  

.changelogs/1.1.6/296_fix_validate_node_presence_when_pinning_guests.yml

@@ -1,3 +0,0 @@
-fixed:
-  - Validate for node presence when pinning guests to avoid crashing (@gyptazy). [#296]
-  

.changelogs/1.1.6/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-09-04

.changelogs/1.1.7/304_add_graceful_shutdown_sigint.yml

@@ -1,2 +0,0 @@
-added:
-  - Add graceful shutdown for SIGINT (e.g., CTRL + C abort). (@gyptazy). [#304]

.changelogs/1.1.7/305_add_conntrack_support_for_migrations.yml

@@ -1,2 +0,0 @@
-added:
-  - Add conntrack state aware migrations of VMs (@gyptazy). [#305]

.changelogs/1.1.7/308_fix_only_validate_valid_jobids.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fix crash when validating absent migration job ids. (@gyptazy). [#308]

.changelogs/1.1.7/310_guest-object-names-not-being-evaluated-in-log.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fix guest object names are not being evaluated in debug log. (@gyptazy). [#310]

.changelogs/1.1.7/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-09-19

.changelogs/1.1.8/317_container_image_non_root.yml

@@ -1,3 +0,0 @@
-changed:
-  - Container image does not run as root anymore (@mikaelkrantz945). [#317]
-  - Container image uses venv for running ProxLB (@mikaelkrantz945). [#317]

.changelogs/1.1.8/318_fix_conntrack_aware_migrations_api_pve8.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Fix API errors when using conntrack aware migration with older PVE versions (@gyptazy). [#318]

.changelogs/1.1.8/329_add_log_prefix.yml

@@ -1,2 +0,0 @@
-fixed:
-  - Add a static ProxLB prefix to the log output when used by journal handler (@gyptazy). [#329]

.changelogs/1.1.8/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-10-09

.changelogs/1.1.9/337_add_pressure_based_balancing.yml

@@ -1,5 +0,0 @@
-added:
-  - Add pressure (PSI) based balancing for memory, cpu, disk (req. PVE9 or greater) (@gyptazy). [#337]
-  - Pressure (PSI) based balancing for nodes
-  - Pressure (PSI) based balancing for guests
-  - Add PVE version evaluation

.changelogs/1.1.9/342_add_memory_balancing_threshold.yml

@@ -1,2 +0,0 @@
-added:
-  - Add an optional memory balancing threshold (@gyptazy). [#342]

.changelogs/1.1.9/343_add_affinity_rules_support_by_pools.yml

@@ -1,2 +0,0 @@
-added:
-  - Add affinity/anti-affinity support by pools (@gyptazy). [#343]

.changelogs/1.1.9/release_meta.yml

@@ -1 +0,0 @@
-date: 2025-10-30

.changelogs/1.2.0/141_add_power_management_for_cluster_nodes.yml

@@ -0,0 +1,2 @@
+added:
+  - Add power management feature for cluster nodes (by @gyptazy) [#141]

.changelogs/1.2.0/release_meta.yml

@@ -0,0 +1 @@
+date: TBD

.github/workflows/20-pipeline-build-deb-package.yml

@@ -58,10 +58,6 @@ jobs:
   integration-test-debian:
     needs: build-package-debian
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        debian_version: [bookworm, trixie]
-    name: Integration Test on Debian ${{ matrix.debian_version }}
     steps:
       - name: Download Debian package artifact
         uses: actions/download-artifact@v4
@@ -70,18 +66,13 @@ jobs:
           path: package/
 
       - name: Set up Docker with Debian image
-        run: docker pull debian:${{ matrix.debian_version }}
+        run: docker pull debian:latest
 
       - name: Install and test Debian package in Docker container
         run: |
-          docker run --rm \
-            -v "$(pwd)/package:/package" \
-            -w /package \
-            debian:${{ matrix.debian_version }} \
-            bash -c "
-              set -e
-              apt-get update
-              apt-get install -y python3 systemd
-              apt-get install -y ./proxlb*.deb
-              python3 -c 'import proxlb; print(\"OK: Debian package successfully installed on ${{ matrix.debian_version }}.\")'
-            "
+          docker run --rm -v $(pwd)/package:/package -w /package debian:latest bash -c "
+            apt-get update && \
+            apt-get install -y systemd && \
+            apt-get install -y ./proxlb*.deb && \
+            python3 -c 'import proxlb; print(\"OK: Debian package successfully installed.\")'
+          "

.github/workflows/30-pipeline-build-container-amd64.yml

@@ -1,26 +0,0 @@
-name: "Build Container Image: AMD64"
-on: [push]
-jobs:
-  build:
-    runs-on: ubuntu-latest
-    steps:
-    - name: Checkout repository
-      uses: actions/checkout@v4
-    - name: Set up QEMU
-      uses: docker/setup-qemu-action@v3
-    - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
-    - name: Build amd64 image and save as tar
-      run: |
-        docker buildx build \
-          --platform linux/amd64 \
-          --load \
-          -t proxlb-image:amd64 \
-          .
-
-        docker save proxlb-image:amd64 -o proxlb_image_amd64.tar
-    - name: Upload Docker image artifact
-      uses: actions/upload-artifact@v4
-      with:
-        name: proxlb-image-amd64
-        path: proxlb_image_amd64.tar

.github/workflows/30-pipeline-build-container-arm64.yml

@@ -1,26 +0,0 @@
-name: "Build Container Image: ARM64"
-on: [push]
-jobs:
-  build:
-    runs-on: ubuntu-latest
-    steps:
-    - name: Checkout repository
-      uses: actions/checkout@v4
-    - name: Set up QEMU
-      uses: docker/setup-qemu-action@v3
-    - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
-    - name: Build arm64 image and save as tar
-      run: |
-        docker buildx build \
-          --platform linux/arm64 \
-          --load \
-          -t proxlb-image:arm64 \
-          .
-
-        docker save proxlb-image:arm64 -o proxlb_image_arm64.tar
-    - name: Upload Docker image artifact
-      uses: actions/upload-artifact@v4
-      with:
-        name: proxlb-image-arm64
-        path: proxlb_image_arm64.tar

.github/workflows/30-pipeline-build-container-multi-arch.yml

@@ -1,23 +0,0 @@
-name: "Build Container Image: Multiarch"
-on: [push]
-jobs:
-  build:
-    runs-on: ubuntu-latest
-    steps:
-    - name: Checkout repository
-      uses: actions/checkout@v4
-    - name: Set up QEMU
-      uses: docker/setup-qemu-action@v3
-    - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
-    - name: Build multi-arch image and save as tar
-      run: |
-        docker buildx build \
-          --platform linux/amd64,linux/arm64 \
-          --output type=tar,dest=proxlb_image_multiarch.tar \
-          .
-    - name: Upload Docker image artifact
-      uses: actions/upload-artifact@v4
-      with:
-        name: proxlb-image-multiarch
-        path: proxlb_image_multiarch.tar

CHANGELOG.md

@@ -5,127 +5,6 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [1.1.10] - 2025-11-25
-
-### Added
-
-- Prevent redundant rebalancing by validating existing affinity enforcement before taking actions (@gyptazy). [#335]
-- Add safety-guard for PVE 8 users when activating conntrack-aware migrations mistakenly (@gyptazy). [#359]
-
-### Fixed
-
-- Fix the Proxmox API connection validation which returned a false-positive logging message of timeouts (@gyptazy). [#361]
-- Refactored Proxmox API connection functions (@gyptazy). [#361]
-- Fix a crash during PVE resource pool enumeration by skipping members not having a 'name' property (@stefanoettl). [#368]
-
-## [1.1.9.1] - 2025-10-30
-
-### Fixed
-
-- Fix quoting in f-strings which may cause issues on PVE 8 / Debian Bookworm systems (@gyptazy). [#352]
-
-## [1.1.9] - 2025-10-30
-
-### Added
-
-- Add an optional memory balancing threshold (@gyptazy). [#342]
-- Add affinity/anti-affinity support by pools (@gyptazy). [#343]
-- Add pressure (PSI) based balancing for memory, cpu, disk (req. PVE9 or greater) (@gyptazy). [#337]
-    - Pressure (PSI) based balancing for nodes
-    - Pressure (PSI) based balancing for guests
-- Add PVE version evaluation
-
-## [1.1.8] - 2025-10-09
-
-### Fixed
-
-- Fix API errors when using conntrack aware migration with older PVE versions (@gyptazy). [#318]
-- Add a static ProxLB prefix to the log output when used by journal handler (@gyptazy). [#329]
-
-### Changed
-- Container image does not run as root anymore (@mikaelkrantz945). [#317]
-- Container image uses venv for running ProxLB (@mikaelkrantz945). [#317]
-
-## [1.1.7] - 2025-09-19
-
-### Added
-
-- Add conntrack state aware migrations of VMs (@gyptazy). [#305]
-- Add graceful shutdown for SIGINT (e.g., CTRL + C abort). (@gyptazy). [#304]
-
-### Fixed
-
-- Fix crash when validating absent migration job ids. (@gyptazy). [#308]
-- Fix guest object names are not being evaluated in debug log. (@gyptazy). [#310]
-
-## [1.1.6.1] - 2025-09-04
-
-### Fixed
-
-- Validate for node presence when pinning VMs to avoid crashing (@gyptazy). [#296]
-
-## [1.1.6] - 2025-09-04
-
-### Added
-
-- Add validation for provided API user token id to avoid confusions (@gyptazy). [#291]
-
-### Fixed
-
-- Fix stacktrace output when validating permissions on non existing users in Proxmox (@gyptazy). [#291]
-- Fix Overprovisioning first node if anti_affinity_group has only one member (@MiBUl-eu). [#295]
-- Validate for node presence when pinning guests to avoid crashing (@gyptazy). [#296]
-- Fix balancing evaluation of guest types (e.g., VM or CT) (@gyptazy). [#268]
-
-## [1.1.5] - 2025-07-14
-
-### Added
-
-- Allow custom API ports instead of fixed tcp/8006 (@gyptazy). [#260]
-
-
-## [1.1.4] - 2025-06-27
-
-### Added
-
-- Allow pinning of guests to a group of nodes (@gyptazy). [#245]
-
-### Fixed
-
-- Modified log levels to make output lighter at INFO level (@pmarasse) [#255]
-- Fixed an issue where balancing was performed in combination of deactivated balancing and dry-run mode (@gyptazy). [#248]
-
-
-## [1.1.3] - 2025-06-19
-
-### Added
-
-- Add relaod (SIGHUP) function to ProxLB to reload the configuration (by @gyptazy). [#189]
-- Add optional wait time parameter to delay execution until the service takes action (by @gyptazy). [#239]
-- Make the amount of parallel migrations configurable (by @gyptazy). [#241]
-
-### Changed
-
-- Use the average CPU consumption of a guest within the last 60 minutes instead of the current CPU usage (by @philslab-ninja & @gyptazy). [#94]
-
-### Fixed
-
-- Align maintenance mode with Proxmox HA maintenance mode (by @gyptazy). [#232]
-
-
-## [1.1.2] - 2025-05-13
-
-### Added
-
-- Add a configurable retry mechanism when connecting to the Proxmox API (by @gyptazy) [#157]
-- Add 1-to-1 relationships between guest and hypervisor node to ping a guest on a node (by @gyptazy) [#218]
-
-### Fixed
-
-- Force type cast cpu count of guests to int for some corner cases where a str got returned (by @gyptazy). [#222]
-- Fix systemd unit file to run after network on non PVE nodes (by @robertdahlem) [#137]
-
-
 ## [1.1.1] - 2025-04-20
 
 ### Added

Dockerfile

@@ -9,33 +9,20 @@ LABEL org.label-schema.vendor="gyptazy"
 LABEL org.label-schema.url="https://proxlb.de"
 LABEL org.label-schema.vcs-url="https://github.com/gyptazy/ProxLB"
 
-# --- Step 1 (root): system deps, user, dirs ---
-RUN apk add --no-cache python3 py3-pip \
-  && addgroup -S plb \
-  && adduser -S -G plb -h /home/plb plb \
-  && mkdir -p /app/conf /opt/venv \
-  && chown -R plb:plb /app /home/plb /opt/venv
+# Install Python3
+RUN apk add --no-cache python3 py3-pip
 
+# Create a directory for the app
 WORKDIR /app
 
-# Copy only requirements first for better layer caching
-COPY --chown=plb:plb requirements.txt /app/requirements.txt
+# Copy the python program from the current directory to /app
+COPY proxlb /app/proxlb
 
-# --- Step 2 (appuser): venv + deps + code ---
-USER plb
+# Copy requirements to the container
+COPY requirements.txt /app/requirements.txt
 
-# Create venv owned by appuser and put it on PATH
-RUN python3 -m venv /opt/venv
-ENV PATH="/opt/venv/bin:${PATH}"
+# Install dependencies in the virtual environment
+RUN pip install --break-system-packages -r /app/requirements.txt
 
-# Install Python dependencies into the venv (no PEP 668 issues)
-RUN pip install --no-cache-dir -r /app/requirements.txt
-
-# Copy application code (owned by appuser)
-COPY --chown=plb:plb proxlb /app/proxlb
-
-# Optional: placeholder config so a bind-mount can override cleanly
-RUN touch /app/conf/proxlb.yaml
-
-# Run as non-root using venv Python
-ENTRYPOINT ["/opt/venv/bin/python", "/app/proxlb/main.py"]
+# Set the entry point to use the virtual environment's python
+ENTRYPOINT ["/usr/bin/python3", "/app/proxlb/main.py"]

README.md

@@ -1,5 +1,5 @@
 # ProxLB - (Re)Balance VM Workloads in Proxmox Clusters
-<img align="left" src="https://cdn.gyptazy.com/img/ProxLB.jpg"/>
+<img align="left" src="https://cdn.gyptazy.com/images/Prox-LB-logo.jpg"/>
 <br>
 
 <p float="center"><img src="https://img.shields.io/github/license/gyptazy/ProxLB"/><img src="https://img.shields.io/github/contributors/gyptazy/ProxLB"/><img src="https://img.shields.io/github/last-commit/gyptazy/ProxLB/main"/><img src="https://img.shields.io/github/issues-raw/gyptazy/ProxLB"/><img src="https://img.shields.io/github/issues-pr/gyptazy/ProxLB"/></p>
@@ -8,32 +8,31 @@
 1. [Introduction](#introduction)
 2. [Features](#features)
 3. [How does it work?](#how-does-it-work)
-4. [Documentation](#documentation)
-5. [Installation](#installation)
+4. [Installation](#installation)
    1. [Requirements / Dependencies](#requirements--dependencies)
    2. [Debian Package](#debian-package)
    4. [Container / Docker](#container--docker)
    5. [Source](#source)
-6. [Usage / Configuration](#usage--configuration)
+5. [Usage / Configuration](#usage--configuration)
    1. [GUI Integration](#gui-integration)
    2. [Proxmox HA Integration](#proxmox-ha-integration)
    3. [Options](#options)
-7. [Affinity & Anti-Affinity Rules](#affinity--anti-affinity-rules)
+6. [Affinity & Anti-Affinity Rules](#affinity--anti-affinity-rules)
    1. [Affinity Rules](#affinity-rules)
    2. [Anti-Affinity Rules](#anti-affinity-rules)
    3. [Ignore VMs](#ignore-vms)
    4. [Pin VMs to Hypervisor Nodes](#pin-vms-to-hypervisor-nodes)
-8. [Maintenance](#maintenance)
-9. [Misc](#misc)
+7. [Maintenance](#maintenance)
+8. [Misc](#misc)
    1. [Bugs](#bugs)
    2. [Contributing](#contributing)
-   3. [Support](#support)
-   4. [Enterprise-Support](#enterprise-support)
-10. [Author(s)](#authors)
+   3. [Documentation](#documentation)
+   4. [Support](#support)
+9. [Author(s)](#authors)
 
 
 ## Introduction
-ProxLB is an advanced load balancing solution specifically designed for Proxmox clusters, addressing the absence of an intelligent and more advanced resource scheduler. As a third-party solution, ProxLB enhances the management and efficiency of Proxmox clusters by intelligently distributing workloads across available nodes. Workloads can be balanced by different times like the guest's memory, CPU or disk usage or their assignment to avoid overprovisioning and ensuring resources.
+ProxLB is an advanced load balancing solution specifically designed for Proxmox clusters, addressing the absence of a Dynamic Resource Scheduler (DRS) that is familiar to VMware users. As a third-party solution, ProxLB enhances the management and efficiency of Proxmox clusters by intelligently distributing workloads across available nodes. Workloads can be balanced by different times like the guest's memory, CPU or disk usage or their assignment to avoid overprovisioning and ensuring resources.
 
 One of the key advantages of ProxLB is that it is fully open-source and free, making it accessible for anyone to use, modify, and contribute to. This ensures transparency and fosters community-driven improvements. ProxLB supports filtering and ignoring specific nodes and guests through configuration files and API calls, providing administrators with the flexibility to tailor the load balancing behavior to their specific needs.
 
@@ -44,51 +43,41 @@ ProxLB can also return the best next node for guest placement, which can be inte
 Overall, ProxLB significantly enhances resource management by intelligently distributing workloads, reducing downtime through its maintenance mode, and providing improved flexibility with affinity and anti-affinity rules. Its seamless integration with CI/CD tools and reliance on the Proxmox API make it a robust and secure solution for optimizing Proxmox cluster performance.
 
 ### Video of Migration
-<img src="https://cdn.gyptazy.com/img/proxlb-rebalancing-demo.gif"/>
+<img src="https://cdn.gyptazy.com/images/proxlb-rebalancing-demo.gif"/>
 
 ## Features
-ProxLB's key features are by enabling automatic rebalancing of VMs and CTs across a Proxmox cluster based on memory, CPU, and local disk usage while identifying optimal nodes for automation. It supports maintenance mode, affinity rules, and seamless Proxmox API integration with ACL support, offering flexible usage as a one-time operation, a daemon, or through the Proxmox Web GUI.
+ProxLB's key features are by enabling automatic rebalancing of VMs and CTs across a Proxmox cluster based on memory, CPU, and local disk usage while identifying optimal nodes for automation. It supports maintenance mode, affinity rules, and seamless Proxmox API integration with ACL support, offering flexible usage as a one-time operation, a daemon, or through the Proxmox Web GUI. In addition, ProxLB also supports additional enterprise alike features like power managements for nodes (often also known as DPM) where nodes can be turned on/off on demand when workloads are higher/lower than usual. Also the automated security-patching of nodes within the cluster (known as ASPM) may help to reduce the manual work from cluster admins, where nodes will install patches, move guests across the cluster, reboot and then reblance the cluster again.
 
 **Features**
-* Rebalance VMs/CTs in the cluster by:
-  * Memory
-  * Disk (only local storage)
-  * CPU
-* Rebalance by different modes:
-  * Used resources
-  * Assigned resources
-  * PSI (Pressure) of resources
-* Get best nodes for further automation
-* Supported Guest Types
-  * VMs
-  * CTs
+* Re-Balancing (DRS)
+  * Supporting VMs & CTs
+  * Balancing by:
+    * CPU
+    * Memory
+    * Disk
+  * Affinity / Anti-Affinity Rules
+    * Affinity: Groups guests together
+    * Anti-Affinity: Ensuring guests run on different nodes
+  * Best node evaluation
+    * Get the best node for guest placement (e.g., CI/CD)
 * Maintenance Mode
-  * Set node(s) into maintenance
-  * Move all workloads to different nodes
-* Affinity / Anti-Affinity Rules
+  * Evacuating a sinlge or multiple nodes
+* Node Power Management (DPM)
+* Auto Node Security-Patch-Management (ASPM)
 * Fully based on Proxmox API
-  * Fully integrated into the Proxmox ACL
-  * No SSH required
-* Usage
-  * One-Time
-  * Daemon
-  * Proxmox Web GUI Integration
+  * Utilizing the Proxmox User Authentications
+  * Supporting API tokens
+  * No SSH or Agents required
+  * Can run everywhere
 
 ## How does it work?
 ProxLB is a load-balancing system designed to optimize the distribution of virtual machines (VMs) and containers (CTs) across a cluster. It works by first gathering resource usage metrics from all nodes in the cluster through the Proxmox API. This includes detailed resource metrics for each VM and CT on every node. ProxLB then evaluates the difference between the maximum and minimum resource usage of the nodes, referred to as "Balanciness." If this difference exceeds a predefined threshold (which is configurable), the system initiates the rebalancing process.
 
 Before starting any migrations, ProxLB validates that rebalancing actions are necessary and beneficial. Depending on the selected balancing mode — such as CPU, memory, or disk — it creates a balancing matrix. This matrix sorts the VMs by their maximum used or assigned resources, identifying the VM with the highest usage. ProxLB then places this VM on the node with the most free resources in the selected balancing type. This process runs recursively until the operator-defined Balanciness is achieved. Balancing can be defined for the used or max. assigned resources of VMs/CTs.
 
-## Documentation
-This `README.md` doesn't contain all information and only highlights the most important facts. Extended information, such like API permissions, creating dedicated user, best-practices in running ProxLB and much more can be found in the [docs/](https://github.com/gyptazy/ProxLB/tree/main/docs) directory. Please consult the documentation before creating issues.
-
 ## Installation
 
 ### Requirements / Dependencies
-* Proxmox
-    * Proxmox 7.x
-    * Proxmox 8.x
-    * Proxmox 9.x
 * Python3.x
 * proxmoxer
 * requests
@@ -142,7 +131,7 @@ wget -O /etc/apt/trusted.gpg.d/proxlb.asc https://repo.gyptazy.com/repository.gp
 
 #### Debian Packages (.deb files)
 If you do not want to use the repository you can also find the debian packages as a .deb file on gyptazy's CDN at:
-* https://cdn.gyptazy.com/debian/proxlb/
+* https://cdn.gyptazy.com/files/os/debian/proxlb/
 
 Afterwards, you can simply install the package by running:
 ```bash
@@ -173,17 +162,6 @@ docker run -it --rm -v $(pwd)/proxlb.yaml:/etc/proxlb/proxlb.yaml proxlb
 | Version | Image |
 |------|:------:|
 | latest | cr.gyptazy.com/proxlb/proxlb:latest |
-| v1.1.10 | cr.gyptazy.com/proxlb/proxlb:v1.1.10 |
-| v1.1.9.1 | cr.gyptazy.com/proxlb/proxlb:v1.1.9.1 |
-| v1.1.9 | cr.gyptazy.com/proxlb/proxlb:v1.1.9 |
-| v1.1.8 | cr.gyptazy.com/proxlb/proxlb:v1.1.8 |
-| v1.1.7 | cr.gyptazy.com/proxlb/proxlb:v1.1.7 |
-| v1.1.6.1 | cr.gyptazy.com/proxlb/proxlb:v1.1.6.1 |
-| v1.1.6 | cr.gyptazy.com/proxlb/proxlb:v1.1.6 |
-| v1.1.5 | cr.gyptazy.com/proxlb/proxlb:v1.1.5 |
-| v1.1.4 | cr.gyptazy.com/proxlb/proxlb:v1.1.4 |
-| v1.1.3 | cr.gyptazy.com/proxlb/proxlb:v1.1.3 |
-| v1.1.2 | cr.gyptazy.com/proxlb/proxlb:v1.1.2 |
 | v1.1.1 | cr.gyptazy.com/proxlb/proxlb:v1.1.1 |
 | v1.1.0 | cr.gyptazy.com/proxlb/proxlb:v1.1.0 |
 | v1.0.6 | cr.gyptazy.com/proxlb/proxlb:v1.0.6 |
@@ -237,7 +215,7 @@ docker run -it --rm -v $(pwd)/proxlb.yaml:/etc/proxlb/proxlb.yaml proxlb
 Running ProxLB is straightforward and versatile, as it only requires `Python3` and the `proxmoxer` library. This means ProxLB can be executed directly on a Proxmox node or on dedicated systems such as Debian, RedHat, or even FreeBSD, provided that the Proxmox API is accessible from the client running ProxLB. ProxLB can also run inside a Container - Docker or LXC - and is simply up to you.
 
 ### GUI Integration
-<img align="left" src="https://cdn.gyptazy.com/img/rebalance-ui.jpg"/> ProxLB can also be accessed through the Proxmox Web UI by installing the optional `pve-proxmoxlb-service-ui` package, which depends on the proxlb package. For full Web UI integration, this package must be installed on all nodes within the cluster. Once installed, a new menu item - `Rebalancing`, appears in the cluster level under the HA section. Once installed, it offers two key functionalities:
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-GUI-integration.jpg"/> ProxLB can also be accessed through the Proxmox Web UI by installing the optional `pve-proxmoxlb-service-ui` package, which depends on the proxlb package. For full Web UI integration, this package must be installed on all nodes within the cluster. Once installed, a new menu item - `Rebalancing`, appears in the cluster level under the HA section. Once installed, it offers two key functionalities:
 * Rebalancing VM workloads
 * Migrate VM workloads away from a defined node (e.g. maintenance preparation)
 
@@ -260,7 +238,7 @@ The following options can be set in the configuration file `proxlb.yaml`:
 | Section | Option | Sub Option | Example | Type | Description |
 |---------|:------:|:----------:|:-------:|:----:|:-----------:|
 | `proxmox_api` |  |  |  |  |  |
-|  | hosts |  | ['virt01.example.com', '10.10.10.10', 'fe01:bad:code::cafe', 'virt01.example.com:443', '[fc00::1]', '[fc00::1]:443', 'fc00::1:8006'] | `List` | List of Proxmox nodes. Can be IPv4, IPv6 or mixed. You can specify custom ports. In case of IPv6 without brackets the port is considered after the last colon |
+|  | hosts |  | ['virt01.example.com', '10.10.10.10', 'fe01::bad:code::cafe'] | `List` | List of Proxmox nodes. Can be IPv4, IPv6 or mixed. |
 |  | user |  | root@pam | `Str` | Username for the API. |
 |  | pass |  | FooBar | `Str` | Password for the API. (Recommended: Use API token authorization!) |
 |  | token_id |  | proxlb | `Str` | Token ID of the user for the API. |
@@ -270,41 +248,38 @@ The following options can be set in the configuration file `proxlb.yaml`:
 |  | retries |  | 1 | `Int` | How often a connection attempt to the defined API host should be performed. |
 |  | wait_time |  | 1 | `Int` | How many seconds should be waited before performing another connection attempt to the API host. |
 | `proxmox_cluster` |  |  |  |  |  |
-|  | maintenance_nodes |  | ['virt66.example.com'] | `List` | A list of Proxmox nodes that are defined to be in a maintenance. (must be the same node names as used within the cluster) |
+|  | maintenance_nodes |  | ['virt66.example.com'] | `List` | A list of Proxmox nodes that are defined to be in a maintenance. |
 |  | ignore_nodes |  | [] | `List` | A list of Proxmox nodes that are defined to be ignored. |
 |  | overprovisioning |  | False | `Bool` | Avoids balancing when nodes would become overprovisioned. |
 | `balancing` |  |  |  |  |  |
 |  | enable |  | True | `Bool` | Enables the guest balancing.|
 |  | enforce_affinity |  | True | `Bool` | Enforcing affinity/anti-affinity rules but balancing might become worse. |
 |  | parallel |  | False | `Bool` | If guests should be moved in parallel or sequentially.|
-|  | parallel_jobs |  | 5 | `Int` | The amount if parallel jobs when migrating guests. (default: `5`)|
 |  | live |  | True | `Bool` | If guests should be moved live or shutdown.|
 |  | with_local_disks |  | True | `Bool` | If balancing of guests should include local disks.|
-|  | with_conntrack_state |  | True | `Bool` | If balancing of guests should including the conntrack state.|
 |  | balance_types |  | ['vm', 'ct'] | `List` | Defined the types of guests that should be honored. [values: `vm`, `ct`]|
 |  | max_job_validation |  | 1800 | `Int` | How long a job validation may take in seconds. (default: 1800) |
 |  | balanciness |  | 10 | `Int` | The maximum delta of resource usage between node with highest and lowest usage. |
-|  | memory_threshold |  | 75 | `Int` | The maximum threshold (in percent) that needs to be hit to perform balancing actions. (Optional) |
 |  | method |  | memory | `Str` | The balancing method that should be used.  [values: `memory` (default), `cpu`, `disk`]|
-|  | mode |  | used | `Str` | The balancing mode that should be used. [values: `used` (default), `assigned`, `psi` (pressure)] |
-|  | psi |  | { nodes: { memory: { pressure_full: 0.20, pressure_some: 0.20, pressure_spikes: 1.00 }}} | `Dict` | A dict of PSI based thresholds for nodes and guests |
-|  | pools |  | pools: { dev: { type: affinity }, de-nbg01-db: { type: anti-affinity }} | `Dict` | A dict of pool names and their type for creating affinity/anti-affinity rules |
+|  | mode |  | used | `Str` | The balancing mode that should be used. [values: `used` (default), `assigned`] |
+| `dpm` |  |  |  |  |  |
+|  | enable |  | True | `Bool` | Enables the Dynamic Power Management functions.|
+|  | method |  | memory | `Str` | The balancing method that should be used.  [values: `memory` (default), `cpu`, `disk`]|
+|  | mode |  | static | `Str` | The balancing mode that should be used. [values: `static` (default), `auto`] |
+|  | cluster_min_free_resources |  | 60 | `Int` | Representing the minimum required free resouzrces in percent within the cluster. [values: `60`% (default)] |
+|  | cluster_min_nodes |  | 3 | `Int` | The minimum of required nodes that should remain in a cluster. [values: `3` (default)] |
 | `service` |  |  |  |  |  |
 |  | daemon |  | True | `Bool` | If daemon mode should be activated. |
 |  | `schedule` |  |  | `Dict` | Schedule config block for rebalancing. |
 |  |  | interval | 12 | `Int` | How often rebalancing should occur in daemon mode.|
 |  |  | format | hours | `Str` | Sets the time format. [values: `hours` (default), `minutes`]|
-|  | `delay` |  |  | `Dict` | Schedule config block for an optional delay until the service starts. |
-|  |  | enable | False | `Bool` | If a delay time should be validated.|
-|  |  | time | 1 | `Int` | Delay time until the service starts after the initial execution.|
-|  |  | format | hours | `Str` | Sets the time format. [values: `hours` (default), `minutes`]|
 |  | log_level |  | INFO | `Str` | Defines the default log level that should be logged. [values: `INFO` (default), `WARNING`, `CRITICAL`, `DEBUG`] |
 
 
 An example of the configuration file looks like:
 ```
 proxmox_api:
-  hosts: ['virt01.example.com', '10.10.10.10', 'fe01:bad:code::cafe']
+  hosts: ['virt01.example.com', '10.10.10.10', 'fe01::bad:code::cafe']
   user: root@pam
   pass: crazyPassw0rd!
   # API Token method
@@ -327,60 +302,26 @@ balancing:
   parallel: False
   live: True
   with_local_disks: True
-  with_conntrack_state: True
   balance_types: ['vm', 'ct']
   max_job_validation: 1800
-  memory_threshold: 75
   balanciness: 5
   method: memory
   mode: used
-# # PSI thresholds only apply when using mode 'psi'
-# # PSI based balancing is currently in beta and req. PVE >= 9
-# psi:
-#   nodes:
-#     memory:
-#       pressure_full: 0.20
-#       pressure_some: 0.20
-#       pressure_spikes: 1.00
-#     cpu:
-#       pressure_full: 0.20
-#       pressure_some: 0.20
-#       pressure_spikes: 1.00
-#     disk:
-#       pressure_full: 0.20
-#       pressure_some: 0.20
-#       pressure_spikes: 1.00
-#   guests:
-#     memory:
-#       pressure_full: 0.20
-#       pressure_some: 0.20
-#       pressure_spikes: 1.00
-#     cpu:
-#       pressure_full: 0.20
-#       pressure_some: 0.20
-#       pressure_spikes: 1.00
-#     disk:
-#       pressure_full: 0.20
-#       pressure_some: 0.20
-#       pressure_spikes: 1.00
-  pools:
-    dev:
-      type: affinity
-    de-nbg01-db
-      type: anti-affinity
-      pin:
-        - virt66
-        - virt77
+
+dpm:
+  # DPM requires you to define the WOL (Wake-on-Lan)
+  # MAC address for each node in Proxmox.
+  enable: True
+  method: memory
+  mode: static
+  cluster_min_free_resources: 60
+  cluster_min_nodes: 1
 
 service:
   daemon: True
   schedule:
     interval: 12
     format: hours
-  delay:
-    enable: False
-    time: 1
-    format: hours
   log_level: INFO
 ```
 
@@ -401,33 +342,19 @@ ProxLB provides an advanced mechanism to define affinity and anti-affinity rules
 ProxLB implements affinity and anti-affinity rules through a tag-based system within the Proxmox web interface. Each guest (virtual machine or container) can be assigned specific tags, which then dictate its placement behavior. This method maintains a streamlined and secure approach to managing VM relationships while preserving Proxmox’s inherent permission model.
 
 ### Affinity Rules
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-affinity-rules.jpg"/> Affinity rules are used to group certain VMs together, ensuring that they run on the same host whenever possible. This can be beneficial for workloads requiring low-latency communication, such as clustered databases or application servers that frequently exchange data. In general, there're two ways to manage affinity rules:
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-affinity-rules.jpg"/> Affinity rules are used to group certain VMs together, ensuring that they run on the same host whenever possible. This can be beneficial for workloads requiring low-latency communication, such as clustered databases or application servers that frequently exchange data.
 
-#### Affinity Rules by Tags
 To define an affinity rule which keeps all guests assigned to this tag together on a node, users assign a tag with the prefix `plb_affinity_$TAG`:
 
 #### Example for Screenshot
 ```
 plb_affinity_talos
 ```
+
 As a result, ProxLB will attempt to place all VMs with the `plb_affinity_web` tag on the same host (see also the attached screenshot with the same node).
 
-#### Affinity Rules by Pools
-Antoher approach is by using pools in Proxmox. This way, it can easily also combined with other resources like backup jobs. However, in this approach you need to modify the ProxLB config file to your needs. Within the `balancing` section you can create a dict of pools, including the pool name and the affinity type. Please see the example for further details:
-
-**Example Config**
-```
-balancing:
-  [...]
-  pools:                              # Optional: Define affinity/anti-affinity rules per pool
-    dev:                              # Pool name: dev
-      type: affinity                  # Type: affinity (keeping VMs together)
-      pin:                            # Pin VMs to Nodes
-        - virt77                        # Pinning to 'virt77' which is maybe an older system for dev labs
-```
-
-### Anti-Affinity Rules by Tags
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-anti-affinity-rules.jpg"/> Conversely, anti-affinity rules ensure that designated VMs do not run on the same physical host. This is particularly useful for high-availability setups, where redundancy is crucial. Ensuring that critical services are distributed across multiple hosts reduces the risk of a single point of failure. In general, there're two ways to manage anti-affinity rules:
+### Anti-Affinity Rules
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-anti-affinity-rules.jpg"/> Conversely, anti-affinity rules ensure that designated VMs do not run on the same physical host. This is particularly useful for high-availability setups, where redundancy is crucial. Ensuring that critical services are distributed across multiple hosts reduces the risk of a single point of failure.
 
 To define an anti-affinity rule that ensures to not move systems within this group to the same node, users assign a tag with the prefix:
 
@@ -438,23 +365,10 @@ plb_anti_affinity_ntp
 
 As a result, ProxLB will try to place the VMs with the `plb_anti_affinity_ntp` tag on different hosts (see also the attached screenshot with the different nodes).
 
-#### Anti-Affinity Rules by Pools
-Antoher approach is by using pools in Proxmox. This way, it can easily also combined with other resources like backup jobs. However, in this approach you need to modify the ProxLB config file to your needs. Within the `balancing` section you can create a dict of pools, including the pool name and the affinity type. Please see the example for further details:
-
-**Example Config**
-```
-balancing:
-  [...]
-  pools:                              # Optional: Define affinity/anti-affinity rules per pool
-    de-nbg01-db:                      # Pool name: de-nbg01-db
-      type: anti-affinity                  # Type: anti-affinity (spreading VMs apart)
-```
-
-
 **Note:** While this ensures that ProxLB tries distribute these VMs across different physical hosts within the Proxmox cluster this may not always work. If you have more guests attached to the group than nodes in the cluster, we still need to run them anywhere. If this case occurs, the next one with the most free resources will be selected.
 
 ### Ignore VMs
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-ignore-vm-movement.jpg"/> Guests, such as VMs or CTs, can also be completely ignored. This means, they won't be affected by any migration (even when (anti-)affinity rules are enforced). To ensure a proper resource evaluation, these guests are still collected and evaluated but simply skipped for balancing actions. Another thing is the implementation. While ProxLB might have a very restricted configuration file including the file permissions, this file is only read- and writeable by the Proxmox administrators. However, we might have user and groups who want to define on their own that their systems shouldn't be moved. Therefore, these users can simpy set a specific tag to the guest object - just like the (anti)affinity rules.
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-ignore-vm-movement.jpg"/> Guests, such as VMs or CTs, can also be completely ignored. This means, they won't be affected by any migration (even when (anti-)affinity rules are enforced). To ensure a proper resource evaluation, these guests are still collected and evaluated but simply skipped for balancing actions. Another thing is the implementation. While ProxLB might have a very restricted configuration file including the file permissions, this file is only read- and writeable by the Proxmox administrators. However, we might have user and groups who want to define on their own that their systems shouldn't be moved. Therefore, these users can simpy set a specific tag to the guest object - just like the (anti)affinity rules.
 
 To define a guest to be ignored from the balancing, users assign a tag with the prefix `plb_ignore_$TAG`:
 
@@ -468,9 +382,8 @@ As a result, ProxLB will not migrate this guest with the `plb_ignore_dev` tag to
 **Note:** Ignored guests are really ignored. Even by enforcing affinity rules this guest will be ignored.
 
 ### Pin VMs to Specific Hypervisor Nodes
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-tag-node-pinning.jpg"/> Guests, such as VMs or CTs, can also be pinned to specific (and multiple) nodes in the cluster. This might be usefull when running applications with some special licensing requirements that are only fulfilled on certain nodes. It might also be interesting, when some physical hardware is attached to a node, that is not available in general within the cluster.
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-tag-node-pinning.jpg"/> Guests, such as VMs or CTs, can also be pinned to specific nodes in the cluster. This might be usefull when running applications with some special licensing requirements that are only fulfilled on certain nodes. It might also be interesting, when some physical hardware is attached to a node, that is not available in general within the cluster.
 
-#### Pinning VMs to (a) specific Hypervisor Node(s) by Tag
 To pin a guest to a specific cluster node, users assign a tag with the prefix `plb_pin_$nodename` to the desired guest:
 
 #### Example for Screenshot
@@ -480,27 +393,11 @@ plb_pin_node03
 
 As a result, ProxLB will pin the guest `dev-vm01` to the node `virt03`.
 
-
-#### Pinning VMs to (a) specific Hypervisor Node(s) by Pools
-Beside the tag approach, you can also pin a resource group to a specific hypervisor or groups of hypervisors by defining a `pin` key of type list.
-
-**Example Config**
-```
-balancing:
-  [...]
-  pools:                              # Optional: Define affinity/anti-affinity rules per pool
-    dev:                              # Pool name: dev
-      type: affinity                  # Type: affinity (keeping VMs together)
-      pin:                            # Pin VMs to Nodes
-        - virt77                        # Pinning to 'virt77' which is maybe an older system for dev labs
-```
-
-
-You can also repeat this step multiple times for different node names to create a potential group of allowed hosts where a the guest may be served on. In this case, ProxLB takes the node with the lowest used resources according to the defined balancing values from this group.
-
 **Note:** The given node names from the tag are validated. This means, ProxLB validated if the given node name is really part of the cluster. In case of a wrongly defined or unavailable node name it continous to use the regular processes to make sure the guest keeps running.
 
 ## Maintenance
+<img src="https://cdn.gyptazy.com/images/proxlb-rebalancing-demo.gif"/>
+
 The `maintenance_nodes` option allows operators to designate one or more Proxmox nodes for maintenance mode. When a node is set to maintenance, no new guest workloads will be assigned to it, and all existing workloads will be migrated to other available nodes within the cluster. This process ensures that (anti)-affinity rules and resource availability are respected, preventing disruptions while maintaining optimal performance across the infrastructure.
 
 ### Adding / Removing Nodes from Maintenance
@@ -518,6 +415,9 @@ Bugs can be reported via the GitHub issue tracker [here](https://github.com/gypt
 ### Contributing
 Feel free to add further documentation, to adjust already existing one or to contribute with code. Please take care about the style guide and naming conventions. You can find more in our [CONTRIBUTING.md](https://github.com/gyptazy/ProxLB/blob/main/CONTRIBUTING.md) file.
 
+### Documentation
+You can also find additional and more detailed documentation within the [docs/](https://github.com/gyptazy/ProxLB/tree/main/docs) directory.
+
 ### Support
 If you need assistance or have any questions, we offer support through our dedicated [chat room](https://matrix.to/#/#proxlb:gyptazy.com) in Matrix or [Discord](https://discord.gg/JemGu7WbfQ). Join our community for real-time help, advice, and discussions. The Matrix and Discord room are bridged to ensure that the communication is not splitted - so simply feel free to join which fits most to you!
 
@@ -532,18 +432,5 @@ Connect with us in our dedicated chat room for immediate support and live intera
 
 **Note:** Please always keep in mind that this is a one-man show project without any further help. This includes coding, testing, packaging and all the infrastructure around it to keep this project up and running.
 
-### Enterprise-Support
-Running critical infrastructure in an enterprise environment often comes with requirements that go far beyond functionality alone. Enterprises typically expect predictable service levels, defined escalation paths, and guaranteed response times. In many cases, organizations also demand 24x7 support availability to ensure that their systems remain stable and resilient, even under unexpected circumstances.
-
-As the creator and maintainer of ProxLB, I operate as a one-man project. While I am continuously working to improve the software, I cannot provide the type of enterprise-grade support that large organizations may require. To address this need, several companies have stepped in to offer professional services around ProxLB in Proxmox VE clusters.
-
-Below is a list of organizations currently known to provide enterprise-level support for ProxLB. If your business relies on ProxLB in production and you require more than community-based support, these providers may be a good fit for your needs:
-
-| Company| Country | Web |
-|------|:------:|:------:|
-| credativ | DE | [credativ.de](https://www.credativ.de/en/portfolio/support/proxmox-virtualization/) |
-
-*Note: If you provide support for ProxLB, feel free to create PR with your addition.*
-
 ### Author(s)
  * Florian Paul Azim Hoberg @gyptazy (https://gyptazy.com)

config/proxlb_example.yaml

@@ -1,5 +1,5 @@
 proxmox_api:
-  hosts: ['virt01.example.com', '10.10.10.10', 'fe01:bad:code::cafe']
+  hosts: ['virt01.example.com', '10.10.10.10', 'fe01::bad:code::cafe']
   user: root@pam
   pass: crazyPassw0rd!
   # API Token method
@@ -20,62 +20,24 @@ balancing:
   enable: True
   enforce_affinity: False
   parallel: False
-  # If running parallel job, you can define
-  # the amount of prallel jobs (default: 5)
-  parallel_jobs: 1
   live: True
   with_local_disks: True
-  with_conntrack_state: True
-  balance_types: ['vm', 'ct']         # 'vm' | 'ct'
-  max_job_validation: 1800            # Maximum time (in seconds) a job validation may take
-  memory_threshold: 75                # Optional: Maximum threshold (in percent) to trigger balancing actions
-  balanciness: 5                      # Maximum delta of resource usage between highest and lowest usage node
-  method: memory                      # 'memory' | 'cpu' | 'disk'
-  mode: used                          # 'assigned' | 'used' | 'psi'
-  # # PSI thresholds only apply when using mode 'psi'
-  # psi:
-  #   nodes:
-  #     memory:
-  #       pressure_full: 0.20
-  #       pressure_some: 0.20
-  #       pressure_spikes: 1.00
-  #     cpu:
-  #       pressure_full: 0.20
-  #       pressure_some: 0.20
-  #       pressure_spikes: 1.00
-  #     disk:
-  #       pressure_full: 0.20
-  #       pressure_some: 0.20
-  #       pressure_spikes: 1.00
-  #   guests:
-  #     memory:
-  #       pressure_full: 0.20
-  #       pressure_some: 0.20
-  #       pressure_spikes: 1.00
-  #     cpu:
-  #       pressure_full: 0.20
-  #       pressure_some: 0.20
-  #       pressure_spikes: 1.00
-  #     disk:
-  #       pressure_full: 0.20
-  #       pressure_some: 0.20
-  #       pressure_spikes: 1.00
-  pools:                              # Optional: Define affinity/anti-affinity rules per pool
-    dev:                              # Pool name: dev
-      type: affinity                  # Type: affinity (keeping VMs together)
-    de-nbg01-db:                      # Pool name: de-nbg01-db
-      type: anti-affinity             # Type: anti-affinity (spreading VMs apart)
-      pin:                            # Define a pinning og guests to specific node(s)
-        - virt66
-        - virt77
+  balance_types: ['vm', 'ct']
+  max_job_validation: 1800
+  balanciness: 5
+  method: memory
+  mode: used
+
+dpm:
+  enable: True
+  method: memory
+  mode: static
+  cluster_min_free_resources: 60
+  cluster_min_nodes: 1
 
 service:
   daemon: True
   schedule:
     interval: 12
     format: hours
-  delay:
-    enable: False
-    time: 1
-    format: hours
   log_level: INFO

debian/changelog

@@ -1,92 +1,8 @@
-proxlb (1.1.10) stable; urgency=medium
+proxlb (1.1.2~b1) stable; urgency=medium
 
-  * Prevent redundant rebalancing by validating existing affinity enforcement before taking actions. (Closes: #335)
-  * Add safety-guard for PVE 8 users when activating conntrack-aware migrations mistakenly. (Closes: #359)
-  * Fix the Proxmox API connection validation which returned a false-positive logging message of timeouts. (Closes: #361)
-  * Refactored the whole Proxmox API connection function. (Closes: #361)
-  * Fix a crash during PVE resource pool enumeration by skipping members not having a 'name' property. (Closes: #368)
+  * Auto-created 1.1.2 beta 1 release.
 
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Tue, 25 Nov 2025 09:12:04 +0001
-
-proxlb (1.1.9.1) stable; urgency=medium
-
-  * Fix quoting in f-strings which may cause issues on PVE 8 / Debian Bookworm systems. (Closes: #352)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 30 Oct 2025 17:41:02 +0001
-
-proxlb (1.1.9) stable; urgency=medium
-
-  * Add pressure (PSI) based balancing for memory, cpu, disk (req. PVE9 or greater). (Closes: #339)
-  * Add (memory) threshold for nodes before running balancing. (Closes: #342)
-  * Add affinity/anti-affinity support by pools. (Closes: #343)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 30 Oct 2025 06:58:43 +0001
-
-proxlb (1.1.8) stable; urgency=medium
-
-  * Fix API errors when using conntrack aware migration with older PVE version. (Closes: #318)
-  * Add a static ProxLB prefix to the log output when used by journal handler. (Closes: #329)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 09 Oct 2025 09:04:13 +0002
-
-proxlb (1.1.7) stable; urgency=medium
-
-  * Add conntrack state aware migrations of VMs. (Closes: #305)
-  * Add graceful shutdown for SIGINT command. (Closes: #304)
-  * Fix crash when validating absent migration job ids. (Closes: #308)
-  * Fix guest object names are not being evaluated in debug log. (Closes: #310)
-  * Note: Have a great Dutch Proxmox Day 2025!
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 04 Sep 2025 19:23:51 +0000
-
-proxlb (1.1.6.1) stable; urgency=medium
-
-  * Validate for node presence when pinning VMs to avoid crashing. (Closes: #296)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 04 Sep 2025 19:23:51 +0000
-
-proxlb (1.1.6) stable; urgency=medium
-
-  * Add validation for provided API user token id to avoid confusions. (Closes: #291)
-  * Fix stacktrace output when validating permissions on non existing users in Proxmox. (Closes: #291)
-  * Fix Overprovisioning first node if anti_affinity_group has only one member. (Closes: #295)
-  * Validate for node presence when pinning guests to avoid crashing. (Closes: #296)
-  * Fix balancing evaluation of guest types (e.g., VM or CT). (Closes: #268)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 04 Sep 2025 05:12:19 +0000
-
-proxlb (1.1.5) stable; urgency=medium
-
-  * Allow custom API ports instead of fixed tcp/8006. (Closes: #260)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Mon, 14 Jul 2025 11:07:34 +0000
-
-proxlb (1.1.4) stable; urgency=medium
-
-  * Allow pinning of guests to a group of nodes. (Closes: #245)
-  * Modified log levels to make output lighter at INFO level. (Closes: #255)
-  * ixed an issue where balancing was performed in combination of deactivated balancing and dry-run mode. (Closes: #248)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Fri, 27 Jun 2025 16:22:58 +0000
-
-proxlb (1.1.3) stable; urgency=medium
-
-  * Add relaod (SIGHUP) function to ProxLB to reload the configuration. (Closes: #189)
-  * Add optional wait time parameter to delay execution until the service takes action. (Closes: #239)
-  * Make the amount of parallel migrations configurable. (Closes: #241)
-  * Use the average CPU consumption of a guest within the last 60 minutes instead of the current CPU usage. (Closes: #94)
-  * Align maintenance mode with Proxmox HA maintenance mode. (Closes: #232)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Thu, 19 Jun 2025 09:10:43 +0000
-
-proxlb (1.1.2) stable; urgency=medium
-
-  * Add a configurable retry mechanism when connecting to the Proxmox API. (Closed: #157)
-  * Add 1-to-1 relationships between guest and hypervisor node to ping a guest on a node. (Closes #218)
-  * Force type cast cpu count of guests to int for some corner cases where a str got returned. (Closed #222)
-  * Fix systemd unit file to run after network on non PVE nodes. (Closes #137)
-
- -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Mon, 13 May 2025 18:12:04 +0000
+ -- Florian Paul Azim Hoberg <gyptazy@gyptazy.com>  Mon, 17 Mar 2025 18:55:02 +0000
 
 proxlb (1.1.1) stable; urgency=medium
 

debian/control

@@ -7,6 +7,6 @@ Build-Depends: debhelper-compat (= 13), dh-python, python3-all, python3-setuptoo
 
 Package: proxlb
 Architecture: all
-Depends: ${python3:Depends}, ${misc:Depends}, python3-requests, python3-urllib3, python3-packaging, python3-proxmoxer, python3-yaml
-Description: An advanced resource scheduler and load balancer for Proxmox clusters
- An advanced resource scheduler and load balancer for Proxmox clusters that also supports maintenance mode and affinity/anti-affinity rules.
+Depends: ${python3:Depends}, ${misc:Depends}, python3-requests, python3-urllib3, python3-proxmoxer, python3-yaml 
+Description: A DRS alike Load Balancer for Proxmox Clusters
+ An advanced DRS alike loadbalancer for Proxmox clusters that also supports maintenance modes and affinity/anti-affinity rules.

docs/01_requirements.md

@@ -55,7 +55,7 @@ ProxLB itself requires minimal system resources to operate. However, for managin
 
 
 ## Where To Run?
-ProxLB is lightweight and flexible where it runs on nearly any environment and only needs access to your Proxmox host’s API endpoint (commonly TCP port 8006).
+ProxLB can run on pretty anthing and only requires you to have a network connectivity to any of the Proxmox host's API (usually on tcp/8006).
 
 Therefore, you can simply run ProxLB on:
 * Bare-metal Systems

docs/03_configuration.md

@@ -19,11 +19,7 @@
     6. [Parallel Migrations](#parallel-migrations)
     7. [Run as a Systemd-Service](#run-as-a-systemd-service)
     8. [SSL Self-Signed Certificates](#ssl-self-signed-certificates)
-    9. [Node Maintenances](#node-maintenances)
-    10. [Balancing Methods](#balancing-methods)
-        1. [Used Resources](#used-resources)
-        2. [Assigned Resources](#assigned-resources)
-        3. [Pressure (PSI) based Resources](#pressure-psi-based-resources)
+    9. [Dynamic Power Management (DPM)](#dynamic-power-management)
 
 ## Authentication / User Accounts / Permissions
 ### Authentication
@@ -80,8 +76,8 @@ ProxLB provides an advanced mechanism to define affinity and anti-affinity rules
 
 ProxLB implements affinity and anti-affinity rules through a tag-based system within the Proxmox web interface. Each guest (virtual machine or container) can be assigned specific tags, which then dictate its placement behavior. This method maintains a streamlined and secure approach to managing VM relationships while preserving Proxmox’s inherent permission model.
 
-#### Affinity Rules by Tags
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-affinity-rules.jpg"/> Affinity rules are used to group certain VMs together, ensuring that they run on the same host whenever possible. This can be beneficial for workloads requiring low-latency communication, such as clustered databases or application servers that frequently exchange data.
+#### Affinity Rules
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-affinity-rules.jpg"/> Affinity rules are used to group certain VMs together, ensuring that they run on the same host whenever possible. This can be beneficial for workloads requiring low-latency communication, such as clustered databases or application servers that frequently exchange data.
 
 To define an affinity rule which keeps all guests assigned to this tag together on a node, users assign a tag with the prefix `plb_affinity_$TAG`:
 
@@ -92,20 +88,8 @@ plb_affinity_talos
 
 As a result, ProxLB will attempt to place all VMs with the `plb_affinity_web` tag on the same host (see also the attached screenshot with the same node).
 
-#### Affinity Rules by Pools
-Antoher approach is by using pools in Proxmox. This way, it can easily also combined with other resources like backup jobs. However, in this approach you need to modify the ProxLB config file to your needs. Within the `balancing` section you can create a dict of pools, including the pool name and the affinity type. Please see the example for further details:
-
-**Example Config**
-```
-balancing:
-  [...]
-  pools:                              # Optional: Define affinity/anti-affinity rules per pool
-    dev:                              # Pool name: dev
-      type: affinity                  # Type: affinity (keeping VMs together)
-```
-
-#### Anti-Affinity Rules by Tags
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-anti-affinity-rules.jpg"/> Conversely, anti-affinity rules ensure that designated VMs do not run on the same physical host. This is particularly useful for high-availability setups, where redundancy is crucial. Ensuring that critical services are distributed across multiple hosts reduces the risk of a single point of failure.
+#### Anti-Affinity Rules
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-anti-affinity-rules.jpg"/> Conversely, anti-affinity rules ensure that designated VMs do not run on the same physical host. This is particularly useful for high-availability setups, where redundancy is crucial. Ensuring that critical services are distributed across multiple hosts reduces the risk of a single point of failure.
 
 To define an anti-affinity rule that ensures to not move systems within this group to the same node, users assign a tag with the prefix:
 
@@ -118,18 +102,6 @@ As a result, ProxLB will try to place the VMs with the `plb_anti_affinity_ntp` t
 
 **Note:** While this ensures that ProxLB tries distribute these VMs across different physical hosts within the Proxmox cluster this may not always work. If you have more guests attached to the group than nodes in the cluster, we still need to run them anywhere. If this case occurs, the next one with the most free resources will be selected.
 
-#### Anti-Affinity Rules by Pools
-Antoher approach is by using pools in Proxmox. This way, it can easily also combined with other resources like backup jobs. However, in this approach you need to modify the ProxLB config file to your needs. Within the `balancing` section you can create a dict of pools, including the pool name and the affinity type. Please see the example for further details:
-
-**Example Config**
-```
-balancing:
-  [...]
-  pools:                              # Optional: Define affinity/anti-affinity rules per pool
-    de-nbg01-db:                      # Pool name: de-nbg01-db
-      type: anti-affinity                  # Type: anti-affinity (spreading VMs apart)
-````
-
 ### Affinity / Anti-Affinity Enforcing
 When a cluster is already balanced and does not require further adjustments, enabling the enforce_affinity parameter ensures that affinity and anti-affinity rules are still respected. This parameter prioritizes the placement of guest objects according to these rules, even if it leads to slight resource imbalances or increased migration overhead. Regularly reviewing and updating these rules, along with monitoring cluster performance, helps maintain optimal performance and reliability. By carefully managing these aspects, you can create a cluster environment that meets your specific needs and maintains a good balance of resources.
 
@@ -141,7 +113,7 @@ balancing:
 *Note: This may have impacts to the cluster. Depending on the created group matrix, the result may also be an unbalanced cluster.*
 
 ### Ignore VMs / CTs
-<img align="left" src="https://cdn.gyptazy.com/img/proxlb-ignore-vm-movement.jpg"/> Guests, such as VMs or CTs, can also be completely ignored. This means, they won't be affected by any migration (even when (anti-)affinity rules are enforced). To ensure a proper resource evaluation, these guests are still collected and evaluated but simply skipped for balancing actions. Another thing is the implementation. While ProxLB might have a very restricted configuration file including the file permissions, this file is only read- and writeable by the Proxmox administrators. However, we might have user and groups who want to define on their own that their systems shouldn't be moved. Therefore, these users can simpy set a specific tag to the guest object - just like the (anti)affinity rules.
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-ignore-vm-movement.jpg"/> Guests, such as VMs or CTs, can also be completely ignored. This means, they won't be affected by any migration (even when (anti-)affinity rules are enforced). To ensure a proper resource evaluation, these guests are still collected and evaluated but simply skipped for balancing actions. Another thing is the implementation. While ProxLB might have a very restricted configuration file including the file permissions, this file is only read- and writeable by the Proxmox administrators. However, we might have user and groups who want to define on their own that their systems shouldn't be moved. Therefore, these users can simpy set a specific tag to the guest object - just like the (anti)affinity rules.
 
 To define a guest to be ignored from the balancing, users assign a tag with the prefix `plb_ignore_$TAG`:
 
@@ -166,20 +138,14 @@ plb_pin_node03
 
 As a result, ProxLB will pin the guest `dev-vm01` to the node `virt03`.
 
-You can also repeat this step multiple times for different node names to create a potential group of allowed hosts where a the guest may be served on. In this case, ProxLB takes the node with the lowest used resources according to the defined balancing values from this group.
-
 **Note:** The given node names from the tag are validated. This means, ProxLB validated if the given node name is really part of the cluster. In case of a wrongly defined or unavailable node name it continous to use the regular processes to make sure the guest keeps running.
 
 ### API Loadbalancing
-ProxLB supports API loadbalancing, where one or more host objects can be defined as a list. This ensures, that you can even operator ProxLB without further changes when one or more nodes are offline or in a maintenance. When defining multiple hosts, the first reachable one will be picked. You can speficy custom ports in the list. There are 4 ways of defining hosts with ports:
-1. Hostname of IPv4 without port (in this case the default 8006 will be used)
-2. Hostname or IPv4 with port
-3. IPv6 in brackets with optional port
-4. IPv6 without brackets, in this case the port is assumed after last colon
+ProxLB supports API loadbalancing, where one or more host objects can be defined as a list. This ensures, that you can even operator ProxLB without further changes when one or more nodes are offline or in a maintenance. When defining multiple hosts, the first reachable one will be picked.
 
 ```
 proxmox_api:
-  hosts: ['virt01.example.com', '10.10.10.10', 'fe01::bad:code::cafe', 'virt01.example.com:443', '[fc00::1]', '[fc00::1]:443', 'fc00::1:8006']
+  hosts: ['virt01.example.com', '10.10.10.10', 'fe01::bad:code::cafe']
 ```
 
 ### Ignore Host-Nodes or Guests
@@ -244,134 +210,32 @@ proxmox_api:
 
 *Note: Disabling SSL certificate validation is not recommended.*
 
-### Node Maintenances
-To exclude specific nodes from receiving any new workloads during the balancing process, the `maintenance_nodes` configuration option can be used. This option allows administrators to define a list of nodes that are currently undergoing maintenance or should otherwise not be used for running virtual machines or containers.
+### Dynamic Power Management (DPM)
+<img align="left" src="https://cdn.gyptazy.com/images/proxlb-proxmox-node-wakeonlan-wol-mac-dpm.jpg"/> Configuring Dynamic Power Management (DPM) in ProxLB within a Proxmox cluster involves a few critical steps to ensure proper operation. The first consideration is that any node intended for automatic shutdown and startup must support Wake-on-LAN (WOL). This is essential because DPM relies on the ability to power nodes back on remotely. For this to work, the ProxLB instance must be able to reach the target node’s MAC address directly over the network.
 
-```yaml
-maintenance_nodes:
-  - virt66.example.com
-```
+To make this possible, you must configure the correct MAC address for WOL within the Proxmox web interface. This is done by selecting the node, going to the “System” section, then “Options,” and finally setting the “MAC address for Wake-on-LAN.” Alternatively, this value can also be submitted using the Proxmox API. Without this MAC address in place, ProxLB will not allow the node to be shut down. This restriction is in place to prevent nodes from being turned off without a way to bring them back online, which could lead to service disruption. By ensuring that each node has a valid WOL MAC address configured, DPM can operate safely and effectively, allowing ProxLB to manage the cluster’s power consumption dynamically.
 
-which can also be written as:
+#### Requirements
+Using the powermanagement feature within clusters comes along with several requirements:
+* ProxLB needs to reach the WOL-Mac address of the node (plain network)
+* WOL must be enabled of the node in general (BIOS/UEFI)
+* The related WOL network interface must be defined
+* The related WOL network interface MAC address must be defined in Proxmox for the node
 
-```yaml
-maintenance_nodes: ['virt66.example.com']
-```
+#### Options
+| Section | Option | Sub Option | Example | Type | Description |
+|---------|:------:|:----------:|:-------:|:----:|:-----------:|
+| `dpm` |  |  |  |  |  |
+|  | enable |  | True | `Bool` | Enables the Dynamic Power Management functions.|
+|  | method |  | memory | `Str` | The balancing method that should be used.  [values: `memory` (default), `cpu`, `disk`]|
+|  | mode |  | static | `Str` | The balancing mode that should be used. [values: `static` (default), `auto`] |
+|  | cluster_min_free_resources |  | 60 | `Int` | Representing the minimum required free resouzrces in percent within the cluster. [values: `60`% (default)] |
+|  | cluster_min_nodes |  | 3 | `Int` | The minimum of required nodes that should remain in a cluster. [values: `3` (default)] |
 
-The maintenance_nodes key must be defined as a list, even if it only includes a single node. Each entry in the list must exactly match the node name as it is known within the Proxmox VE cluster. Do not use IP addresses, alternative DNS names, or aliases—only the actual cluster node names are valid. Once a node is marked as being in maintenance mode:
+#### DPM Modes
+##### Static
+Static mode in DPM lets you set a fixed number of nodes that should always stay powered on in a Proxmox cluster. This is important to keep the cluster working properly, since you need at least three nodes to maintain quorum. The system won’t let you go below that limit to avoid breaking cluster functionality.
 
-* No new workloads will be balanced or migrated onto it.
-* Any existing workloads currently running on the node will be migrated away in accordance with the configured balancing strategies, assuming resources on other nodes allow. 
+Besides the minimum number of active nodes, you can also define a baseline for how many free resources—like CPU or RAM—should always be available when the virtual machines are running. If the available resources drop below that level, ProxLB will try to power on more nodes, as long as they're available and can be started. On the other hand, if the cluster has more than enough resources, ProxLB will begin to shut down nodes again, but only until the free resource threshold is reached.
 
-This feature is particularly useful during planned maintenance, upgrades, or troubleshooting, ensuring that services continue to run with minimal disruption while the specified node is being worked on.
-
-## 10. Balancing Methods
-ProxLB provides multiple balancing modes that define *how* resources are evaluated and compared during cluster balancing.
-Each mode reflects a different strategy for determining load and distributing guests (VMs or containers) between nodes.
-
-Depending on your environment, provisioning strategy, and performance goals, you can choose between:
-
-| Mode | Description | Typical Use Case |
-|------|--------------|------------------|
-| `used` | Uses the *actual runtime resource usage* (e.g. CPU, memory, disk). | Dynamic or lab environments with frequent workload changes and tolerance for overprovisioning. |
-| `assigned` | Uses the *statically defined resource allocations* from guest configurations. | Production or SLA-driven clusters that require guaranteed resources and predictable performance. |
-| `psi` | Uses Linux *Pressure Stall Information (PSI)* metrics to evaluate real system contention and pressure. | Advanced clusters that require pressure-aware decisions for proactive rebalancing. |
-
-### 10.1 Used Resources
-When **mode: `used`** is configured, ProxLB evaluates the *real usage metrics* of guest objects (VMs and CTs).
-It collects the current CPU, memory, and disk usage directly from the Proxmox API to determine the *actual consumption* of each guest and node.
-
-This mode is ideal for **dynamic environments** where workloads frequently change and **overprovisioning is acceptable**. It provides the most reactive balancing behavior, since decisions are based on live usage instead of static assignment.
-
-Typical scenarios include:
-- Production environments to distribute workloads across the nodes.
-- Test or development clusters with frequent VM changes.
-- Clusters where resource spikes are short-lived.
-- Environments where slight resource contention is tolerable.
-
-#### Example Configuration
-```yaml
-balancing:
-  mode: used
-```
-
-### 10.2 Assigned Resources
-When **mode: `assigned`** is configured, ProxLB evaluates the *provisioned or allocated resources* of each guest (VM or CT) instead of their runtime usage.
-It uses data such as **CPU cores**, **memory limits**, and **disk allocations** defined in Proxmox to calculate how much of each node’s capacity is reserved.
-
-This mode is ideal for **production clusters** where:
-- Overcommitment is *not allowed or only minimally tolerated*.
-- Each node’s workload is planned based on the assigned capacities.
-- Administrators want predictable resource distribution aligned with provisioning policies.
-
-Unlike the `used` mode, `assigned` focuses purely on the *declared configuration* of guests and remains stable even if actual usage varies temporarily.
-
-Typical scenarios include:
-- Enterprise environments with SLA or QoS requirements.
-- Clusters where workloads are sized deterministically.
-- Situations where consistent node utilization and capacity awareness are crucial.
-
-#### Example Configuration
-```yaml
-balancing:
-  mode: assigned
-```
-
-### 10.3 Pressure (PSI) based Resources
-> [!IMPORTANT]
-> PSI based balancing is still in beta! If you find any bugs, please raise an issue including metrics of all nodes and affected guests. You can provide metrics directly from PVE or Grafana (via node_exporter or pve_exporter).
-
-When **mode: `psi`** is configured, ProxLB uses the **Linux Pressure Stall Information (PSI)** interface to measure the *real-time pressure* on system resources such as **CPU**, **memory**, and **disk I/O**.
-Unlike the `used` or `assigned` modes, which rely on static or average metrics, PSI provides *direct insight into how often and how long tasks are stalled* because of insufficient resources.
-
-This enables ProxLB to make **proactive balancing decisions** — moving workloads *before* performance degradation becomes visible to the user.
-
-**IMPORTANT**: Predicting distributing workloads is dangerous and might not result into the expected state. Therefore, ProxLB migrates only a single instance each 60 minutes to obtain new real-metrics and to validate if further changes are required. Keep in mind, that migrations are also costly and should be avoided as much as possible.
-
-PSI metrics are available for both **nodes** and **guest objects**, allowing fine-grained balancing decisions:
-- **Node-level PSI:** Detects cluster nodes under systemic load or contention.
-- **Guest-level PSI:** Identifies individual guests suffering from memory, CPU, or I/O stalls.
-
-### PSI Metrics Explained
-Each monitored resource defines three pressure thresholds:
-| Key | Description |
-|-----|--------------|
-| `pressure_some` | Indicates partial stall conditions where some tasks are waiting for a resource. |
-| `pressure_full` | Represents complete stall conditions where *all* tasks are blocked waiting for a resource. |
-| `pressure_spikes` | Defines short-term burst conditions that may signal saturation spikes. |
-
-These thresholds are expressed in **percentages** and represent how much time the kernel reports stalls over specific averaging windows (e.g. 5s, 10s, 60s).
-
-### Example Configuration
-
-```yaml
-balancing:
-  mode: psi
-  psi:
-    nodes:
-      memory:
-        pressure_full: 0.20
-        pressure_some: 0.20
-        pressure_spikes: 1.00
-      cpu:
-        pressure_full: 0.20
-        pressure_some: 0.20
-        pressure_spikes: 1.00
-      disk:
-        pressure_full: 0.20
-        pressure_some: 0.20
-        pressure_spikes: 1.00
-    guests:
-      memory:
-        pressure_full: 0.20
-        pressure_some: 0.20
-        pressure_spikes: 1.00
-      cpu:
-        pressure_full: 0.20
-        pressure_some: 0.20
-        pressure_spikes: 1.00
-      disk:
-        pressure_full: 0.20
-        pressure_some: 0.20
-        pressure_spikes: 1.00
-```
+This mode gives you a more stable setup by always keeping a minimum number of nodes ready while still adjusting the rest of the cluster based on resource usage, but in a controlled and predictable way.

helm/proxlb/Chart.yaml

@@ -1,6 +0,0 @@
-apiVersion: v2
-name: proxlb
-description: A Helm chart for self-hosted ProxLB
-type: application
-version: "1.1.10"
-appVersion: "v1.1.10"

helm/proxlb/templates/_helpers.yaml

@@ -1,13 +0,0 @@
-{{- define "proxlb.fullname" -}}
-{{- printf "%s-%s" .Release.Name .Chart.Name | trunc 63 | trimSuffix "-" -}}
-{{- end -}}
-
-{{ define "proxlb.labels" }}
-app.kubernetes.io/name: {{ .Release.Name }}
-app.kubernetes.io/managed-by: Helm
-app.kubernetes.io/version: {{ .Chart.AppVersion }}
-app.kubernetes.io/component: proxlb
-{{- if .Values.labels }}
-{{ toYaml .Values.labels }}
-{{- end }}
-{{ end }}

helm/proxlb/templates/configmap.yaml

@@ -1,11 +0,0 @@
-{{- if .Values.configmap.create }}
-apiVersion: v1
-kind: ConfigMap
-metadata:
-  name: proxlb-config
-  labels:
-    {{- include "proxlb.labels" . | nindent 4 }}
-data:
-  proxlb.yaml: |
-{{ toYaml .Values.configmap.config | indent 4 }}
-{{ end }}

helm/proxlb/templates/deployment.yaml

@@ -1,44 +0,0 @@
-apiVersion: apps/v1
-kind: Deployment
-metadata:
-  name: {{ .Release.Name }}
-  labels:
-    {{- include "proxlb.labels" . | nindent 4 }}
-spec:
-  replicas: 1 # Number of replicas cannot be more than 1
-  selector:
-    matchLabels:
-      {{- include "proxlb.labels" . | nindent 6 }}
-  template:
-    metadata:
-      labels:
-        {{- include "proxlb.labels" . | nindent 8 }}
-    spec:
-      {{- with .Values.image.imagePullSecrets }}
-      imagePullSecrets:
-        {{- toYaml . | nindent 8 }}
-      {{- end }}
-      # not interacting with the k8s cluster
-      automountServiceAccountToken: False
-      containers:
-      - name: proxlb
-        image: "{{ .Values.image.registry }}/{{ .Values.image.repository }}:{{ .Values.image.tag }}"
-        imagePullPolicy: {{ .Values.image.pullPolicy }}
-        args:
-        {{- if .Values.extraArgs.dryRun }}
-          - --dry-run
-        {{- end }}
-        volumeMounts:
-          - name: config
-            mountPath: /etc/proxlb/proxlb.yaml
-            subPath: proxlb.yaml
-        {{ if .Values.resources }}
-        resources:
-          {{ with .Values.resources }}
-            {{ toYaml . | nindent 10 }}
-          {{ end }}
-        {{ end }}
-      volumes:
-        - name: config
-          configMap:
-            name: proxlb-config

helm/proxlb/values.yaml

@@ -1,61 +0,0 @@
-image:
-  registry: cr.gyptazy.com
-  repository: proxlb/proxlb
-  tag: v1.1.10
-  pullPolicy: IfNotPresent
-  imagePullSecrets: [ ]
-
-resources:
-  limits:
-    cpu: "1000m"
-    memory: "2Gi"
-  requests:
-    cpu: "100m"
-    memory: "100Mi"
-
-labels: {}
-
-extraArgs:
-  dryRun: false
-
-configmap:
-  create: true
-  config:
-    proxmox_api:
-      hosts: []
-#Can be either a user or a token
-#      user: ""
-#      pass: ""
-#      token_id: ""
-#      token_secret: ""
-      ssl_verification: True
-      timeout: 10
-    proxmox_cluster:
-      maintenance_nodes: [ ]
-      ignore_nodes: [ ]
-      overprovisioning: True
-    balancing:
-      enable: True
-      enforce_affinity: False
-      parallel: False
-      # If running parallel job, you can define
-      # the amount of prallel jobs (default: 5)
-      parallel_jobs: 1
-      live: True
-      with_local_disks: True
-      with_conntrack_state: True
-      balance_types: [ 'vm', 'ct' ]
-      max_job_validation: 1800
-      balanciness: 5
-      method: memory
-      mode: used
-    service:
-      daemon: True
-      schedule:
-        interval: 12
-        format: "hours"
-      delay:
-        enable: False
-        time: 1
-        format: "hours"
-      log_level: INFO

misc/01-replace-version.sh

@@ -1,13 +1,6 @@
 #!/usr/bin/env bash
-VERSION="1.1.9.1"
+VERSION="1.1.2b"
 
-# ProxLB
 sed -i "s/^__version__ = .*/__version__ = \"$VERSION\"/" "proxlb/utils/version.py"
 sed -i "s/version=\"[0-9]*\.[0-9]*\.[0-9]*\"/version=\"$VERSION\"/" setup.py
-
-# Helm Chart
-sed -i "s/^version: .*/version: \"$VERSION\"/" helm/proxlb/Chart.yaml
-sed -i "s/^appVersion: .*/appVersion: \"v$VERSION\"/" helm/proxlb/Chart.yaml
-sed -i "s/^tag: .*/tag: \"v$VERSION\"/" helm/proxlb/values.yaml
-
 echo "OK: Versions have been sucessfully set to $VERSION"

proxlb/main.py

@@ -13,18 +13,16 @@ __license__ = "GPL-3.0"
 
 
 import logging
-import signal
 from utils.logger import SystemdLogger
 from utils.cli_parser import CliParser
 from utils.config_parser import ConfigParser
 from utils.proxmox_api import ProxmoxApi
+from models.dpm import DPM
 from models.nodes import Nodes
-from models.features import Features
 from models.guests import Guests
 from models.groups import Groups
 from models.calculations import Calculations
 from models.balancing import Balancing
-from models.pools import Pools
 from utils.helper import Helper
 
 
@@ -35,10 +33,6 @@ def main():
     # Initialize logging handler
     logger = SystemdLogger(level=logging.INFO)
 
-    # Initialize handlers
-    signal.signal(signal.SIGHUP, Helper.handler_sighup)
-    signal.signal(signal.SIGINT, Helper.handler_sigint)
-
     # Parses arguments passed from the CLI
     cli_parser = CliParser()
     cli_args = cli_parser.parse_args()
@@ -51,9 +45,6 @@ def main():
     # Update log level from config and fallback to INFO if not defined
     logger.set_log_level(proxlb_config.get('service', {}).get('log_level', 'INFO'))
 
-    # Validate of an optional service delay
-    Helper.get_service_delay(proxlb_config)
-
     # Connect to Proxmox API & create API object
     proxmox_api = ProxmoxApi(proxlb_config)
 
@@ -61,45 +52,35 @@ def main():
     proxlb_config["proxmox_api"]["pass"] = "********"
 
     while True:
-
-        # Validate if reload signal was sent during runtime
-        # and reload the ProxLB configuration and adjust log level
-        if Helper.proxlb_reload:
-            logger.info("Reloading ProxLB configuration.")
-            proxlb_config = config_parser.get_config()
-            logger.set_log_level(proxlb_config.get('service', {}).get('log_level', 'INFO'))
-            Helper.proxlb_reload = False
-
         # Get all required objects from the Proxmox cluster
         meta = {"meta": proxlb_config}
-        nodes = Nodes.get_nodes(proxmox_api, proxlb_config)
-        pools = Pools.get_pools(proxmox_api)
-        guests = Guests.get_guests(proxmox_api, pools, nodes, meta, proxlb_config)
+        nodes, cluster = Nodes.get_nodes(proxmox_api, proxlb_config)
+        guests = Guests.get_guests(proxmox_api, nodes, meta)
         groups = Groups.get_groups(guests, nodes)
 
         # Merge obtained objects from the Proxmox cluster for further usage
-        proxlb_data = {**meta, **nodes, **guests, **pools, **groups}
+        proxlb_data = {**meta, **cluster, **nodes, **guests, **groups}
         Helper.log_node_metrics(proxlb_data)
 
-        # Validate usable features by PVE versions
-        Features.validate_available_features(proxlb_data)
+        # Evaluate the dynamic power management for nodes in the clustet
+        DPM(proxlb_data)
 
         # Update the initial node resource assignments
         # by the previously created groups.
         Calculations.set_node_assignments(proxlb_data)
-        Calculations.set_node_hot(proxlb_data)
-        Calculations.set_guest_hot(proxlb_data)
         Calculations.get_most_free_node(proxlb_data, cli_args.best_node)
-        Calculations.validate_affinity_map(proxlb_data)
         Calculations.relocate_guests_on_maintenance_nodes(proxlb_data)
         Calculations.get_balanciness(proxlb_data)
         Calculations.relocate_guests(proxlb_data)
         Helper.log_node_metrics(proxlb_data, init=False)
 
-        # Perform balancing actions via Proxmox API
-        if proxlb_data["meta"]["balancing"].get("enable", False):
-            if not cli_args.dry_run:
-                Balancing(proxmox_api, proxlb_data)
+        # Perform balancing
+        if not cli_args.dry_run or not proxlb_data["meta"]["balancing"].get("enable", False):
+            Balancing(proxmox_api, proxlb_data)
+
+        # Perform DPM
+        if not cli_args.dry_run:
+            DPM.dpm_shutdown_nodes(proxmox_api, proxlb_data)
 
         # Validate if the JSON output should be
         # printed to stdout

proxlb/models/balancing.py

@@ -12,7 +12,6 @@ __license__ = "GPL-3.0"
 
 import proxmoxer
 import time
-from itertools import islice
 from utils.logger import SystemdLogger
 from typing import Dict, Any
 
@@ -49,81 +48,34 @@ class Balancing:
         Initializes the Balancing class with the provided ProxLB data.
 
         Args:
-            proxmox_api (object): The Proxmox API client instance used to interact with the Proxmox cluster.
-            proxlb_data (dict): A dictionary containing data related to the ProxLB load balancing configuration.
+            proxlb_data (dict): The data required for balancing VMs and CTs.
         """
-        def chunk_dict(data, size):
-            """
-            Splits a dictionary into chunks of a specified size.
-            Args:
-                data (dict): The dictionary to be split into chunks.
-                size (int): The size of each chunk.
-            Yields:
-                dict: A chunk of the original dictionary with the specified size.
-            """
-            logger.debug("Starting: chunk_dict.")
-            it = iter(data.items())
-            for chunk in range(0, len(data), size):
-                yield dict(islice(it, size))
+        for guest_name, guest_meta in proxlb_data["guests"].items():
 
-        # Validate if balancing should be performed in parallel or sequentially.
-        # If parallel balancing is enabled, set the number of parallel jobs.
-        parallel_jobs = proxlb_data["meta"]["balancing"].get("parallel_jobs", 5)
-        if not proxlb_data["meta"]["balancing"].get("parallel", False):
-            parallel_jobs = 1
-            logger.debug("Balancing: Parallel balancing is disabled. Running sequentially.")
-        else:
-            logger.debug(f"Balancing: Parallel balancing is enabled. Running with {parallel_jobs} parallel jobs.")
+            # Check if the guest's target is not the same as the current node
+            if guest_meta["node_current"] != guest_meta["node_target"]:
+                # Check if the guest is not ignored and perform the balancing
+                # operation based on the guest type
+                if not guest_meta["ignore"]:
+                    guest_id = guest_meta["id"]
+                    guest_node_current = guest_meta["node_current"]
+                    guest_node_target = guest_meta["node_target"]
 
-        for chunk in chunk_dict(proxlb_data["guests"], parallel_jobs):
-            jobs_to_wait = []
+                    # VM Balancing
+                    if guest_meta["type"] == "vm":
+                        self.exec_rebalancing_vm(proxmox_api, proxlb_data, guest_name)
 
-            for guest_name, guest_meta in chunk.items():
-
-                # Check if the guest's target is not the same as the current node
-                if guest_meta["node_current"] != guest_meta["node_target"]:
-
-                    # Check if the guest is not ignored and perform the balancing
-                    # operation based on the guest type
-                    if not guest_meta["ignore"]:
-                        job_id = None
-
-                        # VM Balancing
-                        if guest_meta["type"] == "vm":
-                            if 'vm' in proxlb_data["meta"]["balancing"].get("balance_types", []):
-                                logger.debug(f"Balancing: Balancing for guest {guest_name} of type VM started.")
-                                job_id = self.exec_rebalancing_vm(proxmox_api, proxlb_data, guest_name)
-                            else:
-                                logger.debug(
-                                    f"Balancing: Balancing for guest {guest_name} will not be performed. "
-                                    "Guest is of type VM which is not included in allowed balancing types.")
-
-                        # CT Balancing
-                        elif guest_meta["type"] == "ct":
-                            if 'ct' in proxlb_data["meta"]["balancing"].get("balance_types", []):
-                                logger.debug(f"Balancing: Balancing for guest {guest_name} of type CT started.")
-                                job_id = self.exec_rebalancing_ct(proxmox_api, proxlb_data, guest_name)
-                            else:
-                                logger.debug(
-                                    f"Balancing: Balancing for guest {guest_name} will not be performed. "
-                                    "Guest is of type CT which is not included in allowed balancing types.")
-
-                        # Just in case we get a new type of guest in the future
-                        else:
-                            logger.critical(f"Balancing: Got unexpected guest type: {guest_meta['type']}. Cannot proceed guest: {guest_meta['name']}.")
-
-                        if job_id:
-                            jobs_to_wait.append((guest_name, guest_meta["node_current"], job_id))
+                    # CT Balancing
+                    elif guest_meta["type"] == "ct":
+                        self.exec_rebalancing_ct(proxmox_api, proxlb_data, guest_name)
 
+                    # Just in case we get a new type of guest in the future
                     else:
-                        logger.debug(f"Balancing: Guest {guest_name} is ignored and will not be rebalanced.")
+                        logger.critical(f"Balancing: Got unexpected guest type: {guest_meta['type']}. Cannot proceed guest: {guest_meta['name']}.")
                 else:
-                    logger.debug(f"Balancing: Guest {guest_name} is already on the target node {guest_meta['node_target']} and will not be rebalanced.")
-
-            # Wait for all jobs in the current chunk to complete
-            for guest_name, node, job_id in jobs_to_wait:
-                if job_id:
-                    self.get_rebalancing_job_status(proxmox_api, proxlb_data, guest_name, node, job_id)
+                    logger.debug(f"Balancing: Guest {guest_name} is ignored and will not be rebalanced.")
+            else:
+                logger.debug(f"Balancing: Guest {guest_name} is already on the target node {guest_meta['node_target']} and will not be rebalanced.")
 
     def exec_rebalancing_vm(self, proxmox_api: any, proxlb_data: Dict[str, Any], guest_name: str) -> None:
         """
@@ -144,7 +96,6 @@ class Balancing:
         guest_id = proxlb_data["guests"][guest_name]["id"]
         guest_node_current = proxlb_data["guests"][guest_name]["node_current"]
         guest_node_target = proxlb_data["guests"][guest_name]["node_target"]
-        job_id = None
 
         if proxlb_data["meta"]["balancing"].get("live", True):
             online_migration = 1
@@ -157,25 +108,19 @@ class Balancing:
             with_local_disks = 0
 
         migration_options = {
-            'target': guest_node_target,
+            'target': {guest_node_target},
             'online': online_migration,
-            'with-local-disks': with_local_disks,
+            'with-local-disks': with_local_disks
         }
 
-        # Conntrack state aware migrations are not supported in older
-        # PVE versions, so we should not add it by default.
-        if proxlb_data["meta"]["balancing"].get("with_conntrack_state", True):
-            migration_options['with-conntrack-state'] = 1
-
         try:
-            logger.info(f"Balancing: Starting to migrate VM guest {guest_name} from {guest_node_current} to {guest_node_target}.")
+            logger.debug(f"Balancing: Starting to migrate guest {guest_name} of type VM.")
             job_id = proxmox_api.nodes(guest_node_current).qemu(guest_id).migrate().post(**migration_options)
+            self.get_rebalancing_job_status(proxmox_api, proxlb_data, guest_name, guest_node_current, job_id)
         except proxmoxer.core.ResourceException as proxmox_api_error:
             logger.critical(f"Balancing: Failed to migrate guest {guest_name} of type VM due to some Proxmox errors. Please check if resource is locked or similar.")
             logger.debug(f"Balancing: Failed to migrate guest {guest_name} of type VM due to some Proxmox errors: {proxmox_api_error}")
-
         logger.debug("Finished: exec_rebalancing_vm.")
-        return job_id
 
     def exec_rebalancing_ct(self, proxmox_api: any, proxlb_data: Dict[str, Any], guest_name: str) -> None:
         """
@@ -196,17 +141,15 @@ class Balancing:
         guest_id = proxlb_data["guests"][guest_name]["id"]
         guest_node_current = proxlb_data["guests"][guest_name]["node_current"]
         guest_node_target = proxlb_data["guests"][guest_name]["node_target"]
-        job_id = None
 
         try:
-            logger.info(f"Balancing: Starting to migrate CT guest {guest_name} from {guest_node_current} to {guest_node_target}.")
+            logger.debug(f"Balancing: Starting to migrate guest {guest_name} of type CT.")
             job_id = proxmox_api.nodes(guest_node_current).lxc(guest_id).migrate().post(target=guest_node_target, restart=1)
+            self.get_rebalancing_job_status(proxmox_api, proxlb_data, guest_name, guest_node_current, job_id)
         except proxmoxer.core.ResourceException as proxmox_api_error:
             logger.critical(f"Balancing: Failed to migrate guest {guest_name} of type CT due to some Proxmox errors. Please check if resource is locked or similar.")
             logger.debug(f"Balancing: Failed to migrate guest {guest_name} of type CT due to some Proxmox errors: {proxmox_api_error}")
-
         logger.debug("Finished: exec_rebalancing_ct.")
-        return job_id
 
     def get_rebalancing_job_status(self, proxmox_api: any, proxlb_data: Dict[str, Any], guest_name: str, guest_current_node: str, job_id: int, retry_counter: int = 1) -> bool:
         """
@@ -224,32 +167,35 @@ class Balancing:
             bool: True if the job completed successfully, False otherwise.
         """
         logger.debug("Starting: get_rebalancing_job_status.")
-        job = proxmox_api.nodes(guest_current_node).tasks(job_id).status().get()
+        # Parallel migrations can take a huge time and create a higher load, if not defined by an
+        # operator we will use a sequential mode by default
+        if not proxlb_data["meta"]["balancing"].get("parallel", False):
+            job = proxmox_api.nodes(guest_current_node).tasks(job_id).status().get()
 
-        # Watch job id until it finalizes
-        if job["status"] == "running":
-            # Do not hammer the API while
-            # watching the job status
-            time.sleep(10)
-            retry_counter += 1
+            # Watch job id until it finalizes
+            if job["status"] == "running":
+                # Do not hammer the API while
+                # watching the job status
+                time.sleep(10)
+                retry_counter += 1
 
-            # Run recursion until we hit the soft-limit of maximum migration time for a guest
-            if retry_counter < proxlb_data["meta"]["balancing"].get("max_job_validation", 1800):
-                logger.debug(f"Balancing: Job ID {job_id} (guest: {guest_name}) for migration is still running... (Run: {retry_counter})")
-                self.get_rebalancing_job_status(proxmox_api, proxlb_data, guest_name, guest_current_node, job_id, retry_counter)
-            else:
-                logger.warning(f"Balancing: Job ID {job_id} (guest: {guest_name}) for migration took too long. Please check manually.")
-                logger.debug("Finished: get_rebalancing_job_status.")
-                return False
+                # Run recursion until we hit the soft-limit of maximum migration time for a guest
+                if retry_counter < proxlb_data["meta"]["balancing"].get("max_job_validation", 1800):
+                    logger.debug(f"Balancing: Job ID {job_id} (guest: {guest_name}) for migration is still running... (Run: {retry_counter})")
+                    self.get_rebalancing_job_status(proxmox_api, proxlb_data, guest_name, guest_current_node, job_id, retry_counter)
+                else:
+                    logger.warning(f"Balancing: Job ID {job_id} (guest: {guest_name}) for migration took too long. Please check manually.")
+                    logger.debug("Finished: get_rebalancing_job_status.")
+                    return False
 
-        # Validate job output for errors when finished
-        if job["status"] == "stopped":
+            # Validate job output for errors when finished
+            if job["status"] == "stopped":
 
-            if job["exitstatus"] == "OK":
-                logger.debug(f"Balancing: Job ID {job_id} (guest: {guest_name}) was successfully.")
-                logger.debug("Finished: get_rebalancing_job_status.")
-                return True
-            else:
-                logger.critical(f"Balancing: Job ID {job_id} (guest: {guest_name}) went into an error! Please check manually.")
-                logger.debug("Finished: get_rebalancing_job_status.")
-                return False
+                if job["exitstatus"] == "OK":
+                    logger.debug(f"Balancing: Job ID {job_id} (guest: {guest_name}) was successfully.")
+                    logger.debug("Finished: get_rebalancing_job_status.")
+                    return True
+                else:
+                    logger.critical(f"Balancing: Job ID {job_id} (guest: {guest_name}) went into an error! Please check manually.")
+                    logger.debug("Finished: get_rebalancing_job_status.")
+                    return False

proxlb/models/calculations.py

@@ -80,9 +80,8 @@ class Calculations:
 
             for guest_name in group_meta["guests"]:
                 guest_node_current = proxlb_data["guests"][guest_name]["node_current"]
-                # Update resource assignments
+                # Update Hardware assignments
                 # Update assigned values for the current node
-                logger.debug(f"set_node_assignment of guest {guest_name} on node {guest_node_current} with cpu_total: {proxlb_data['guests'][guest_name]['cpu_total']}, memory_total: {proxlb_data['guests'][guest_name]['memory_total']}, disk_total: {proxlb_data['guests'][guest_name]['disk_total']}.")
                 proxlb_data["nodes"][guest_node_current]["cpu_assigned"] += proxlb_data["guests"][guest_name]["cpu_total"]
                 proxlb_data["nodes"][guest_node_current]["memory_assigned"] += proxlb_data["guests"][guest_name]["memory_total"]
                 proxlb_data["nodes"][guest_node_current]["disk_assigned"] += proxlb_data["guests"][guest_name]["disk_total"]
@@ -93,83 +92,6 @@ class Calculations:
 
         logger.debug("Finished: set_node_assignments.")
 
-    def set_node_hot(proxlb_data: Dict[str, Any]) -> Dict[str, Any]:
-        """
-        Evaluates node 'full' pressure metrics for memory, cpu, and io
-        against defined thresholds and sets <metric>_pressure_hot = True
-        when a node is considered HOT.
-
-        Returns the modified proxlb_data dict.
-        """
-        logger.debug("Starting: set_node_hot.")
-        balancing_cfg = proxlb_data.get("meta", {}).get("balancing", {})
-        thresholds = balancing_cfg.get("psi_thresholds", balancing_cfg.get("psi", {}).get("nodes", {}))
-        nodes = proxlb_data.get("nodes", {})
-
-        for node_name, node in nodes.items():
-
-            if node.get("maintenance"):
-                continue
-
-            if node.get("ignore"):
-                continue
-
-            # PSI metrics are only availavble on Proxmox VE 9.0 and higher.
-            if proxlb_data["meta"]["balancing"].get("mode", "used") == "psi":
-
-                if tuple(map(int, proxlb_data["nodes"][node["name"]]["pve_version"].split('.'))) < tuple(map(int, "9.0".split('.'))):
-                    logger.critical(f"Proxmox node {node['name']} runs Proxmox VE version {proxlb_data['nodes'][node['name']]['pve_version']}."
-                                    " PSI metrics require Proxmox VE 9.0 or higher. Balancing deactivated!")
-
-            for metric, threshold in thresholds.items():
-                pressure_full = node.get(f"{metric}_pressure_full_percent", 0.0)
-                pressure_some = node.get(f"{metric}_pressure_some_percent", 0.0)
-                pressure_spikes = node.get(f"{metric}_pressure_full_spikes_percent", 0.0)
-                is_hot = (pressure_full >= threshold["pressure_full"] and pressure_some >= threshold["pressure_some"]) or (pressure_spikes >= threshold["pressure_spikes"])
-
-                if is_hot:
-                    logger.debug(f"Set node {node['name']} as hot based on {metric} pressure metrics.")
-                    proxlb_data["nodes"][node["name"]][f"{metric}_pressure_hot"] = True
-                    proxlb_data["nodes"][node["name"]][f"pressure_hot"] = True
-                else:
-                    logger.debug(f"Node {node['name']} is not hot based on {metric} pressure metrics.")
-
-        logger.debug("Finished: set_node_hot.")
-        return proxlb_data
-
-    def set_guest_hot(proxlb_data: Dict[str, Any]) -> Dict[str, Any]:
-        """
-        Evaluates guest 'full' pressure metrics for memory, cpu, and io
-        against defined thresholds and sets <metric>_pressure_hot = True
-        when a guest is considered HOT.
-
-        Returns the modified proxlb_data dict.
-        """
-        logger.debug("Starting: set_guest_hot.")
-        balancing_cfg = proxlb_data.get("meta", {}).get("balancing", {})
-        thresholds = balancing_cfg.get("psi_thresholds", balancing_cfg.get("psi", {}).get("guests", {}))
-        guests = proxlb_data.get("guests", {})
-
-        for guest_name, guest in guests.items():
-            if guest.get("ignore"):
-                continue
-
-            for metric, threshold in thresholds.items():
-                pressure_full = guest.get(f"{metric}_pressure_full_percent", 0.0)
-                pressure_some = guest.get(f"{metric}_pressure_some_percent", 0.0)
-                pressure_spikes = guest.get(f"{metric}_pressure_full_spikes_percent", 0.0)
-                is_hot = (pressure_full >= threshold["pressure_full"] and pressure_some >= threshold["pressure_some"]) or (pressure_spikes >= threshold["pressure_spikes"])
-
-                if is_hot:
-                    logger.debug(f"Set guest {guest['name']} as hot based on {metric} pressure metrics.")
-                    proxlb_data["guests"][guest["name"]][f"{metric}_pressure_hot"] = True
-                    proxlb_data["guests"][guest["name"]][f"pressure_hot"] = True
-                else:
-                    logger.debug(f"guest {guest['name']} is not hot based on {metric} pressure metrics.")
-
-        logger.debug("Finished: set_guest_hot.")
-        return proxlb_data
-
     @staticmethod
     def get_balanciness(proxlb_data: Dict[str, Any]) -> Dict[str, Any]:
         """
@@ -190,66 +112,7 @@ class Calculations:
             method = proxlb_data["meta"]["balancing"].get("method", "memory")
             mode = proxlb_data["meta"]["balancing"].get("mode", "used")
             balanciness = proxlb_data["meta"]["balancing"].get("balanciness", 10)
-
-            if mode == "assigned":
-                method_value = [node_meta[f"{method}_{mode}_percent"] for node_meta in proxlb_data["nodes"].values()]
-
-                if proxlb_data["meta"]["balancing"].get(f"{method}_threshold", None):
-                    threshold = proxlb_data["meta"]["balancing"].get(f"{method}_threshold")
-                    highest_usage_node = max(proxlb_data["nodes"].values(), key=lambda x: x[f"{method}_{mode}_percent"])
-                    highest_node_value = highest_usage_node[f"{method}_{mode}_percent"]
-
-                    if highest_node_value >= threshold:
-                        logger.debug(f"Guest balancing is required. Highest {method} usage node {highest_usage_node['name']} is above the defined threshold of {threshold}% with a value of {highest_node_value}%.")
-                        proxlb_data["meta"]["balancing"]["balance"] = True
-                    else:
-                        logger.debug(f"Guest balancing is ok. Highest {method} usage node {highest_usage_node['name']} is below the defined threshold of {threshold}% with a value of {highest_node_value}%.")
-                        proxlb_data["meta"]["balancing"]["balance"] = False
-
-                else:
-                    logger.debug(f"No {method} threshold defined for balancing. Skipping threshold check.")
-
-            elif mode == "used":
-                method_value = [node_meta[f"{method}_{mode}_percent"] for node_meta in proxlb_data["nodes"].values()]
-
-                if proxlb_data["meta"]["balancing"].get(f"{method}_threshold", None):
-                    threshold = proxlb_data["meta"]["balancing"].get(f"{method}_threshold")
-                    highest_usage_node = max(proxlb_data["nodes"].values(), key=lambda x: x[f"{method}_{mode}_percent"])
-                    highest_node_value = highest_usage_node[f"{method}_{mode}_percent"]
-
-                    if highest_node_value >= threshold:
-                        logger.debug(f"Guest balancing is required. Highest {method} usage node {highest_usage_node['name']} is above the defined threshold of {threshold}% with a value of {highest_node_value}%.")
-                        proxlb_data["meta"]["balancing"]["balance"] = True
-                    else:
-                        logger.debug(f"Guest balancing is ok. Highest {method} usage node {highest_usage_node['name']} is below the defined threshold of {threshold}% with a value of {highest_node_value}%.")
-                        proxlb_data["meta"]["balancing"]["balance"] = False
-
-                else:
-                    logger.debug(f"No {method} threshold defined for balancing. Skipping threshold check.")
-
-            elif mode == "psi":
-                method_value = [node_meta[f"{method}_pressure_full_spikes_percent"] for node_meta in proxlb_data["nodes"].values()]
-                any_node_hot = any(node.get(f"{method}_pressure_hot", False) for node in proxlb_data["nodes"].values())
-                any_guest_hot = any(node.get(f"{method}_pressure_hot", False) for node in proxlb_data["guests"].values())
-
-                if any_node_hot:
-                    logger.debug(f"Guest balancing is required. A node is marked as HOT based on {method} pressure metrics.")
-                    proxlb_data["meta"]["balancing"]["balance"] = True
-                else:
-                    logger.debug(f"Guest balancing is ok. No node is marked as HOT based on {method} pressure metrics.")
-
-                if any_guest_hot:
-                    logger.debug(f"Guest balancing is required. A guest is marked as HOT based on {method} pressure metrics.")
-                    proxlb_data["meta"]["balancing"]["balance"] = True
-                else:
-                    logger.debug(f"Guest balancing is ok. No guest is marked as HOT based on {method} pressure metrics.")
-
-                return proxlb_data
-
-            else:
-                logger.critical(f"Unknown balancing mode: {mode} provided. Cannot get balanciness.")
-                sys.exit(1)
-
+            method_value = [node_meta[f"{method}_{mode}_percent"] for node_meta in proxlb_data["nodes"].values()]
             method_value_highest = max(method_value)
             method_value_lowest = min(method_value)
 
@@ -265,7 +128,7 @@ class Calculations:
         logger.debug("Finished: get_balanciness.")
 
     @staticmethod
-    def get_most_free_node(proxlb_data: Dict[str, Any], return_node: bool = False, guest_node_relation_list: list = []) -> Dict[str, Any]:
+    def get_most_free_node(proxlb_data: Dict[str, Any], return_node: bool = False) -> Dict[str, Any]:
         """
         Get the name of the Proxmox node in the cluster with the most free resources based on
         the user defined method (e.g.: memory) and mode (e.g.: used).
@@ -274,8 +137,6 @@ class Calculations:
             proxlb_data (Dict[str, Any]): The data holding all content of all objects.
             return_node (bool): The indicator to simply return the best node for further
                                 assignments.
-            guest_node_relation_list (list): A list of nodes that have a tag on the given
-                                             guest relationship for pinning.
 
         Returns:
             Dict[str, Any]: Updated meta data section of the node with the most free resources that should
@@ -284,34 +145,11 @@ class Calculations:
         logger.debug("Starting: get_most_free_node.")
         proxlb_data["meta"]["balancing"]["balance_next_node"] = ""
 
-        # Filter and exclude nodes that are in maintenance mode
+        # Do not include nodes that are marked in 'maintenance'
         filtered_nodes = [node for node in proxlb_data["nodes"].values() if not node["maintenance"]]
-
-        # Filter and include nodes that given by a relationship between guest and node. This is only
-        # used if the guest has a relationship to a node defined by "pin" tags.
-        if len(guest_node_relation_list) > 0:
-            filtered_nodes = [node for node in proxlb_data["nodes"].values() if node["name"] in guest_node_relation_list]
-
-        # Filter by the defined methods and modes for balancing
         method = proxlb_data["meta"]["balancing"].get("method", "memory")
         mode = proxlb_data["meta"]["balancing"].get("mode", "used")
-
-        if mode == "assigned":
-            logger.debug(f"Get best node for balancing by assigned {method} resources.")
-            lowest_usage_node = min(filtered_nodes, key=lambda x: x[f"{method}_{mode}_percent"])
-
-        elif mode == "used":
-            logger.debug(f"Get best node for balancing by used {method} resources.")
-            lowest_usage_node = min(filtered_nodes, key=lambda x: x[f"{method}_{mode}_percent"])
-
-        elif mode == "psi":
-            logger.debug(f"Get best node for balancing by pressure of {method} resources.")
-            lowest_usage_node = min(filtered_nodes, key=lambda x: x[f"{method}_pressure_full_spikes_percent"])
-
-        else:
-            logger.critical(f"Unknown balancing mode: {mode} provided. Cannot get best node.")
-            sys.exit(1)
-
+        lowest_usage_node = min(filtered_nodes, key=lambda x: x[f"{method}_{mode}_percent"])
         proxlb_data["meta"]["balancing"]["balance_reason"] = 'resources'
         proxlb_data["meta"]["balancing"]["balance_next_node"] = lowest_usage_node["name"]
 
@@ -324,7 +162,7 @@ class Calculations:
         logger.debug("Finished: get_most_free_node.")
 
     @staticmethod
-    def relocate_guests_on_maintenance_nodes(proxlb_data: Dict[str, Any]):
+    def relocate_guests_on_maintenance_nodes(proxlb_data: Dict[str, Any]) -> None:
         """
         Relocates guests that are currently on nodes marked for maintenance to
         nodes with the most available resources.
@@ -340,7 +178,7 @@ class Calculations:
         Returns:
         None
         """
-        logger.debug("Starting: relocate_guests_on_maintenance_nodes.")
+        logger.debug("Starting: get_most_free_node.")
         proxlb_data["meta"]["balancing"]["balance_next_guest"] = ""
 
         for guest_name in proxlb_data["groups"]["maintenance"]:
@@ -351,10 +189,10 @@ class Calculations:
             Calculations.update_node_resources(proxlb_data)
             logger.warning(f"Warning: Balancing may not be perfect because guest {guest_name} was located on a node which is in maintenance mode.")
 
-        logger.debug("Finished: relocate_guests_on_maintenance_nodes.")
+        logger.debug("Finished: get_most_free_node.")
 
     @staticmethod
-    def relocate_guests(proxlb_data: Dict[str, Any]):
+    def relocate_guests(proxlb_data: Dict[str, Any]) -> None:
         """
         Relocates guests within the provided data structure to ensure affinity groups are
         placed on nodes with the most free resources.
@@ -385,34 +223,15 @@ class Calculations:
                 Calculations.get_most_free_node(proxlb_data)
 
                 for guest_name in proxlb_data["groups"]["affinity"][group_name]["guests"]:
-                    mode = proxlb_data["meta"]["balancing"].get("mode", "used")
-
-                    if mode == 'psi':
-                        logger.debug(f"Evaluating guest relocation based on {mode} mode.")
-                        method = proxlb_data["meta"]["balancing"].get("method", "memory")
-                        processed_guests_psi = proxlb_data["meta"]["balancing"].setdefault("processed_guests_psi", [])
-                        unprocessed_guests_psi = [guest for guest in proxlb_data["guests"].values() if guest["name"] not in processed_guests_psi]
-
-                        # Filter by the defined methods and modes for balancing
-                        highest_usage_guest = max(unprocessed_guests_psi, key=lambda x: x[f"{method}_pressure_full_spikes_percent"])
-
-                        # Append guest to the psi based processed list of guests
-                        if highest_usage_guest["name"] == guest_name and guest_name not in proxlb_data["meta"]["balancing"]["processed_guests_psi"]:
-                            proxlb_data["meta"]["balancing"]["processed_guests_psi"].append(guest_name)
-                            proxlb_data["meta"]["balancing"]["balance_next_guest"] = guest_name
-
-                    else:
-                        logger.debug(f"Evaluating guest relocation based on {mode} mode.")
-                        proxlb_data["meta"]["balancing"]["balance_next_guest"] = guest_name
-
+                    proxlb_data["meta"]["balancing"]["balance_next_guest"] = guest_name
                     Calculations.val_anti_affinity(proxlb_data, guest_name)
-                    Calculations.val_node_relationships(proxlb_data, guest_name)
+                    Calculations.val_node_relationship(proxlb_data, guest_name)
                     Calculations.update_node_resources(proxlb_data)
 
         logger.debug("Finished: relocate_guests.")
 
     @staticmethod
-    def val_anti_affinity(proxlb_data: Dict[str, Any], guest_name: str):
+    def val_anti_affinity(proxlb_data: Dict[str, Any], guest_name: str) -> None:
         """
         Validates and assigns nodes to guests based on anti-affinity rules.
 
@@ -437,28 +256,23 @@ class Calculations:
             if guest_name in proxlb_data["groups"]["anti_affinity"][group_name]['guests'] and not proxlb_data["guests"][guest_name]["processed"]:
                 logger.debug(f"Anti-Affinity: Guest: {guest_name} is included in anti-affinity group: {group_name}.")
 
-                # Check if the group has only one member. If so skip new guest node assignment.
-                if proxlb_data["groups"]["anti_affinity"][group_name]["counter"] > 1:
-                    logger.debug(f"Anti-Affinity: Group has more than 1 member.")
-                    # Iterate over all available nodes
-                    for node_name in proxlb_data["nodes"].keys():
+                # Iterate over all available nodes
+                for node_name in proxlb_data["nodes"].keys():
 
-                        # Only select node if it was not used before and is not in a
-                        # maintenance mode. Afterwards, add it to the list of already
-                        # used nodes for the current anti-affinity group
-                        if node_name not in proxlb_data["groups"]["anti_affinity"][group_name]["used_nodes"]:
+                    # Only select node if it was not used before and is not in a
+                    # maintenance mode. Afterwards, add it to the list of already
+                    # used nodes for the current anti-affinity group
+                    if node_name not in proxlb_data["groups"]["anti_affinity"][group_name]["used_nodes"]:
 
-                            if not proxlb_data["nodes"][node_name]["maintenance"]:
-                                # If the node has not been used yet, we assign this node to the guest
-                                proxlb_data["meta"]["balancing"]["balance_next_node"] = node_name
-                                proxlb_data["groups"]["anti_affinity"][group_name]["used_nodes"].append(node_name)
-                                logger.debug(f"Node: {node_name} marked as used for anti-affinity group: {group_name} with guest {guest_name}")
-                                break
+                        if not proxlb_data["nodes"][node_name]["maintenance"]:
+                            # If the node has not been used yet, we assign this node to the guest
+                            proxlb_data["meta"]["balancing"]["balance_next_node"] = node_name
+                            proxlb_data["groups"]["anti_affinity"][group_name]["used_nodes"].append(node_name)
+                            logger.debug(f"Node: {node_name} marked as used for anti-affinity group: {group_name} with guest {guest_name}")
+                            break
 
-                        else:
-                            logger.critical(f"Node: {node_name} already got used for anti-affinity group:: {group_name}. (Tried for guest: {guest_name})")
-                else:
-                    logger.debug(f"Anti-Affinity: Group has less than 2 members. Skipping node calculation for the group.")
+                    else:
+                        logger.critical(f"Node: {node_name} already got used for anti-affinity group:: {group_name}. (Tried for guest: {guest_name})")
 
             else:
                 logger.debug(f"Guest: {guest_name} is not included in anti-affinity group: {group_name}. Skipping.")
@@ -466,7 +280,7 @@ class Calculations:
         logger.debug("Finished: val_anti_affinity.")
 
     @staticmethod
-    def val_node_relationships(proxlb_data: Dict[str, Any], guest_name: str):
+    def val_node_relationship(proxlb_data: Dict[str, Any], guest_name: str) -> None:
         """
         Validates and assigns guests to nodes based on defined relationships based on tags.
 
@@ -477,29 +291,27 @@ class Calculations:
         Returns:
         None
         """
-        logger.debug("Starting: val_node_relationships.")
+        logger.debug("Starting: val_node_relationship.")
         proxlb_data["guests"][guest_name]["processed"] = True
 
-        if len(proxlb_data["guests"][guest_name]["node_relationships"]) > 0:
-            logger.debug(f"Guest '{guest_name}' has relationships defined to node(s): {','.join(proxlb_data['guests'][guest_name]['node_relationships'])}. Pinning to node.")
-
-            # Get the node with the most free resources of the group
-            guest_node_relation_list = proxlb_data["guests"][guest_name]["node_relationships"]
-            Calculations.get_most_free_node(proxlb_data, False, guest_node_relation_list)
+        if proxlb_data["guests"][guest_name]["node_relationship"]:
+            logger.info(f"Guest '{guest_name}' has a specific relationship defined to node: {proxlb_data['guests'][guest_name]['node_relationship']}. Pinning to node.")
 
             # Validate if the specified node name is really part of the cluster
-            if proxlb_data["meta"]["balancing"]["balance_next_node"] in proxlb_data["nodes"].keys():
-                logger.debug(f"Guest '{guest_name}' has a specific relationship defined to node: {proxlb_data['meta']['balancing']['balance_next_node']} is a known hypervisor node in the cluster.")
+            if proxlb_data['guests'][guest_name]['node_relationship'] in proxlb_data["nodes"].keys():
+                logger.info(f"Guest '{guest_name}' has a specific relationship defined to node: {proxlb_data['guests'][guest_name]['node_relationship']} is a known hypervisor node in the cluster.")
+                # Pin the guest to the specified hypervisor node.
+                proxlb_data["meta"]["balancing"]["balance_next_node"] = proxlb_data['guests'][guest_name]['node_relationship']
             else:
-                logger.warning(f"Guest '{guest_name}' has a specific relationship defined to node: {proxlb_data['meta']['balancing']['balance_next_node']} but this node name is not known in the cluster!")
+                logger.warning(f"Guest '{guest_name}' has a specific relationship defined to node: {proxlb_data['guests'][guest_name]['node_relationship']} but this node name is not known in the cluster!")
 
         else:
-            logger.debug(f"Guest '{guest_name}' does not have any specific node relationships.")
+            logger.info(f"Guest '{guest_name}' does not have any specific node relationships.")
 
-        logger.debug("Finished: val_node_relationships.")
+        logger.debug("Finished: val_node_relationship.")
 
     @staticmethod
-    def update_node_resources(proxlb_data: Dict[str, Any]):
+    def update_node_resources(proxlb_data: Dict[str, Any]) -> None:
         """
         Updates the resource allocation and usage statistics for nodes when a guest
         is moved from one node to another.
@@ -519,11 +331,6 @@ class Calculations:
         """
         logger.debug("Starting: update_node_resources.")
         guest_name = proxlb_data["meta"]["balancing"]["balance_next_guest"]
-
-        if guest_name == "":
-            logger.debug("No guest defined to update node resources for.")
-            return
-
         node_current = proxlb_data["guests"][guest_name]["node_current"]
         node_target = proxlb_data["meta"]["balancing"]["balance_next_node"]
 
@@ -569,141 +376,67 @@ class Calculations:
 
         logger.debug("Finished: update_node_resources.")
 
-    def validate_affinity_map(proxlb_data: Dict[str, Any]):
-        """
-        Validates the affinity and anti-affinity constraints for all guests in the ProxLB data structure.
-
-        This function iterates through each guest and checks both affinity and anti-affinity rules.
-        If any guest violates these constraints, it sets the enforce_affinity flag to trigger rebalancing
-        and skips further validation for efficiency.
-
-        Args:
-            proxlb_data (Dict[str, Any]): A dictionary containing ProxLB configuration with the following structure:
-                - "guests" (list): List of guest identifiers to validate
-                - "meta" (dict): Metadata dictionary containing:
-                    - "balancing" (dict): Balancing configuration with "enforce_affinity" flag
-
-        Returns:
-            None: Modifies proxlb_data in-place by updating the "enforce_affinity" flag in meta.balancing
-
-        Raises:
-            None: Function handles validation gracefully and logs outcomes
-        """
-        logger.debug("Starting: validate_current_affinity.")
-        balancing_ok = True
-
-        for guest in proxlb_data["guests"]:
-
-            # We do not need to validate anymore if rebalancing is required
-            if balancing_ok is False:
-                proxlb_data["meta"]["balancing"]["enforce_affinity"] = True
-                logger.debug(f"Rebalancing based on affinity/anti-affinity map is required. Skipping further validation...")
-                break
-
-            balancing_state_affinity = Calculations.validate_current_affinity(proxlb_data, guest)
-            balancing_state_anti_affinity = Calculations.validate_current_anti_affinity(proxlb_data, guest)
-            logger.debug(f"Affinity for guest {guest} is {'valid' if balancing_state_affinity else 'NOT valid'}")
-            logger.debug(f"Anti-affinity for guest {guest} is {'valid' if balancing_state_anti_affinity else 'NOT valid'}")
-
-            balancing_ok = not balancing_state_affinity or not balancing_state_anti_affinity
-
-        if balancing_ok:
-            logger.debug(f"Rebalancing based on affinity/anti-affinity map is not required.")
-            proxlb_data["meta"]["balancing"]["enforce_affinity"] = False
-
-        logger.debug("Finished: validate_current_affinity.")
-
     @staticmethod
-    def get_guest_node(proxlb_data: Dict[str, Any], guest_name: str) -> str:
+    def update_cluster_resources(proxlb_data: Dict[str, Any], node: str, action: str) -> None:
         """
-        Return a currently assoicated PVE node where the guest is running on.
+        Updates the cluster resource statistics based on the specified action and node.
 
-        Args:
-            proxlb_data (Dict[str, Any]): A dictionary containing ProxLB configuration.
+        This method modifies the cluster-level resource data (such as CPU, memory, disk usage,
+        and node counts) based on the action performed ('add' or 'remove') for the specified node.
+        It calculates the updated statistics after adding or removing a node and logs the results.
+
+        Parameters:
+            proxlb_data (Dict[str, Any]): The data representing the current state of the cluster,
+                                        including node-level statistics for CPU, memory, and disk.
+            node (str): The identifier of the node whose resources are being added or removed from the cluster.
+            action (str): The action to perform, either 'add' or 'remove'. 'add' will include the node's
+                        resources in the cluster, while 'remove' will exclude the node's resources.
 
         Returns:
-            node_name_current (str): The name of the current node where the guest runs on.
-
+            None: The function modifies the `proxlb_data` dictionary in place to update the cluster resources.
         """
-        return proxlb_data["guests"][guest_name]["node_current"]
+        logger.debug("Starting: update_cluster_resources.")
+        logger.debug(f"DPM: Updating cluster statistics by online node {node}. Action: {action}")
+        logger.debug(f"DPM: update_cluster_resources - Before {action}: {proxlb_data['cluster']['memory_free_percent']}")
 
-    @staticmethod
-    def validate_current_affinity(proxlb_data: Dict[str, Any], guest_name: str) -> bool:
-        """
-        Validate that all guests in affinity groups containing the specified guest are on the same non-maintenance node.
+        if action == "add":
+            proxlb_data["cluster"]["node_count"] = proxlb_data["cluster"].get("node_count", 0) + 1
+            proxlb_data["cluster"]["cpu_total"] = proxlb_data["cluster"].get("cpu_total", 0) + proxlb_data["nodes"][node]["cpu_total"]
+            proxlb_data["cluster"]["cpu_used"] = proxlb_data["cluster"].get("cpu_used", 0) + proxlb_data["nodes"][node]["cpu_used"]
+            proxlb_data["cluster"]["cpu_free"] = proxlb_data["cluster"].get("cpu_free", 0) + proxlb_data["nodes"][node]["cpu_free"]
+            proxlb_data["cluster"]["cpu_free_percent"] = proxlb_data["cluster"].get("cpu_free", 0) / proxlb_data["cluster"].get("cpu_total", 0) * 100
+            proxlb_data["cluster"]["cpu_used_percent"] = proxlb_data["cluster"].get("cpu_used", 0) / proxlb_data["cluster"].get("cpu_total", 0) * 100
+            proxlb_data["cluster"]["memory_total"] = proxlb_data["cluster"].get("memory_total", 0) + proxlb_data["nodes"][node]["memory_total"]
+            proxlb_data["cluster"]["memory_used"] = proxlb_data["cluster"].get("memory_used", 0) + proxlb_data["nodes"][node]["memory_used"]
+            proxlb_data["cluster"]["memory_free"] = proxlb_data["cluster"].get("memory_free", 0) + proxlb_data["nodes"][node]["memory_free"]
+            proxlb_data["cluster"]["memory_free_percent"] = proxlb_data["cluster"].get("memory_free", 0) / proxlb_data["cluster"].get("memory_total", 0) * 100
+            proxlb_data["cluster"]["memory_used_percent"] = proxlb_data["cluster"].get("memory_used", 0) / proxlb_data["cluster"].get("memory_total", 0) * 100
+            proxlb_data["cluster"]["disk_total"] = proxlb_data["cluster"].get("disk_total", 0) + proxlb_data["nodes"][node]["disk_total"]
+            proxlb_data["cluster"]["disk_used"] = proxlb_data["cluster"].get("disk_used", 0) + proxlb_data["nodes"][node]["disk_used"]
+            proxlb_data["cluster"]["disk_free"] = proxlb_data["cluster"].get("disk_free", 0) + proxlb_data["nodes"][node]["disk_free"]
+            proxlb_data["cluster"]["disk_free_percent"] = proxlb_data["cluster"].get("disk_free", 0) / proxlb_data["cluster"].get("disk_total", 0) * 100
+            proxlb_data["cluster"]["disk_used_percent"] = proxlb_data["cluster"].get("disk_used", 0) / proxlb_data["cluster"].get("disk_total", 0) * 100
+            proxlb_data["cluster"]["node_count_available"] = proxlb_data["cluster"].get("node_count_available", 0) + 1
+            proxlb_data["cluster"]["node_count_overall"] = proxlb_data["cluster"].get("node_count_overall", 0) + 1
 
-        This function checks affinity group constraints for a given guest. It ensures that:
-        1. All guests within an affinity group are located on the same physical node
-        2. The node hosting the affinity group is not in maintenance mode
+        if action == "remove":
+            proxlb_data["cluster"]["node_count"] = proxlb_data["cluster"].get("node_count", 0) - 1
+            proxlb_data["cluster"]["cpu_total"] = proxlb_data["cluster"].get("cpu_total", 0) - proxlb_data["nodes"][node]["cpu_total"]
+            proxlb_data["cluster"]["cpu_used"] = proxlb_data["cluster"].get("cpu_used", 0) - proxlb_data["nodes"][node]["cpu_used"]
+            proxlb_data["cluster"]["cpu_free"] = proxlb_data["cluster"].get("cpu_free", 0) - proxlb_data["nodes"][node]["cpu_free"]
+            proxlb_data["cluster"]["cpu_free_percent"] = proxlb_data["cluster"].get("cpu_free", 0) / proxlb_data["cluster"].get("cpu_total", 0) * 100
+            proxlb_data["cluster"]["cpu_used_percent"] = proxlb_data["cluster"].get("cpu_used", 0) / proxlb_data["cluster"].get("cpu_total", 0) * 100
+            proxlb_data["cluster"]["memory_total"] = proxlb_data["cluster"].get("memory_total", 0) - proxlb_data["nodes"][node]["memory_total"]
+            proxlb_data["cluster"]["memory_used"] = proxlb_data["cluster"].get("memory_used") - proxlb_data["nodes"][node]["memory_used"]
+            proxlb_data["cluster"]["memory_free"] = proxlb_data["cluster"].get("memory_free") - proxlb_data["nodes"][node]["memory_free"]
+            proxlb_data["cluster"]["memory_free_percent"] = proxlb_data["cluster"].get("memory_free") / proxlb_data["cluster"].get("memory_total", 0) * 100
+            proxlb_data["cluster"]["memory_used_percent"] = proxlb_data["cluster"].get("memory_used") / proxlb_data["cluster"].get("memory_total", 0) * 100
+            proxlb_data["cluster"]["disk_total"] = proxlb_data["cluster"].get("disk_total", 0) - proxlb_data["nodes"][node]["disk_total"]
+            proxlb_data["cluster"]["disk_used"] = proxlb_data["cluster"].get("disk_used", 0) - proxlb_data["nodes"][node]["disk_used"]
+            proxlb_data["cluster"]["disk_free"] = proxlb_data["cluster"].get("disk_free", 0) - proxlb_data["nodes"][node]["disk_free"]
+            proxlb_data["cluster"]["disk_free_percent"] = proxlb_data["cluster"].get("disk_free", 0) / proxlb_data["cluster"].get("disk_total", 0) * 100
+            proxlb_data["cluster"]["disk_used_percent"] = proxlb_data["cluster"].get("disk_used", 0) / proxlb_data["cluster"].get("disk_total", 0) * 100
+            proxlb_data["cluster"]["node_count_available"] = proxlb_data["cluster"].get("node_count_available", 0) - 1
 
-        Args:
-            proxlb_data (Dict[str, Any]): A dictionary containing the complete ProxLB state including:
-                - "groups": Dictionary with "affinity" key containing affinity group definitions
-                - "guests": Dictionary with guest information
-                - "nodes": Dictionary with node information including maintenance status
-            guest_name (str): The name of the guest to validate affinity for
-
-        Returns:
-            bool: True if all affinity groups containing the guest are valid (all members on same
-                non-maintenance node), False otherwise
-        """
-        logger.debug("Starting: validate_current_affinity.")
-        for group_name, grp in proxlb_data["groups"]["affinity"].items():
-            if guest_name not in grp["guests"]:
-                continue
-
-            nodes = []
-            for group in grp["guests"]:
-                if group not in proxlb_data["guests"]:
-                    continue
-
-                node = Calculations.get_guest_node(proxlb_data, group)
-                if proxlb_data["nodes"][node]["maintenance"]:
-                    logger.debug(f"Group '{group_name}' invalid: node '{node}' in maintenance.")
-                    return False
-                nodes.append(node)
-
-            if len(set(nodes)) != 1:
-                logger.debug(f"Group '{group_name}' invalid: guests spread across nodes {set(nodes)}.")
-                return False
-
-        return True
-
-    @staticmethod
-    def validate_current_anti_affinity(proxlb_data: Dict[str, Any], guest_name: str) -> bool:
-        """
-        Validate that all guests in anti-affinity groups containing the specified guest are not on the same node.
-
-        This function checks anti-affinity group constraints for a given guest. It ensures that:
-        1. All guests within an anti-affinity group are located on the same physical node
-        2. The node hosting the anti-affinity group is not in maintenance mode
-
-        Args:
-            proxlb_data (Dict[str, Any]): A dictionary containing the complete ProxLB state including:
-                - "groups": Dictionary with "affinity" key containing affinity group definitions
-                - "guests": Dictionary with guest information
-                - "nodes": Dictionary with node information including maintenance status
-            guest_name (str): The name of the guest to validate affinity for
-
-        Returns:
-            bool: True if all anti-affinity groups containing the guest are valid (all members on different
-                non-maintenance node), False otherwise
-        """
-        logger.debug("Starting: validate_current_anti_affinity.")
-        for group_name, grp in proxlb_data["groups"]["anti_affinity"].items():
-            if guest_name not in grp["guests"]:
-                continue
-            nodes = []
-            for group in grp["guests"]:
-                if group not in proxlb_data["guests"]:
-                    continue
-
-                node = Calculations.get_guest_node(proxlb_data, group)
-                if proxlb_data["nodes"][node]["maintenance"]:
-                    return False
-                nodes.append(node)
-
-            if len(nodes) != len(set(nodes)):
-                return False
-
-        return True
+        logger.debug(f"DPM: update_cluster_resources - After {action}: {proxlb_data['cluster']['memory_free_percent']}")
+        logger.debug("Finished: update_cluster_resources.")

proxlb/models/dpm.py

@@ -0,0 +1,255 @@
+"""
+The DPM (Dynamic Power Management) class is responsible for the dynamic management
+of nodes within a Proxmox cluster, optimizing resource utilization by controlling
+node power states based on specified schedules and conditions.
+
+This class provides functionality for:
+- Tracking and validating schedules for dynamic power management.
+- Shutting down nodes that are underutilized or not needed.
+- Starting up nodes using Wake-on-LAN (WOL) based on certain conditions.
+- Ensuring that nodes are properly flagged for maintenance and startup/shutdown actions.
+
+The DPM class can operate in different modes, such as static and automatic,
+to either perform predefined actions or dynamically adjust based on real-time resource usage.
+"""
+
+__author__ = "Florian Paul Azim Hoberg <gyptazy>"
+__copyright__ = "Copyright (C) 2025 Florian Paul Azim Hoberg (@gyptazy)"
+__license__ = "GPL-3.0"
+
+
+import proxmoxer
+from typing import Dict, Any
+from models.calculations import Calculations
+from utils.logger import SystemdLogger
+
+logger = SystemdLogger()
+
+
+class DPM:
+    """
+    The DPM (Dynamic Power Management) class is responsible for the dynamic management
+    of nodes within a Proxmox cluster, optimizing resource utilization by controlling
+    node power states based on specified schedules and conditions.
+
+    This class provides functionality for:
+    - Tracking and validating schedules for dynamic power management.
+    - Shutting down nodes that are underutilized or not needed.
+    - Starting up nodes using Wake-on-LAN (WOL) based on certain conditions.
+    - Ensuring that nodes are properly flagged for maintenance and startup/shutdown actions.
+
+    The DPM class can operate in different modes, such as static and automatic,
+    to either perform predefined actions or dynamically adjust based on real-time resource usage.
+
+    Attributes:
+        None directly defined for the class; instead, all actions are based on input data
+        and interactions with the Proxmox API and other helper functions.
+
+    Methods:
+        __init__(proxlb_data: Dict[str, Any]):
+            Initializes the DPM class, checking whether DPM is enabled and operating in the
+            appropriate mode (static or auto).
+
+        dpm_static(proxlb_data: Dict[str, Any]) -> None:
+            Evaluates the cluster's resource availability and performs static power management
+            actions by removing nodes that are not required.
+
+        dpm_shutdown_nodes(proxmox_api, proxlb_data) -> None:
+            Shuts down nodes flagged for DPM shutdown by using the Proxmox API, ensuring
+            that Wake-on-LAN (WOL) is available for proper node recovery.
+
+        dpm_startup_nodes(proxmox_api, proxlb_data) -> None:
+            Powers on nodes that are flagged for startup and are not in maintenance mode,
+            leveraging Wake-on-LAN (WOL) functionality.
+
+        dpm_validate_wol_mac(proxmox_api, node) -> None:
+            Validates and retrieves the Wake-on-LAN (WOL) MAC address for a given node,
+            ensuring that a valid address is set for powering on the node remotely.
+    """
+
+    def __init__(self, proxlb_data: Dict[str, Any]):
+        """
+        Initializes the DPM class with the provided ProxLB data.
+
+        Args:
+            proxlb_data (dict): The data required for balancing VMs and CTs.
+        """
+        logger.debug("Starting: dpm class.")
+
+        if proxlb_data["meta"].get("dpm", {}).get("enable", False):
+            logger.debug("DPM function is enabled.")
+            mode = proxlb_data["meta"].get("dpm", {}).get("mode", None)
+
+            if mode == "static":
+                self.dpm_static(proxlb_data)
+
+            if mode == "auto":
+                self.dpm_auto(proxlb_data)
+
+        else:
+            logger.debug("DPM function is not enabled.")
+
+        logger.debug("Finished: dpm class.")
+
+    def dpm_static(self, proxlb_data: Dict[str, Any]) -> None:
+        """
+        Evaluates and performs static Distributed Power Management (DPM) actions based on current cluster state.
+
+        This method monitors cluster resource availability and attempts to reduce the number of active nodes
+        when sufficient free resources are available. It ensures a minimum number of nodes remains active
+        and prioritizes shutting down nodes with the least utilized resources to minimize impact. Nodes selected
+        for shutdown are marked for maintenance and flagged for DPM shutdown.
+
+        Parameters:
+            proxlb_data (Dict[str, Any]): A dictionary containing metadata, cluster status, and node-level information
+                including resource utilization, configuration settings, and DPM thresholds.
+
+        Returns:
+            None: Modifies the input dictionary in-place to reflect updated cluster state and node flags.
+        """
+        logger.debug("Starting: dpm_static.")
+
+        method = proxlb_data["meta"].get("dpm", {}).get("method", "memory")
+        cluster_nodes_overall = proxlb_data["cluster"]["node_count_overall"]
+        cluster_nodes_available = proxlb_data["cluster"]["node_count_available"]
+        cluster_free_resources_percent = int(proxlb_data["cluster"][f"{method}_free_percent"])
+        cluster_free_resources_req_min = proxlb_data["meta"].get("dpm", {}).get("cluster_min_free_resources", 0)
+        cluster_mind_nodes = proxlb_data["meta"].get("dpm", {}).get("cluster_min_nodes", 3)
+        logger.debug(f"DPM: Cluster Nodes: {cluster_nodes_overall} | Nodes available: {cluster_nodes_available} | Nodes offline: {cluster_nodes_overall - cluster_nodes_available}")
+
+        # Only proceed removing nodes if the cluster has enough resources
+        while cluster_free_resources_percent > cluster_free_resources_req_min:
+            logger.debug(f"DPM: More free resources {cluster_free_resources_percent}% available than required: {cluster_free_resources_req_min}%. DPM evaluation starting...")
+
+            # Ensure that we have at least a defined minimum of nodes left
+            if cluster_nodes_available > cluster_mind_nodes:
+                logger.debug(f"DPM: A minimum of {cluster_mind_nodes} nodes is required. {cluster_nodes_available} are available. Proceeding...")
+
+                # Get the node with the fewest used resources to keep migrations low
+                Calculations.get_most_free_node(proxlb_data, False)
+                dpm_node = proxlb_data["meta"]["balancing"]["balance_next_node"]
+
+                # Perform cluster calculation for evaluating how many nodes can safely leave
+                # the cluster. Further object calculations are being processed afterwards by
+                # the calculation class
+                logger.debug(f"DPM: Removing node {dpm_node} from cluster. Node will be turned off later.")
+                Calculations.update_cluster_resources(proxlb_data, dpm_node, "remove")
+                cluster_free_resources_percent = int(proxlb_data["cluster"][f"{method}_free_percent"])
+                logger.debug(f"DPM: Free cluster resources changed to: {int(proxlb_data['cluster'][f'{method}_free_percent'])}%.")
+
+                # Set node to maintenance and DPM shutdown
+                proxlb_data["nodes"][dpm_node]["maintenance"] = True
+                proxlb_data["nodes"][dpm_node]["dpm_shutdown"] = True
+            else:
+                logger.warning(f"DPM: A minimum of {cluster_mind_nodes} nodes is required. {cluster_nodes_available} are available. Cannot proceed!")
+
+        logger.debug(f"DPM: Not enough free resources {cluster_free_resources_percent}% available than required: {cluster_free_resources_req_min}%. DPM evaluation stopped.")
+        logger.debug("Finished: dpm_static.")
+        return proxlb_data
+
+    @staticmethod
+    def dpm_shutdown_nodes(proxmox_api, proxlb_data: Dict[str, Any]) -> None:
+        """
+        Shuts down cluster nodes that are marked for maintenance and flagged for DPM shutdown.
+
+        This method iterates through the cluster nodes in the provided data and attempts to
+        power off any node that has both the 'maintenance' and 'dpm_shutdown' flags set.
+        It communicates with the Proxmox API to issue shutdown commands and logs any failures.
+
+        Parameters:
+            proxmox_api: An instance of the Proxmox API client used to issue node shutdown commands.
+            proxlb_data: A dictionary containing node status information, including flags for
+                maintenance and DPM shutdown readiness.
+
+        Returns:
+            None: Performs shutdown operations and logs outcomes; modifies no data directly.
+        """
+        logger.debug("Starting: dpm_shutdown_nodes.")
+        for node, node_info in proxlb_data["nodes"].items():
+
+            if node_info["maintenance"] and node_info["dpm_shutdown"]:
+                logger.debug(f"DPM: Node: {node} is flagged as maintenance mode and to be powered off.")
+
+                # Ensure that the node has a valid WOL MAC defined. If not
+                # we would be unable to power on that system again
+                valid_wol_mac = DPM.dpm_validate_wol_mac(proxmox_api, node)
+
+                if valid_wol_mac:
+                    try:
+                        logger.debug(f"DPM: Shutting down node: {node}.")
+                        job_id = proxmox_api.nodes(node).status.post(command="shutdown")
+                    except proxmoxer.core.ResourceException as proxmox_api_error:
+                        logger.critical(f"DPM: Error while powering off node {node}. Please check job-id: {job_id}")
+                        logger.debug(f"DPM: Error while powering off node {node}. Please check job-id: {job_id}")
+                else:
+                    logger.critical(f"DPM: Node {node} cannot be powered off due to missing WOL MAC. Please define a valid WOL MAC for this node.")
+
+        logger.debug("Finished: dpm_shutdown_nodes.")
+
+    @staticmethod
+    def dpm_startup_nodes(proxmox_api, proxlb_data: Dict[str, Any]) -> None:
+        """
+        Starts uo cluster nodes that are marked for DPM start up.
+
+        This method iterates through the cluster nodes in the provided data and attempts to
+        power on any node that is not flagged as 'maintenance' but flagged as 'dpm_startup'.
+        It communicates with the Proxmox API to issue poweron commands and logs any failures.
+
+        Parameters:
+            proxmox_api: An instance of the Proxmox API client used to issue node startup commands.
+            proxlb_data: A dictionary containing node status information, including flags for
+                maintenance and DPM shutdown readiness.
+
+        Returns:
+            None: Performs poweron operations and logs outcomes; modifies no data directly.
+        """
+        logger.debug("Starting: dpm_startup_nodes.")
+        for node, node_info in proxlb_data["nodes"].items():
+
+            if not node_info["maintenance"]:
+                logger.debug(f"DPM: Node: {node} is not in maintenance mode.")
+
+                if node_info["dpm_startup"]:
+                    logger.debug(f"DPM: Node: {node} is flagged as to be started.")
+
+                    try:
+                        logger.debug(f"DPM: Powering on node: {node}.")
+                        # Important: This requires Proxmox Operators to define the
+                        # WOL address for each node within the Proxmox webinterface
+                        job_id = proxmox_api.nodes().wakeonlan.post(node=node)
+                    except proxmoxer.core.ResourceException as proxmox_api_error:
+                        logger.critical(f"DPM: Error while powering on node {node}. Please check job-id: {job_id}")
+                        logger.debug(f"DPM: Error while powering on node {node}. Please check job-id: {job_id}")
+
+        logger.debug("Finished: dpm_startup_nodes.")
+
+    @staticmethod
+    def dpm_validate_wol_mac(proxmox_api, node: Dict[str, Any]) -> str:
+        """
+        Retrieves and validates the Wake-on-LAN (WOL) MAC address for a specified node.
+
+        This method fetches the MAC address configured for Wake-on-LAN (WOL) from the Proxmox API.
+        If the MAC address is found, it is logged. In case of failure to retrieve the address,
+        a critical log is generated indicating the absence of a WOL MAC address for the node.
+
+        Parameters:
+            proxmox_api: An instance of the Proxmox API client used to query node configurations.
+            node: The identifier (name or ID) of the node for which the WOL MAC address is to be validated.
+
+        Returns:
+            node_wol_mac_address: The WOL MAC address for the specified node if found, otherwise `None`.
+        """
+        logger.debug("Starting: dpm_validate_wol_mac.")
+
+        try:
+            logger.debug(f"DPM: Getting WOL MAC address for node {node} from API.")
+            node_wol_mac_address = proxmox_api.nodes(node).config.get(property="wakeonlan")
+            node_wol_mac_address = node_wol_mac_address.get("wakeonlan")
+            logger.debug(f"DPM: Node {node} has MAC address: {node_wol_mac_address} for WOL.")
+        except proxmoxer.core.ResourceException as proxmox_api_error:
+            logger.debug(f"DPM: Failed to get WOL MAC address for node {node} from API.")
+            node_wol_mac_address = None
+            logger.critical(f"DPM: Node {node} has no MAC address defined for WOL.")
+
+        logger.debug("Finished: dpm_validate_wol_mac.")
+        return node_wol_mac_address

proxlb/models/features.py

@@ -1,90 +0,0 @@
-"""
-ProxLB Features module for validating and adjusting feature flags
-based on Proxmox VE node versions and cluster compatibility.
-"""
-
-__author__ = "Florian Paul Azim Hoberg <gyptazy>"
-__copyright__ = "Copyright (C) 2025 Florian Paul Azim Hoberg (@gyptazy)"
-__license__ = "GPL-3.0"
-
-
-from typing import List
-from typing import Dict, Any
-from utils.logger import SystemdLogger
-from packaging import version
-
-logger = SystemdLogger()
-
-
-class Features:
-    """
-    ProxLB Features module for validating and adjusting feature flags
-    based on Proxmox VE node versions and cluster compatibility.
-
-    Responsibilities:
-        - Validate and adjust feature flags based on Proxmox VE node versions.
-
-    Methods:
-        __init__():
-            No-op initializer.
-
-        validate_available_features(proxlb_data: dict) -> None:
-            Static method that inspects proxlb_data["nodes"] versions and disables
-            incompatible balancing features for Proxmox VE versions < 9.0.0.
-            This function mutates proxlb_data in place.
-
-    Notes:
-        - Expects proxlb_data to be a dict with "nodes" and "meta" keys.
-    """
-    def __init__(self):
-        """
-        Initializes the Features class.
-        """
-
-    @staticmethod
-    def validate_available_features(proxlb_data: any) -> None:
-        """
-        Validate and adjust feature flags in the provided proxlb_data according to Proxmox VE versions.
-
-        This function inspects the cluster node versions in proxlb_data and disables features
-        that are incompatible with Proxmox VE versions older than 9.0.0. Concretely, if any node
-        reports a 'pve_version' lower than "9.0.0":
-        - If meta.balancing.with_conntrack_state is truthy, it is set to False and a warning is logged.
-        - If meta.balancing.mode equals "psi", meta.balancing.enable is set to False and a warning is logged.
-
-            proxlb_data (dict): Cluster data structure that must contain:
-                - "nodes": a mapping (e.g., dict) whose values are mappings containing a 'pve_version' string.
-                - "meta": a mapping that may contain a "balancing" mapping with keys:
-                    - "with_conntrack_state" (bool, optional)
-                    - "mode" (str, optional)
-                    - "enable" (bool, optional)
-
-            None: The function mutates proxlb_data in place to disable incompatible features.
-
-        Side effects:
-            - Mutates proxlb_data["meta"]["balancing"] when incompatible features are detected.
-            - Emits debug and warning log messages.
-
-        Notes:
-            - Unexpected or missing keys/types in proxlb_data may raise KeyError or TypeError.
-            - Version comparison uses semantic version parsing; callers should provide versions as strings.
-
-        Returns:
-            None
-        """
-        logger.debug("Starting: validate_available_features.")
-
-        any_non_pve9_node = any(version.parse(n['pve_version']) < version.parse("9.0.0") for n in proxlb_data["nodes"].values())
-        if any_non_pve9_node:
-
-            with_conntrack_state = proxlb_data["meta"].get("balancing", {}).get("with_conntrack_state", False)
-            if with_conntrack_state:
-                logger.warning("Non Proxmox VE 9 systems detected: Deactivating migration option 'with-conntrack-state'!")
-                proxlb_data["meta"]["balancing"]["with_conntrack_state"] = False
-
-            psi_balancing = proxlb_data["meta"].get("balancing", {}).get("mode", None)
-            if psi_balancing == "psi":
-                logger.warning("Non Proxmox VE 9 systems detected: Deactivating balancing!")
-                proxlb_data["meta"]["balancing"]["enable"] = False
-
-        logger.debug("Finished: validate_available_features.")

proxlb/models/guests.py

@@ -10,7 +10,6 @@ __license__ = "GPL-3.0"
 
 from typing import Dict, Any
 from utils.logger import SystemdLogger
-from models.pools import Pools
 from models.tags import Tags
 import time
 
@@ -36,7 +35,7 @@ class Guests:
         """
 
     @staticmethod
-    def get_guests(proxmox_api: any, pools: Dict[str, Any], nodes: Dict[str, Any], meta: Dict[str, Any], proxlb_config: Dict[str, Any]) -> Dict[str, Any]:
+    def get_guests(proxmox_api: any, nodes: Dict[str, Any], meta: Dict[str, Any]) -> Dict[str, Any]:
         """
         Get metrics of all guests in a Proxmox cluster.
 
@@ -47,8 +46,6 @@ class Guests:
         Args:
             proxmox_api (any): The Proxmox API client instance.
             nodes (Dict[str, Any]): A dictionary containing information about the nodes in the Proxmox cluster.
-            meta (Dict[str, Any]): A dictionary containing metadata information.
-            proxmox_config (Dict[str, Any]): A dictionary containing the ProxLB configuration.
 
         Returns:
             Dict[str, Any]: A dictionary containing metrics and information for all running guests.
@@ -65,43 +62,36 @@ class Guests:
             # resource metrics for rebalancing to ensure that we do not overprovisiong the node.
             for guest in proxmox_api.nodes(node).qemu.get():
                 if guest['status'] == 'running':
+
+                    # If the balancing method is set to cpu, we need to wait for the guest to report
+                    # cpu usage. This is important for the balancing process to ensure that we do not
+                    # wait for a guest for an infinite time.
+                    if meta["meta"]["balancing"]["method"] == "cpu":
+                        retry_counter = 0
+                        while guest['cpu'] == 0 and retry_counter < 10:
+                            guest = proxmox_api.nodes(node).qemu(guest['vmid']).status.current.get()
+                            logger.debug(f"Guest {guest['name']} (type VM) is reporting {guest['cpu']} cpu usage on retry {retry_counter}.")
+                            retry_counter += 1
+                            time.sleep(1)
+
                     guests['guests'][guest['name']] = {}
                     guests['guests'][guest['name']]['name'] = guest['name']
-                    guests['guests'][guest['name']]['cpu_total'] = int(guest['cpus'])
-                    guests['guests'][guest['name']]['cpu_used'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', None)
-                    guests['guests'][guest['name']]['cpu_pressure_some_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'some')
-                    guests['guests'][guest['name']]['cpu_pressure_full_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'full')
-                    guests['guests'][guest['name']]['cpu_pressure_some_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'some', spikes=True)
-                    guests['guests'][guest['name']]['cpu_pressure_full_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'full', spikes=True)
-                    guests['guests'][guest['name']]['cpu_pressure_hot'] = False
+                    guests['guests'][guest['name']]['cpu_total'] = guest['cpus']
+                    guests['guests'][guest['name']]['cpu_used'] = guest['cpu'] * guest['cpus']
                     guests['guests'][guest['name']]['memory_total'] = guest['maxmem']
                     guests['guests'][guest['name']]['memory_used'] = guest['mem']
-                    guests['guests'][guest['name']]['memory_pressure_some_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'some')
-                    guests['guests'][guest['name']]['memory_pressure_full_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'full')
-                    guests['guests'][guest['name']]['memory_pressure_some_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'some', spikes=True)
-                    guests['guests'][guest['name']]['memory_pressure_full_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'full', spikes=True)
-                    guests['guests'][guest['name']]['memory_pressure_hot'] = False
                     guests['guests'][guest['name']]['disk_total'] = guest['maxdisk']
                     guests['guests'][guest['name']]['disk_used'] = guest['disk']
-                    guests['guests'][guest['name']]['disk_pressure_some_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'some')
-                    guests['guests'][guest['name']]['disk_pressure_full_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'full')
-                    guests['guests'][guest['name']]['disk_pressure_some_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'some', spikes=True)
-                    guests['guests'][guest['name']]['disk_pressure_full_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'full', spikes=True)
-                    guests['guests'][guest['name']]['disk_pressure_hot'] = False
                     guests['guests'][guest['name']]['id'] = guest['vmid']
                     guests['guests'][guest['name']]['node_current'] = node
                     guests['guests'][guest['name']]['node_target'] = node
                     guests['guests'][guest['name']]['processed'] = False
-                    guests['guests'][guest['name']]['pressure_hot'] = False
                     guests['guests'][guest['name']]['tags'] = Tags.get_tags_from_guests(proxmox_api, node, guest['vmid'], 'vm')
-                    guests['guests'][guest['name']]['pools'] = Pools.get_pools_for_guest(guest['name'], pools)
-                    guests['guests'][guest['name']]['affinity_groups'] = Tags.get_affinity_groups(guests['guests'][guest['name']]['tags'], guests['guests'][guest['name']]['pools'], proxlb_config)
-                    guests['guests'][guest['name']]['anti_affinity_groups'] = Tags.get_anti_affinity_groups(guests['guests'][guest['name']]['tags'], guests['guests'][guest['name']]['pools'], proxlb_config)
+                    guests['guests'][guest['name']]['affinity_groups'] = Tags.get_affinity_groups(guests['guests'][guest['name']]['tags'])
+                    guests['guests'][guest['name']]['anti_affinity_groups'] = Tags.get_anti_affinity_groups(guests['guests'][guest['name']]['tags'])
                     guests['guests'][guest['name']]['ignore'] = Tags.get_ignore(guests['guests'][guest['name']]['tags'])
-                    guests['guests'][guest['name']]['node_relationships'] = Tags.get_node_relationships(guests['guests'][guest['name']]['tags'], nodes, guests['guests'][guest['name']]['pools'], proxlb_config)
+                    guests['guests'][guest['name']]['node_relationship'] = Tags.get_node_relationship(guests['guests'][guest['name']]['tags'])
                     guests['guests'][guest['name']]['type'] = 'vm'
-
-                    logger.debug(f"Resources of Guest {guest['name']} (type VM) added: {guests['guests'][guest['name']]}")
                 else:
                     logger.debug(f'Metric for VM {guest["name"]} ignored because VM is not running.')
 
@@ -112,97 +102,24 @@ class Guests:
                 if guest['status'] == 'running':
                     guests['guests'][guest['name']] = {}
                     guests['guests'][guest['name']]['name'] = guest['name']
-                    guests['guests'][guest['name']]['cpu_total'] = int(guest['cpus'])
-                    guests['guests'][guest['name']]['cpu_used'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', None)
-                    guests['guests'][guest['name']]['cpu_pressure_some_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'some')
-                    guests['guests'][guest['name']]['cpu_pressure_full_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'full')
-                    guests['guests'][guest['name']]['cpu_pressure_some_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'some', spikes=True)
-                    guests['guests'][guest['name']]['cpu_pressure_full_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'cpu', 'full', spikes=True)
-                    guests['guests'][guest['name']]['cpu_pressure_hot'] = False
+                    guests['guests'][guest['name']]['cpu_total'] = guest['cpus']
+                    guests['guests'][guest['name']]['cpu_used'] = guest['cpu']
                     guests['guests'][guest['name']]['memory_total'] = guest['maxmem']
                     guests['guests'][guest['name']]['memory_used'] = guest['mem']
-                    guests['guests'][guest['name']]['memory_pressure_some_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'some')
-                    guests['guests'][guest['name']]['memory_pressure_full_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'full')
-                    guests['guests'][guest['name']]['memory_pressure_some_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'some', spikes=True)
-                    guests['guests'][guest['name']]['memory_pressure_full_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'memory', 'full', spikes=True)
-                    guests['guests'][guest['name']]['memory_pressure_hot'] = False
                     guests['guests'][guest['name']]['disk_total'] = guest['maxdisk']
                     guests['guests'][guest['name']]['disk_used'] = guest['disk']
-                    guests['guests'][guest['name']]['disk_pressure_some_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'some')
-                    guests['guests'][guest['name']]['disk_pressure_full_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'full')
-                    guests['guests'][guest['name']]['disk_pressure_some_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'some', spikes=True)
-                    guests['guests'][guest['name']]['disk_pressure_full_spikes_percent'] = Guests.get_guest_rrd_data(proxmox_api, node, guest['vmid'], guest['name'], 'disk', 'full', spikes=True)
-                    guests['guests'][guest['name']]['disk_pressure_hot'] = False
                     guests['guests'][guest['name']]['id'] = guest['vmid']
                     guests['guests'][guest['name']]['node_current'] = node
                     guests['guests'][guest['name']]['node_target'] = node
                     guests['guests'][guest['name']]['processed'] = False
-                    guests['guests'][guest['name']]['pressure_hot'] = False
                     guests['guests'][guest['name']]['tags'] = Tags.get_tags_from_guests(proxmox_api, node, guest['vmid'], 'ct')
-                    guests['guests'][guest['name']]['pools'] = Pools.get_pools_for_guest(guest['name'], pools)
-                    guests['guests'][guest['name']]['affinity_groups'] = Tags.get_affinity_groups(guests['guests'][guest['name']]['tags'], guests['guests'][guest['name']]['pools'], proxlb_config)
-                    guests['guests'][guest['name']]['anti_affinity_groups'] = Tags.get_anti_affinity_groups(guests['guests'][guest['name']]['tags'], guests['guests'][guest['name']]['pools'], proxlb_config)
+                    guests['guests'][guest['name']]['affinity_groups'] = Tags.get_affinity_groups(guests['guests'][guest['name']]['tags'])
+                    guests['guests'][guest['name']]['anti_affinity_groups'] = Tags.get_anti_affinity_groups(guests['guests'][guest['name']]['tags'])
                     guests['guests'][guest['name']]['ignore'] = Tags.get_ignore(guests['guests'][guest['name']]['tags'])
-                    guests['guests'][guest['name']]['node_relationships'] = Tags.get_node_relationships(guests['guests'][guest['name']]['tags'], nodes, guests['guests'][guest['name']]['pools'], proxlb_config)
+                    guests['guests'][guest['name']]['node_relationship'] = Tags.get_node_relationship(guests['guests'][guest['name']]['tags'])
                     guests['guests'][guest['name']]['type'] = 'ct'
-
-                    logger.debug(f"Resources of Guest {guest['name']} (type CT) added: {guests['guests'][guest['name']]}")
                 else:
                     logger.debug(f'Metric for CT {guest["name"]} ignored because CT is not running.')
 
         logger.debug("Finished: get_guests.")
         return guests
-
-    @staticmethod
-    def get_guest_rrd_data(proxmox_api, node_name: str, vm_id: int, vm_name: str, object_name: str, object_type: str, spikes=False) -> float:
-        """
-        Retrieves the rrd data metrics for a specific resource (CPU, memory, disk) of a guest VM or CT.
-
-        Args:
-            proxmox_api (Any): The Proxmox API client instance.
-            node_name (str): The name of the node hosting the guest.
-            vm_id (int): The ID of the guest VM or CT.
-            vm_name (str): The name of the guest VM or CT.
-            object_name (str): The resource type to query (e.g., 'cpu', 'memory', 'disk').
-            object_type (str, optional): The pressure type ('some', 'full') or None for average usage.
-            spikes (bool, optional): Whether to consider spikes in the calculation. Defaults to False.
-
-        Returns:
-            float: The calculated average usage value for the specified resource.
-        """
-        logger.debug("Starting: get_guest_rrd_data.")
-        time.sleep(0.1)
-
-        try:
-            if spikes:
-                logger.debug(f"Getting spike RRD data for {object_name} from guest: {vm_name}.")
-                guest_data_rrd = proxmox_api.nodes(node_name).qemu(vm_id).rrddata.get(timeframe="hour", cf="MAX")
-            else:
-                logger.debug(f"Getting average RRD data for {object_name} from guest: {vm_name}.")
-                guest_data_rrd = proxmox_api.nodes(node_name).qemu(vm_id).rrddata.get(timeframe="hour", cf="AVERAGE")
-        except Exception:
-            logger.error(f"Failed to retrieve RRD data for guest: {vm_name} (ID: {vm_id}) on node: {node_name}. Using 0.0 as value.")
-            logger.debug("Finished: get_guest_rrd_data.")
-            return float(0.0)
-
-        if object_type:
-
-            lookup_key = f"pressure{object_name}{object_type}"
-            if spikes:
-                # RRD data is collected every minute, so we look at the last 6 entries
-                # and take the maximum value to represent the spike
-                logger.debug(f"Getting RRD data (spike: {spikes}) of pressure for {object_name} {object_type} from guest: {vm_name}.")
-                rrd_data_value = [row.get(lookup_key) for row in guest_data_rrd if row.get(lookup_key) is not None]
-                rrd_data_value = max(rrd_data_value[-6:], default=0.0)
-            else:
-                # Calculate the average value from the RRD data entries
-                logger.debug(f"Getting RRD data (spike: {spikes}) of pressure for {object_name} {object_type} from guest: {vm_name}.")
-                rrd_data_value = sum(entry.get(lookup_key, 0.0) for entry in guest_data_rrd) / len(guest_data_rrd)
-
-        else:
-            logger.debug(f"Getting RRD data of cpu usage from guest: {vm_name}.")
-            rrd_data_value = sum(entry.get("cpu", 0.0) for entry in guest_data_rrd) / len(guest_data_rrd)
-
-        logger.debug(f"RRD data (spike: {spikes}) for {object_name} from guest: {vm_name}: {rrd_data_value}")
-        logger.debug("Finished: get_guest_rrd_data.")
-        return rrd_data_value

proxlb/models/nodes.py

@@ -21,7 +21,6 @@ __copyright__ = "Copyright (C) 2025 Florian Paul Azim Hoberg (@gyptazy)"
 __license__ = "GPL-3.0"
 
 
-import time
 from typing import Dict, Any
 from utils.logger import SystemdLogger
 
@@ -48,7 +47,6 @@ class Nodes:
 
         Args:
             proxmox_api (any): The Proxmox API client instance.
-            proxmox_config (Dict[str, Any]): A dictionary containing the ProxLB configuration.
             nodes (Dict[str, Any]): A dictionary containing information about the nodes in the Proxmox cluster.
 
         Returns:
@@ -56,15 +54,16 @@ class Nodes:
         """
         logger.debug("Starting: get_nodes.")
         nodes = {"nodes": {}}
+        cluster = {"cluster": {}}
 
         for node in proxmox_api.nodes.get():
             # Ignoring a node results into ignoring all placed guests on the ignored node!
             if node["status"] == "online" and not Nodes.set_node_ignore(proxlb_config, node["node"]):
                 nodes["nodes"][node["node"]] = {}
                 nodes["nodes"][node["node"]]["name"] = node["node"]
-                nodes["nodes"][node["node"]]["pve_version"] = Nodes.get_node_pve_version(proxmox_api, node["node"])
-                nodes["nodes"][node["node"]]["pressure_hot"] = False
                 nodes["nodes"][node["node"]]["maintenance"] = False
+                nodes["nodes"][node["node"]]["dpm_shutdown"] = False
+                nodes["nodes"][node["node"]]["dpm_startup"] = False
                 nodes["nodes"][node["node"]]["cpu_total"] = node["maxcpu"]
                 nodes["nodes"][node["node"]]["cpu_assigned"] = 0
                 nodes["nodes"][node["node"]]["cpu_used"] = node["cpu"] * node["maxcpu"]
@@ -72,11 +71,6 @@ class Nodes:
                 nodes["nodes"][node["node"]]["cpu_assigned_percent"] = nodes["nodes"][node["node"]]["cpu_assigned"] / nodes["nodes"][node["node"]]["cpu_total"] * 100
                 nodes["nodes"][node["node"]]["cpu_free_percent"] = nodes["nodes"][node["node"]]["cpu_free"] / node["maxcpu"] * 100
                 nodes["nodes"][node["node"]]["cpu_used_percent"] = nodes["nodes"][node["node"]]["cpu_used"] / node["maxcpu"] * 100
-                nodes["nodes"][node["node"]]["cpu_pressure_some_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "cpu", "some")
-                nodes["nodes"][node["node"]]["cpu_pressure_full_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "cpu", "full")
-                nodes["nodes"][node["node"]]["cpu_pressure_some_spikes_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "cpu", "some", spikes=True)
-                nodes["nodes"][node["node"]]["cpu_pressure_full_spikes_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "cpu", "full", spikes=True)
-                nodes["nodes"][node["node"]]["cpu_pressure_hot"] = False
                 nodes["nodes"][node["node"]]["memory_total"] = node["maxmem"]
                 nodes["nodes"][node["node"]]["memory_assigned"] = 0
                 nodes["nodes"][node["node"]]["memory_used"] = node["mem"]
@@ -84,11 +78,6 @@ class Nodes:
                 nodes["nodes"][node["node"]]["memory_assigned_percent"] = nodes["nodes"][node["node"]]["memory_assigned"] / nodes["nodes"][node["node"]]["memory_total"] * 100
                 nodes["nodes"][node["node"]]["memory_free_percent"] = nodes["nodes"][node["node"]]["memory_free"] / node["maxmem"] * 100
                 nodes["nodes"][node["node"]]["memory_used_percent"] = nodes["nodes"][node["node"]]["memory_used"] / node["maxmem"] * 100
-                nodes["nodes"][node["node"]]["memory_pressure_some_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "memory", "some")
-                nodes["nodes"][node["node"]]["memory_pressure_full_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "memory", "full")
-                nodes["nodes"][node["node"]]["memory_pressure_some_spikes_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "memory", "some", spikes=True)
-                nodes["nodes"][node["node"]]["memory_pressure_full_spikes_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "memory", "full", spikes=True)
-                nodes["nodes"][node["node"]]["memory_pressure_hot"] = False
                 nodes["nodes"][node["node"]]["disk_total"] = node["maxdisk"]
                 nodes["nodes"][node["node"]]["disk_assigned"] = 0
                 nodes["nodes"][node["node"]]["disk_used"] = node["disk"]
@@ -96,22 +85,43 @@ class Nodes:
                 nodes["nodes"][node["node"]]["disk_assigned_percent"] = nodes["nodes"][node["node"]]["disk_assigned"] / nodes["nodes"][node["node"]]["disk_total"] * 100
                 nodes["nodes"][node["node"]]["disk_free_percent"] = nodes["nodes"][node["node"]]["disk_free"] / node["maxdisk"] * 100
                 nodes["nodes"][node["node"]]["disk_used_percent"] = nodes["nodes"][node["node"]]["disk_used"] / node["maxdisk"] * 100
-                nodes["nodes"][node["node"]]["disk_pressure_some_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "disk", "some")
-                nodes["nodes"][node["node"]]["disk_pressure_full_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "disk", "full")
-                nodes["nodes"][node["node"]]["disk_pressure_some_spikes_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "disk", "some", spikes=True)
-                nodes["nodes"][node["node"]]["disk_pressure_full_spikes_percent"] = Nodes.get_node_rrd_data(proxmox_api, node["node"], "disk", "full", spikes=True)
-                nodes["nodes"][node["node"]]["disk_pressure_hot"] = False
 
                 # Evaluate if node should be set to maintenance mode
-                if Nodes.set_node_maintenance(proxmox_api, proxlb_config, node["node"]):
+                if Nodes.set_node_maintenance(proxlb_config, node["node"]):
                     nodes["nodes"][node["node"]]["maintenance"] = True
 
-        logger.debug(f"Node metrics collected: {nodes}")
+                # Generate the intial cluster statistics within the same loop to avoid a further one.
+                logger.debug(f"Updating cluster statistics by online node {node['node']}.")
+                cluster["cluster"]["node_count"] = cluster["cluster"].get("node_count", 0) + 1
+                cluster["cluster"]["cpu_total"] = cluster["cluster"].get("cpu_total", 0) + nodes["nodes"][node["node"]]["cpu_total"]
+                cluster["cluster"]["cpu_used"] = cluster["cluster"].get("cpu_used", 0) + nodes["nodes"][node["node"]]["cpu_used"]
+                cluster["cluster"]["cpu_free"] = cluster["cluster"].get("cpu_free", 0) + nodes["nodes"][node["node"]]["cpu_free"]
+                cluster["cluster"]["cpu_free_percent"] = cluster["cluster"].get("cpu_free", 0) / cluster["cluster"].get("cpu_total", 0) * 100
+                cluster["cluster"]["cpu_used_percent"] = cluster["cluster"].get("cpu_used", 0) / cluster["cluster"].get("cpu_total", 0) * 100
+                cluster["cluster"]["memory_total"] = cluster["cluster"].get("memory_total", 0) + nodes["nodes"][node["node"]]["memory_total"]
+                cluster["cluster"]["memory_used"] = cluster["cluster"].get("memory_used", 0) + nodes["nodes"][node["node"]]["memory_used"]
+                cluster["cluster"]["memory_free"] = cluster["cluster"].get("memory_free", 0) + nodes["nodes"][node["node"]]["memory_free"]
+                cluster["cluster"]["memory_free_percent"] = cluster["cluster"].get("memory_free", 0) / cluster["cluster"].get("memory_total", 0) * 100
+                cluster["cluster"]["memory_used_percent"] = cluster["cluster"].get("memory_used", 0) / cluster["cluster"].get("memory_total", 0) * 100
+                cluster["cluster"]["disk_total"] = cluster["cluster"].get("disk_total", 0) + nodes["nodes"][node["node"]]["disk_total"]
+                cluster["cluster"]["disk_used"] = cluster["cluster"].get("disk_used", 0) + nodes["nodes"][node["node"]]["disk_used"]
+                cluster["cluster"]["disk_free"] = cluster["cluster"].get("disk_free", 0) + nodes["nodes"][node["node"]]["disk_free"]
+                cluster["cluster"]["disk_free_percent"] = cluster["cluster"].get("disk_free", 0) / cluster["cluster"].get("disk_total", 0) * 100
+                cluster["cluster"]["disk_used_percent"] = cluster["cluster"].get("disk_used", 0) / cluster["cluster"].get("disk_total", 0) * 100
+
+                cluster["cluster"]["node_count_available"] = cluster["cluster"].get("node_count_available", 0) + 1
+                cluster["cluster"]["node_count_overall"] = cluster["cluster"].get("node_count_overall", 0) + 1
+
+            # Update the cluster statistics by offline nodes to have the overall count of nodes in the cluster
+            else:
+                logger.debug(f"Updating cluster statistics by offline node {node['node']}.")
+                cluster["cluster"]["node_count_overall"] = cluster["cluster"].get("node_count_overall", 0) + 1
+
         logger.debug("Finished: get_nodes.")
-        return nodes
+        return nodes, cluster
 
     @staticmethod
-    def set_node_maintenance(proxmox_api, proxlb_config: Dict[str, Any], node_name: str) -> Dict[str, Any]:
+    def set_node_maintenance(proxlb_config: Dict[str, Any], node_name: str) -> Dict[str, Any]:
         """
         Set nodes to maintenance mode based on the provided configuration.
 
@@ -119,7 +129,6 @@ class Nodes:
         based on the configuration provided in proxlb_config.
 
         Args:
-            proxmox_api (any): The Proxmox API client instance.
             proxlb_config (Dict[str, Any]): A dictionary containing the ProxLB configuration, including maintenance nodes.
             node_name: (str): The current node name within the outer iteration.
 
@@ -128,24 +137,11 @@ class Nodes:
         """
         logger.debug("Starting: set_node_maintenance.")
 
-        # Evaluate maintenance mode by config
         if proxlb_config.get("proxmox_cluster", None).get("maintenance_nodes", None) is not None:
             if len(proxlb_config.get("proxmox_cluster", {}).get("maintenance_nodes", [])) > 0:
                 if node_name in proxlb_config.get("proxmox_cluster", {}).get("maintenance_nodes", []):
-                    logger.info(f"Node: {node_name} has been set to maintenance mode (by ProxLB config).")
+                    logger.warning(f"Node: {node_name} has been set to maintenance mode.")
                     return True
-                else:
-                    logger.debug(f"Node: {node_name} is not in maintenance mode by ProxLB config.")
-
-        # Evaluate maintenance mode by Proxmox HA
-        for ha_element in proxmox_api.cluster.ha.status.current.get():
-            if ha_element.get("status"):
-                if "maintenance mode" in ha_element.get("status"):
-                    if ha_element.get("node") == node_name:
-                        logger.info(f"Node: {node_name} has been set to maintenance mode (by Proxmox HA API).")
-                        return True
-                    else:
-                        logger.debug(f"Node: {node_name} is not in maintenance mode by Proxmox HA API.")
 
         logger.debug("Finished: set_node_maintenance.")
 
@@ -169,87 +165,7 @@ class Nodes:
         if proxlb_config.get("proxmox_cluster", None).get("ignore_nodes", None) is not None:
             if len(proxlb_config.get("proxmox_cluster", {}).get("ignore_nodes", [])) > 0:
                 if node_name in proxlb_config.get("proxmox_cluster", {}).get("ignore_nodes", []):
-                    logger.info(f"Node: {node_name} has been set to be ignored. Not adding node!")
+                    logger.warning(f"Node: {node_name} has been set to be ignored. Not adding node!")
                     return True
 
         logger.debug("Finished: set_node_ignore.")
-
-    @staticmethod
-    def get_node_rrd_data(proxmox_api, node_name: str, object_name: str, object_type: str, spikes=False) -> float:
-        """
-        Retrieves the rrd data metrics for a specific resource (CPU, memory, disk) of a node.
-
-        Args:
-            proxmox_api (Any): The Proxmox API client instance.
-            node_name (str): The name of the node hosting the guest.
-            object_name (str): The resource type to query (e.g., 'cpu', 'memory', 'disk').
-            object_type (str, optional): The pressure type ('some', 'full') or None for average usage.
-            spikes (bool, optional): Whether to consider spikes in the calculation. Defaults to False.
-
-        Returns:
-            float: The calculated average usage value for the specified resource.
-        """
-        logger.debug("Starting: get_node_rrd_data.")
-        time.sleep(0.1)
-
-        try:
-            if spikes:
-                logger.debug(f"Getting spike RRD data for {object_name} from node: {node_name}.")
-                node_data_rrd = proxmox_api.nodes(node_name).rrddata.get(timeframe="hour", cf="MAX")
-            else:
-                logger.debug(f"Getting average RRD data for {object_name} from node: {node_name}.")
-                node_data_rrd = proxmox_api.nodes(node_name).rrddata.get(timeframe="hour", cf="AVERAGE")
-
-        except Exception:
-            logger.error(f"Failed to retrieve RRD data for guest: {node_name}. Using 0.0 as value.")
-            logger.debug("Finished: get_node_rrd_data.")
-            return 0.0
-
-        lookup_key = f"pressure{object_name}{object_type}"
-
-        if spikes:
-            # RRD data is collected every minute, so we look at the last 6 entries
-            # and take the maximum value to represent the spike
-            rrd_data_value = [row.get(lookup_key) for row in node_data_rrd if row.get(lookup_key) is not None]
-            rrd_data_value = max(rrd_data_value[-6:], default=0.0)
-        else:
-            # Calculate the average value from the RRD data entries
-            rrd_data_value = sum(entry.get(lookup_key, 0.0) for entry in node_data_rrd) / len(node_data_rrd)
-
-        logger.debug(f"RRD data (spike: {spikes}) for {object_name} from node: {node_name}: {rrd_data_value}")
-        logger.debug("Finished: get_node_rrd_data.")
-        return rrd_data_value
-
-    @staticmethod
-    def get_node_pve_version(proxmox_api, node_name: str) -> float:
-        """
-        Return the Proxmox VE (PVE) version for a given node by querying the Proxmox API.
-
-        This function calls proxmox_api.nodes(node_name).version.get() and extracts the
-        'version' field from the returned mapping. The value is expected to be numeric
-        (or convertible to float) and is returned as a float.
-
-        Args:
-            proxmox_api (Any): The Proxmox API client instance.
-            node_name (str): The name of the node hosting the guest.
-
-        Returns:
-            float: The PVE version for the specified node as a floating point number.
-
-        Raises:
-        Exception: If the proxmox_api call fails, returns an unexpected structure, or the
-                   'version' field is missing or cannot be converted to float. Callers should
-                    handle or propagate exceptions as appropriate.
-        """
-        logger.debug("Starting: get_node_pve_version.")
-        time.sleep(0.1)
-
-        try:
-            logger.debug(f"Trying to get PVE version for node: {node_name}.")
-            version = proxmox_api.nodes(node_name).version.get()
-        except Exception:
-            logger.error(f"Failed to get PVE version for node: {node_name}.")
-
-        logger.debug(f"Got version {version['version']} for node {node_name}.")
-        logger.debug("Finished: get_node_pve_version.")
-        return version["version"]

proxlb/models/pools.py

@@ -1,117 +0,0 @@
-"""
-The Pools class retrieves all present pools defined on a Proxmox cluster
-including the chield objects.
-"""
-
-__author__ = "Florian Paul Azim Hoberg <gyptazy>"
-__copyright__ = "Copyright (C) 2025 Florian Paul Azim Hoberg (@gyptazy)"
-__license__ = "GPL-3.0"
-
-
-from typing import Dict, Any
-from utils.logger import SystemdLogger
-from models.tags import Tags
-import time
-
-logger = SystemdLogger()
-
-
-class Pools:
-    """
-    The Pools class retrieves all present pools defined on a Proxmox cluster
-    including the chield objects.
-
-    Methods:
-        __init__:
-            Initializes the Pools class.
-
-        get_pools(proxmox_api: any) -> Dict[str, Any]:
-            Retrieve pool definitions and membership from the Proxmox cluster.
-            Returns a dict with a top-level "pools" mapping each poolid to
-            {"name": <poolid>, "members": [<member_names>...]}.
-            This method does not collect per-member metrics or perform node filtering.
-    """
-    def __init__(self):
-        """
-        Initializes the Pools class with the provided ProxLB data.
-        """
-
-    @staticmethod
-    def get_pools(proxmox_api: any) -> Dict[str, Any]:
-        """
-        Retrieve all pools and their members from a Proxmox cluster.
-
-        Queries the Proxmox API for pool definitions and returns a dictionary
-        containing each pool's id/name and a list of its member VM/CT names.
-        This function does not perform per-member metric collection or node
-        filtering — it only gathers pool membership information.
-
-        Args:
-            proxmox_api (any): Proxmox API client instance.
-
-        Returns:
-            Dict[str, Any]: Dictionary with a top-level "pools" key mapping poolid
-                    to {"name": <poolid>, "members": [<member_names>...]}.
-        """
-        logger.debug("Starting: get_pools.")
-        pools = {"pools": {}}
-
-        # Pool objects: iterate over all pools in the cluster.
-        # We keep pool members even if their nodes are ignored so resource accounting
-        # for rebalancing remains correct and we avoid overprovisioning nodes.
-        for pool in proxmox_api.pools.get():
-            logger.debug(f"Got pool: {pool['poolid']}")
-            pools['pools'][pool['poolid']] = {}
-            pools['pools'][pool['poolid']]['name'] = pool['poolid']
-            pools['pools'][pool['poolid']]['members'] = []
-
-            # Fetch pool details and collect member names
-            pool_details = proxmox_api.pools(pool['poolid']).get()
-            for member in pool_details.get("members", []):
-
-                # We might also have objects without the key "name", e.g. storage pools
-                if "name" not in member:
-                    logger.debug(f"Skipping member without name in pool: {pool['poolid']}")
-                    continue
-
-                logger.debug(f"Got member: {member['name']} for pool: {pool['poolid']}")
-                pools['pools'][pool['poolid']]['members'].append(member["name"])
-
-        logger.debug("Finished: get_pools.")
-        return pools
-
-    @staticmethod
-    def get_pools_for_guest(guest_name: str, pools: Dict[str, Any]) -> Dict[str, Any]:
-        """
-        Return the list of pool names that include the given guest.
-
-        Args:
-            guest_name (str): Name of the VM or CT to look up.
-            pools (Dict[str, Any]): Pools structure as returned by get_pools(),
-            expected to contain a top-level "pools" mapping each poolid to
-            {"name": <poolid>, "members": [<member_names>...]}.
-
-        Returns:
-            list[str]: Names of pools the guest is a member of (empty list if none).
-        """
-        logger.debug("Starting: get_pools_for_guests.")
-        guest_pools = []
-
-        for pool in pools.items():
-            for pool_id, pool_data in pool[1].items():
-
-                if type(pool_data) is dict:
-                    pool_name = pool_data.get("name", "")
-                    pool_name_members = pool_data.get("members", [])
-
-                    if guest_name in pool_name_members:
-                        logger.debug(f"Guest: {guest_name} is member of Pool: {pool_name}.")
-                        guest_pools.append(pool_name)
-                    else:
-                        logger.debug(f"Guest: {guest_name} is NOT member of Pool: {pool_name}.")
-
-                else:
-                    logger.debug(f"Pool data for pool_id {pool_id} is not a dict: {pool_data}")
-
-        logger.debug("Finished: get_pools_for_guests.")
-        return guest_pools

proxlb/models/tags.py

@@ -12,9 +12,7 @@ __license__ = "GPL-3.0"
 
 import time
 from typing import List
-from typing import Dict, Any
 from utils.logger import SystemdLogger
-from utils.helper import Helper
 
 logger = SystemdLogger()
 
@@ -80,18 +78,15 @@ class Tags:
         return tags
 
     @staticmethod
-    def get_affinity_groups(tags: List[str], pools: List[str], proxlb_config: Dict[str, Any]) -> List[str]:
+    def get_affinity_groups(tags: List[str]) -> List[str]:
         """
         Get affinity tags for a guest from the Proxmox cluster by the API.
 
-        This method retrieves all tags for a given guest or based on a
-        membership of a pool and evaluates the affinity groups which are
-        required during the balancing calculations.
+        This method retrieves all tags for a given guest and evaluates the
+        affinity tags which are required during the balancing calculations.
 
         Args:
             tags (List): A list holding all defined tags for a given guest.
-            pools (List): A list holding all defined pools for a given guest.
-            proxlb_config (Dict): A dict holding the ProxLB configuration.
 
         Returns:
             List: A list including all affinity tags for the given guest.
@@ -102,36 +97,21 @@ class Tags:
         if len(tags) > 0:
             for tag in tags:
                 if tag.startswith("plb_affinity"):
-                    logger.debug(f"Adding affinity group for tag {tag}.")
                     affinity_tags.append(tag)
-                else:
-                    logger.debug(f"Skipping affinity group for tag {tag}.")
-
-        if len(pools) > 0:
-            for pool in pools:
-                if pool in (proxlb_config['balancing'].get('pools') or {}):
-                    if proxlb_config['balancing']['pools'][pool].get('type', None) == 'affinity':
-                        logger.debug(f"Adding affinity group for pool {pool}.")
-                        affinity_tags.append(pool)
-                else:
-                    logger.debug(f"Skipping affinity group for pool {pool}.")
 
         logger.debug("Finished: get_affinity_groups.")
         return affinity_tags
 
     @staticmethod
-    def get_anti_affinity_groups(tags: List[str], pools: List[str], proxlb_config: Dict[str, Any]) -> List[str]:
+    def get_anti_affinity_groups(tags: List[str]) -> List[str]:
         """
         Get anti-affinity tags for a guest from the Proxmox cluster by the API.
 
-        This method retrieves all tags for a given guest or based on a
-        membership of a pool and evaluates the anti-affinity groups which
-        are required during the balancing calculations.
+        This method retrieves all tags for a given guest and evaluates the
+        anti-affinity tags which are required during the balancing calculations.
 
         Args:
             tags (List): A list holding all defined tags for a given guest.
-            pools (List): A list holding all defined pools for a given guest.
-            proxlb_config (Dict): A dict holding the ProxLB configuration.
 
         Returns:
             List: A list including all anti-affinity tags for the given guest..
@@ -142,19 +122,7 @@ class Tags:
         if len(tags) > 0:
             for tag in tags:
                 if tag.startswith("plb_anti_affinity"):
-                    logger.debug(f"Adding anti-affinity group for tag {tag}.")
                     anti_affinity_tags.append(tag)
-                else:
-                    logger.debug(f"Skipping anti-affinity group for tag {tag}.")
-
-        if len(pools) > 0:
-            for pool in pools:
-                if pool in (proxlb_config['balancing'].get('pools') or {}):
-                    if proxlb_config['balancing']['pools'][pool].get('type', None) == 'anti-affinity':
-                        logger.debug(f"Adding anti-affinity group for pool {pool}.")
-                        anti_affinity_tags.append(pool)
-                else:
-                    logger.debug(f"Skipping anti-affinity group for pool {pool}.")
 
         logger.debug("Finished: get_anti_affinity_groups.")
         return anti_affinity_tags
@@ -185,55 +153,27 @@ class Tags:
         return ignore_tag
 
     @staticmethod
-    def get_node_relationships(tags: List[str], nodes: Dict[str, Any], pools: List[str], proxlb_config: Dict[str, Any]) -> str:
+    def get_node_relationship(tags: List[str]) -> str:
         """
         Get a node relationship tag for a guest from the Proxmox cluster by the API to pin
-        a guest to a node or by defined pools from ProxLB configuration.
+        a guest to a node.
 
         This method retrieves a relationship tag between a guest and a specific
         hypervisor node to pin the guest to a specific node (e.g., for licensing reason).
 
         Args:
             tags (List): A list holding all defined tags for a given guest.
-            nodes (Dict): A dictionary holding all available nodes in the cluster.
-            pools (List): A list holding all defined pools for a given guest.
-            proxlb_config (Dict): A dict holding the ProxLB configuration.
 
         Returns:
-            Str: The related hypervisor node name(s).
+            Str: The related hypervisor node name.
         """
-        logger.debug("Starting: get_node_relationships.")
-        node_relationship_tags = []
+        logger.debug("Starting: get_node_relationship.")
+        node_relationship_tag = False
 
         if len(tags) > 0:
-            logger.debug("Validating node pinning by tags.")
             for tag in tags:
                 if tag.startswith("plb_pin"):
                     node_relationship_tag = tag.replace("plb_pin_", "")
 
-                    # Validate if the node to pin is present in the cluster
-                    if Helper.validate_node_presence(node_relationship_tag, nodes):
-                        logger.debug(f"Tag {node_relationship_tag} is valid! Defined node exists in the cluster.")
-                        logger.debug(f"Setting node relationship because of tag {tag} to {node_relationship_tag}.")
-                        node_relationship_tags.append(node_relationship_tag)
-                    else:
-                        logger.warning(f"Tag {node_relationship_tag} is invalid! Defined node does not exist in the cluster. Not applying pinning.")
-
-        if len(pools) > 0:
-            logger.debug("Validating node pinning by pools.")
-            for pool in pools:
-                if pool in (proxlb_config['balancing'].get('pools') or {}):
-
-                    node = proxlb_config['balancing']['pools'][pool].get('pin', None)
-                    # Validate if the node to pin is present in the cluster
-                    if Helper.validate_node_presence(node, nodes):
-                        logger.debug(f"Pool pinning tag {node} is valid! Defined node exists in the cluster.")
-                        logger.debug(f"Setting node relationship because of pool {pool} to {node}.")
-                        node_relationship_tags.append(node)
-                    else:
-                        logger.warning(f"Pool pinning tag {node} is invalid! Defined node does not exist in the cluster. Not applying pinning.")
-                else:
-                    logger.debug(f"Skipping pinning for pool {pool}. Pool is not defined in ProxLB configuration.")
-
-        logger.debug("Finished: get_node_relationships.")
-        return node_relationship_tags
+        logger.debug("Finished: get_node_relationship.")
+        return node_relationship_tag

proxlb/utils/helper.py

@@ -10,14 +10,11 @@ __license__ = "GPL-3.0"
 
 import json
 import uuid
-import re
-import socket
 import sys
 import time
 import utils.version
 from utils.logger import SystemdLogger
 from typing import Dict, Any
-from types import FrameType
 
 logger = SystemdLogger()
 
@@ -43,8 +40,6 @@ class Helper:
         get_daemon_mode(proxlb_config: Dict[str, Any]) -> None:
             Checks if the daemon mode is active and handles the scheduling accordingly.
     """
-    proxlb_reload = False
-
     def __init__(self):
         """
         Initializes the general Helper clas.
@@ -148,39 +143,6 @@ class Helper:
 
         logger.debug("Finished: get_daemon_mode.")
 
-    @staticmethod
-    def get_service_delay(proxlb_config: Dict[str, Any]) -> None:
-        """
-        Checks if a start up delay for the service is defined and waits to proceed until
-        the time is up.
-
-        Parameters:
-            proxlb_config (Dict[str, Any]): A dictionary containing the ProxLB configuration.
-
-        Returns:
-            None
-        """
-        logger.debug("Starting: get_service_delay.")
-        if proxlb_config.get("service", {}).get("delay", {}).get("enable", False):
-
-            # Convert hours to seconds
-            if proxlb_config["service"]["delay"].get("format", "hours") == "hours":
-                sleep_seconds = proxlb_config.get("service", {}).get("delay", {}).get("time", 1) * 3600
-            # Convert minutes to seconds
-            elif proxlb_config["service"]["delay"].get("format", "hours") == "minutes":
-                sleep_seconds = proxlb_config.get("service", {}).get("delay", {}).get("time", 60) * 60
-            else:
-                logger.error("Invalid format for service delay. Please use 'hours' or 'minutes'.")
-                sys.exit(1)
-
-            logger.info(f"Service delay active: First run in: {proxlb_config.get('service', {}).get('delay', {}).get('time', 1)} {proxlb_config['service']['delay'].get('format', 'hours')}.")
-            time.sleep(sleep_seconds)
-
-        else:
-            logger.debug("Service delay not active. Proceeding without delay.")
-
-        logger.debug("Finished: get_service_delay.")
-
     @staticmethod
     def print_json(proxlb_config: Dict[str, Any], print_json: bool = False) -> None:
         """
@@ -200,136 +162,3 @@ class Helper:
             print(json.dumps(filtered_data, indent=4))
 
         logger.debug("Finished: print_json.")
-
-    @staticmethod
-    def handler_sighup(signum: int, frame: FrameType) -> None:
-        """
-        Signal handler for SIGHUP.
-
-        This method is triggered when the process receives a SIGHUP signal.
-        It sets the `proxlb_reload` class variable to True to indicate that
-        configuration should be reloaded in the main loop.
-
-        Args:
-            signum (int): The signal number (expected to be signal.SIGHUP).
-            frame (frame object): Current stack frame (unused but required by signal handler signature).
-        """
-        logger.debug("Starting: handle_sighup.")
-        logger.debug("Got SIGHUP signal. Reloading...")
-        Helper.proxlb_reload = True
-        logger.debug("Finished: handle_sighup.")
-
-    @staticmethod
-    def handler_sigint(signum: int, frame: FrameType) -> None:
-        """
-        Signal handler for SIGINT. (triggered by CTRL+C).
-
-        Args:
-            signum (int): The signal number (e.g., SIGINT).
-            frame (FrameType): The current stack frame when the signal was received.
-
-        Returns:
-            None
-        """
-        exit_message = "ProxLB has been successfully terminated by user."
-        logger.debug(exit_message)
-        print(f"\n {exit_message}")
-        sys.exit(0)
-
-    @staticmethod
-    def get_host_port_from_string(host_object):
-        """
-        Parses a string containing a host (IPv4, IPv6, or hostname) and an optional port, and returns a tuple of (host, port).
-
-        Supported formats:
-        - Hostname or IPv4 without port: "example.com" or "192.168.0.1"
-        - Hostname or IPv4 with port: "example.com:8006" or "192.168.0.1:8006"
-        - IPv6 in brackets with optional port: "[fc00::1]" or "[fc00::1]:8006"
-        - IPv6 without brackets, port is assumed after last colon: "fc00::1:8006"
-
-        If no port is specified, port 8006 is used as the default.
-
-        Args:
-            host_object (str): A string representing a host with or without a port.
-
-        Returns:
-            tuple: A tuple (host: str, port: int)
-        """
-        logger.debug("Starting: get_host_port_from_string.")
-
-        # IPv6 (with or without port, written in brackets)
-        match = re.match(r'^\[(.+)\](?::(\d+))?$', host_object)
-        if match:
-            host = match.group(1)
-            port = int(match.group(2)) if match.group(2) else 8006
-            return host, port
-
-        # Count colons to identify IPv6 addresses without brackets
-        colon_count = host_object.count(':')
-
-        # IPv4 or hostname without port
-        if colon_count == 0:
-            return host_object, 8006
-
-        # IPv4 or hostname with port
-        elif colon_count == 1:
-            host, port = host_object.split(':')
-            return host, int(port)
-
-        # IPv6 (with or without port, assume last colon is port)
-        else:
-            parts = host_object.rsplit(':', 1)
-            try:
-                port = int(parts[1])
-                return parts[0], port
-            except ValueError:
-                return host_object, 8006
-
-    @staticmethod
-    def validate_node_presence(node: str, nodes: Dict[str, Any]) -> bool:
-        """
-        Validates whether a given node exists in the provided cluster nodes dictionary.
-
-        Args:
-            node (str): The name of the node to validate.
-            nodes (Dict[str, Any]): A dictionary containing cluster information.
-                                    Must include a "nodes" key mapping to a dict of available nodes.
-
-        Returns:
-            bool: True if the node exists in the cluster, False otherwise.
-        """
-        logger.debug("Starting: validate_node_presence.")
-
-        if node in nodes["nodes"].keys():
-            logger.info(f"Node {node} found in cluster. Applying pinning.")
-            logger.debug("Finished: validate_node_presence.")
-            return True
-        else:
-            logger.warning(f"Node {node} not found in cluster. Not applying pinning!")
-            logger.debug("Finished: validate_node_presence.")
-            return False
-
-    @staticmethod
-    def tcp_connect_test(addr_family: int, host: str, port: int, timeout: int) -> tuple[bool, int | None]:
-        """
-        Attempt a TCP connection to the specified host and port to test the reachability.
-
-        Args:
-            addr_family (int): Address family for the socket (e.g., socket.AF_INET for IPv4, socket.AF_INET6 for IPv6).
-            host (str): The hostname or IP address to connect to.
-            port (int): The port number to connect to.
-            timeout (int): Connection timeout in seconds.
-
-        Returns:
-            tuple[bool, int | None]: A tuple containing:
-                - bool: True if the connection was successful, False otherwise.
-                - int | None: None if the connection was successful, otherwise the errno code indicating the reason for failure.
-        """
-        test_socket = socket.socket(addr_family, socket.SOCK_STREAM)
-        test_socket.settimeout(timeout)
-
-        try:
-            rc = test_socket.connect_ex((host, port))
-            return (rc == 0, rc if rc != 0 else None)
-        finally:
-            test_socket.close()

proxlb/utils/logger.py

@@ -88,7 +88,7 @@ class SystemdLogger:
         # logging is preferred.
         if SYSTEMD_PRESENT:
             # Add a JournalHandler for systemd integration
-            handler = JournalHandler(SYSLOG_IDENTIFIER="ProxLB")
+            handler = JournalHandler()
         else:
             # Add a stdout handler as a fallback
             handler = logging.StreamHandler(sys.stdout)

proxlb/utils/proxmox_api.py

@@ -13,7 +13,6 @@ __copyright__ = "Copyright (C) 2025 Florian Paul Azim Hoberg (@gyptazy)"
 __license__ = "GPL-3.0"
 
 
-import errno
 try:
     import proxmoxer
     PROXMOXER_PRESENT = True
@@ -34,7 +33,6 @@ try:
 except ImportError:
     URLLIB3_PRESENT = False
 from typing import Dict, Any
-from utils.helper import Helper
 from utils.logger import SystemdLogger
 
 
@@ -136,14 +134,6 @@ class ProxmoxApi:
         proxlb_credentials = proxlb_config["proxmox_api"]
         present_auth_pass = "pass" in proxlb_credentials
         present_auth_secret = "token_secret" in proxlb_credentials
-        token_id = proxlb_credentials.get("token_id", None)
-
-        if token_id:
-            non_allowed_chars = ["@", "!"]
-            for char in non_allowed_chars:
-                if char in token_id:
-                    logger.error(f"Wrong user/token format defined. User and token id must be splitted! Please see: https://github.com/gyptazy/ProxLB/blob/main/docs/03_configuration.md#required-permissions-for-a-user")
-                    sys.exit(1)
 
         if present_auth_pass and present_auth_secret:
             logger.critical(f"Username/password and API token authentication are mutal exclusive. Please use only one!")
@@ -176,40 +166,49 @@ class ProxmoxApi:
         logger.debug("Starting: api_connect_get_hosts.")
         # Pre-validate the given API endpoints
         if not isinstance(proxmox_api_endpoints, list):
-            logger.critical("The proxmox_api hosts are not defined as a list type.")
+            logger.critical(f"The proxmox_api hosts are not defined as a list type.")
             sys.exit(1)
+
         if not proxmox_api_endpoints:
-            logger.critical("No proxmox_api hosts are defined.")
+            logger.critical(f"No proxmox_api hosts are defined.")
             sys.exit(1)
 
-        validated_api_hosts: list[tuple[str, int]] = []
+        if len(proxmox_api_endpoints) == 0:
+            logger.critical(f"No proxmox_api hosts are defined.")
+            sys.exit(1)
 
+        # If we have multiple Proxmox API endpoints, we need to check each one by
+        # doing a connection attempt for IPv4 and IPv6. If we find a working one,
+        # we return that one. This allows us to define multiple endpoints in a cluster.
+        validated_api_hosts = []
         for host in proxmox_api_endpoints:
-            retries = proxlb_config["proxmox_api"].get("retries", 1)
-            wait_time = proxlb_config["proxmox_api"].get("wait_time", 1)
 
-            for attempt in range(retries):
-                candidate_host, candidate_port = self.test_api_proxmox_host(host)
-                if candidate_host:
-                    validated_api_hosts.append((candidate_host, candidate_port))
+            # Get or set a default value for a maximum of retries when connecting to
+            # the Proxmox API
+            api_connection_retries = proxlb_config["proxmox_api"].get("retries", 1)
+            api_connection_wait_time = proxlb_config["proxmox_api"].get("wait_time", 1)
+
+            for api_connection_attempt in range(api_connection_retries):
+                validated = self.test_api_proxmox_host(host)
+                if validated:
+                    validated_api_hosts.append(validated)
                     break
                 else:
-                    logger.warning(
-                        f"Attempt {attempt + 1}/{retries} failed for host {host}. "
-                        f"Retrying in {wait_time} seconds..."
-                    )
-                    time.sleep(wait_time)
+                    logger.warning(f"Attempt {api_connection_attempt + 1}/{api_connection_retries} failed for host {host}. Retrying in {api_connection_wait_time} seconds...")
+                    time.sleep(api_connection_wait_time)
 
-        if validated_api_hosts:
-            chosen_host, chosen_port = random.choice(validated_api_hosts)
-            return chosen_host, chosen_port
+        if len(validated_api_hosts) > 0:
+            # Choose a random host to distribute the load across the cluster
+            # as a simple load balancing mechanism.
+            return random.choice(validated_api_hosts)
 
         logger.critical("No valid Proxmox API hosts found.")
         print("No valid Proxmox API hosts found.")
+
         logger.debug("Finished: api_connect_get_hosts.")
         sys.exit(1)
 
-    def test_api_proxmox_host(self, host: str) -> tuple[str, int | None, None]:
+    def test_api_proxmox_host(self, host: str) -> str:
         """
         Tests the connectivity to a Proxmox host by resolving its IP address and
         checking both IPv4 and IPv6 addresses.
@@ -229,36 +228,27 @@ class ProxmoxApi:
         """
         logger.debug("Starting: test_api_proxmox_host.")
 
-        # Validate for custom port configurations (e.g., by given external
-        # loadbalancer systems)
-        host, port = Helper.get_host_port_from_string(host)
-        if port is None:
-            port = 8006
-
         # Try resolving DNS to IP and log non-resolvable ones
         try:
-            infos = socket.getaddrinfo(host, None, socket.AF_UNSPEC)
+            ip = socket.getaddrinfo(host, None, socket.AF_UNSPEC)
         except socket.gaierror:
             logger.warning(f"Could not resolve {host}.")
-            return (None, None)
+            return False
 
-        # Check both families that are actually present
-        saw_family = set()
-        for family, *_rest in infos:
-            saw_family.add(family)
+        # Validate if given object is IPv4 or IPv6
+        for address_type in ip:
+            if address_type[0] == socket.AF_INET:
+                logger.debug(f"{host} is type ipv4.")
+                if self.test_api_proxmox_host_ipv4(host):
+                    return host
+            elif address_type[0] == socket.AF_INET6:
+                logger.debug(f"{host} is type ipv6.")
+                if self.test_api_proxmox_host_ipv6(host):
+                    return host
+            else:
+                return False
 
-        if socket.AF_INET in saw_family:
-            logger.debug(f"{host} has IPv4.")
-            if self.test_api_proxmox_host_ipv4(host, port):
-                return (host, port)
-
-        if socket.AF_INET6 in saw_family:
-            logger.debug(f"{host} has IPv6.")
-            if self.test_api_proxmox_host_ipv6(host, port):
-                return (host, port)
-
-        logger.debug("Finished: test_api_proxmox_host (unreachable).")
-        return (None, None)
+        logger.debug("Finished: test_api_proxmox_host.")
 
     def test_api_proxmox_host_ipv4(self, host: str, port: int = 8006, timeout: int = 1) -> bool:
         """
@@ -277,16 +267,18 @@ class ProxmoxApi:
             bool: True if the host is reachable on the specified port, False otherwise.
         """
         logger.debug("Starting: test_api_proxmox_host_ipv4.")
-        ok, rc = Helper.tcp_connect_test(socket.AF_INET, host, port, timeout)
-        if ok:
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.settimeout(timeout)
+        logger.warning(f"Warning: Host {host} ran into a timeout when connecting on IPv4 for tcp/{port}.")
+        result = sock.connect_ex((host, port))
+
+        if result == 0:
+            sock.close()
             logger.debug(f"Host {host} is reachable on IPv4 for tcp/{port}.")
-            logger.debug("Finished: test_api_proxmox_host_ipv4.")
             return True
 
-        if rc == errno.ETIMEDOUT:
-            logger.warning(f"Timeout connecting to {host} on IPv4 tcp/{port}.")
-        else:
-            logger.warning(f"Host {host} is unreachable on IPv4 for tcp/{port} (errno {rc}).")
+        sock.close()
+        logger.warning(f"Host {host} is unreachable on IPv4 for tcp/{port}.")
 
         logger.debug("Finished: test_api_proxmox_host_ipv4.")
         return False
@@ -308,16 +300,18 @@ class ProxmoxApi:
             bool: True if the host is reachable on the specified port, False otherwise.
         """
         logger.debug("Starting: test_api_proxmox_host_ipv6.")
-        ok, rc = Helper.tcp_connect_test(socket.AF_INET6, host, port, timeout)
-        if ok:
+        sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
+        sock.settimeout(timeout)
+        logger.warning(f"Host {host} ran into a timeout when connecting via IPv6 for tcp/{port}.")
+        result = sock.connect_ex((host, port))
+
+        if result == 0:
+            sock.close()
             logger.debug(f"Host {host} is reachable on IPv6 for tcp/{port}.")
-            logger.debug("Finished: test_api_proxmox_host_ipv6.")
             return True
 
-        if rc == errno.ETIMEDOUT:
-            logger.warning(f"Timeout connecting to {host} on IPv6 tcp/{port}.")
-        else:
-            logger.warning(f"Host {host} is unreachable on IPv6 for tcp/{port} (errno {rc}).")
+        sock.close()
+        logger.warning(f"Host {host} is unreachable on IPv6 for tcp/{port}.")
 
         logger.debug("Finished: test_api_proxmox_host_ipv6.")
         return False
@@ -337,15 +331,7 @@ class ProxmoxApi:
         permissions_available = []
 
         # Get the permissions for the current user/token from API
-        try:
-            permissions = proxmox_api.access.permissions.get()
-        except proxmoxer.core.ResourceException as api_error:
-            if "no such user" in str(api_error):
-                logger.error("Authentication to Proxmox API not possible: User not known - please check your username and config file.")
-                sys.exit(1)
-            else:
-                logger.error(f"Proxmox API error: {api_error}")
-                sys.exit(1)
+        permissions = proxmox_api.access.permissions.get()
 
         # Get all available permissions of the current user/token
         for path, permission in permissions.items():
@@ -392,7 +378,7 @@ class ProxmoxApi:
         self.validate_config(proxlb_config)
 
         # Get a valid Proxmox API endpoint
-        proxmox_api_endpoint, proxmox_api_port = self.api_connect_get_hosts(proxlb_config, proxlb_config.get("proxmox_api", {}).get("hosts", []))
+        proxmox_api_endpoint = self.api_connect_get_hosts(proxlb_config, proxlb_config.get("proxmox_api", {}).get("hosts", []))
 
         # Disable warnings for SSL certificate validation
         if not proxlb_config.get("proxmox_api").get("ssl_verification", True):
@@ -406,7 +392,6 @@ class ProxmoxApi:
             if proxlb_config.get("proxmox_api").get("token_secret", False):
                 proxmox_api = proxmoxer.ProxmoxAPI(
                     proxmox_api_endpoint,
-                    port=proxmox_api_port,
                     user=proxlb_config.get("proxmox_api").get("user", True),
                     token_name=proxlb_config.get("proxmox_api").get("token_id", True),
                     token_value=proxlb_config.get("proxmox_api").get("token_secret", True),
@@ -416,7 +401,6 @@ class ProxmoxApi:
             else:
                 proxmox_api = proxmoxer.ProxmoxAPI(
                     proxmox_api_endpoint,
-                    port=proxmox_api_port,
                     user=proxlb_config.get("proxmox_api").get("user", True),
                     password=proxlb_config.get("proxmox_api").get("pass", True),
                     verify_ssl=proxlb_config.get("proxmox_api").get("ssl_verification", True),
@@ -436,5 +420,6 @@ class ProxmoxApi:
             sys.exit(2)
 
         logger.info(f"API connection to host {proxmox_api_endpoint} succeeded.")
+
         logger.debug("Finished: api_connect.")
         return proxmox_api

proxlb/utils/version.py

@@ -1,7 +1,7 @@
 __app_name__ = "ProxLB"
-__app_desc__ = "An advanced resource scheduler and load balancer for Proxmox clusters."
+__app_desc__ = "A DRS alike loadbalancer for Proxmox clusters."
 __author__ = "Florian Paul Azim Hoberg <gyptazy>"
 __copyright__ = "Copyright (C) 2025 Florian Paul Azim Hoberg (@gyptazy)"
 __license__ = "GPL-3.0"
-__version__ = "1.1.10"
+__version__ = "1.1.2b"
 __url__ = "https://github.com/gyptazy/ProxLB"

requirements.txt

@@ -1,5 +1,4 @@
-packaging
 proxmoxer
 requests
 urllib3
-PyYAML
+PyYAML

service/proxlb.service

@@ -6,8 +6,6 @@ Wants=network-online.target pveproxy.service
 [Service]
 ExecStart=python3 /usr/lib/python3/dist-packages/proxlb/main.py -c /etc/proxlb/proxlb.yaml
 User=plb
-ExecReload=/bin/kill -HUP $MAINPID
-KillMode=process
 
 [Install]
 WantedBy=multi-user.target

setup.py

@@ -2,9 +2,9 @@ from setuptools import setup
 
 setup(
     name="proxlb",
-    version="1.1.10",
-    description="An advanced resource scheduler and load balancer for Proxmox clusters.",
-    long_description="An advanced resource scheduler and load balancer for Proxmox clusters that also supports maintenance modes and affinity/anti-affinity rules.",
+    version="1.1.2b",
+    description="A DRS alike loadbalancer for Proxmox clusters.",
+    long_description="An advanced DRS alike loadbalancer for Proxmox clusters that also supports maintenance modes and affinity/anti-affinity rules.",
     author="Florian Paul Azim Hoberg",
     author_email="gyptazy@gyptazy.com",
     maintainer="Florian Paul Azim Hoberg",