From da13337a18d2f109ef772d506854530b4b5b50bf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alo=C3=AFs=20Micard?= <a.micard@vold.lu>
Date: Wed, 12 Feb 2025 20:46:58 +0100
Subject: [PATCH] Handle disk backup information

---
 app/__main__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/app/__main__.py b/app/__main__.py
index 0491a08..b303e32 100644
--- a/app/__main__.py
+++ b/app/__main__.py
@@ -314,6 +314,7 @@ def _process_pve_virtual_machine_disks(
             _nb_virtual_machine,
             _disk_definition['name'],
             _process_pve_disk_size(_disk_definition['size']),
+            _disk_definition.get('backup', '1') == '1',
         )
 
     return _nb_objects
@@ -325,6 +326,7 @@ def _process_pve_virtual_machine_disk(
         _nb_virtual_machine: any,
         _disk_name: str,
         _disk_size: int,
+        _has_backup: bool,
 ) -> dict:
     nb_disk = _nb_objects['disks'].get(_nb_virtual_machine.id, {}).get(_disk_name)
     if nb_disk is None:
@@ -332,9 +334,13 @@ def _process_pve_virtual_machine_disk(
             name=_disk_name,
             size=_disk_size,
             virtual_machine=_nb_virtual_machine.id,
+            custom_fields={
+                'backup': _has_backup,
+            }
         )
     else:
         nb_disk.size = _disk_size
+        nb_disk.custom_fields['backup'] = _has_backup
         nb_disk.save()
 
     return _nb_objects