From c892d596f49cb1f89e3529f98726ee7be91d1902 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alo=C3=AFs=20Micard?= <a.micard@vold.lu>
Date: Thu, 1 May 2025 08:49:38 +0200
Subject: [PATCH] [#7] Improve tag handling

Closes: #7
---
 Changelog.md                | 4 ++++
 netbox_pve_sync/__init__.py | 4 +++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/Changelog.md b/Changelog.md
index c0ce759..57c612a 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -2,6 +2,10 @@
 
 ## [Unreleased]
 
+### Fixed
+
+- [#7] Improve tag handling.
+
 ## [0.2.0] - 21/04/2025
 
 ### New
diff --git a/netbox_pve_sync/__init__.py b/netbox_pve_sync/__init__.py
index 882b503..906bc41 100644
--- a/netbox_pve_sync/__init__.py
+++ b/netbox_pve_sync/__init__.py
@@ -88,6 +88,7 @@ def _process_pve_tags(
                 slug=f'pool-{_pve_pool["poolid"]}'.lower(),
                 description=f'Proxmox pool {_pve_pool["poolid"]}',
             )
+            _nb_objects['tags'][_nb_tag.name] = _nb_tag
 
     return _nb_objects
 
@@ -433,7 +434,8 @@ def main():
     for pve_vm_resource in pve_api.cluster.resources.get(type='vm'):
         pve_vm_tags[pve_vm_resource['vmid']] = []
 
-        pve_vm_tags[pve_vm_resource['vmid']].append(f'Pool/{pve_vm_resource["pool"]}')
+        if 'pool' in pve_vm_resource:
+            pve_vm_tags[pve_vm_resource['vmid']].append(f'Pool/{pve_vm_resource["pool"]}')
 
         if 'tags' in pve_vm_resource:
             pass  # TODO: pve_vm_tags[pve_vm_resource['vmid']].append(pve_vm_resource['tags'])