From 8669da435a6f0368e177c32133be107e1da4a9f0 Mon Sep 17 00:00:00 2001
From: mvrp21 <mvrp21@inf.ufpr.br>
Date: Tue, 25 Jun 2024 10:30:37 -0300
Subject: [PATCH] fix: vms ips update

Signed-off-by: mvrp21 <mvrp21@inf.ufpr.br>
---
 netbox_proxmox_sync/__init__.py           | 2 +-
 netbox_proxmox_sync/api/proxmox/update.py | 9 +++++++++
 setup.py                                  | 2 +-
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/netbox_proxmox_sync/__init__.py b/netbox_proxmox_sync/__init__.py
index b596542..ece8f09 100644
--- a/netbox_proxmox_sync/__init__.py
+++ b/netbox_proxmox_sync/__init__.py
@@ -5,7 +5,7 @@ class NetBoxProxmoxSyncConfig(PluginConfig):
     name = 'netbox_proxmox_sync'
     verbose_name = 'NetBox Proxmox Sync'
     description = 'Import cluster information from Proxmox into NetBox'
-    version = '1.0.1'
+    version = '1.0.2'
     base_url = 'netbox-proxmox-sync'
 
 
diff --git a/netbox_proxmox_sync/api/proxmox/update.py b/netbox_proxmox_sync/api/proxmox/update.py
index f437760..007bafa 100644
--- a/netbox_proxmox_sync/api/proxmox/update.py
+++ b/netbox_proxmox_sync/api/proxmox/update.py
@@ -64,6 +64,15 @@ def update_old_vms(old_vms, new_vms):
         # Without this the update is messed, because it's a dict, same for all the "del"s
         # below these ifs (long story, just believe me here)
         del old_vms[name]['site']
+        if old_vms[name].get('primary_ip4') is not None:
+            ip = old_vms[name]['primary_ip4']['address']
+            old_vms[name]['primary_ip4'] = {'address': ip}
+        if old_vms[name].get('primary_ip') is not None:
+            ip = old_vms[name]['primary_ip']['address']
+            old_vms[name]['primary_ip'] = {'address': ip}
+        if old_vms[name].get('primary_ip6') is not None:
+            ip = old_vms[name]['primary_ip6']['address']
+            old_vms[name]['primary_ip6'] = {'address': ip}
         # Iterate over properties, set new stuff
         for key in new_vms[name]:
             # Edge cases... 4 of them...
diff --git a/setup.py b/setup.py
index f17aa92..d9b13f8 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name='netbox-proxmox-sync',
-    version='1.0.1',
+    version='1.0.2',
     description='Import Proxmox cluster info into NetBox.',
     install_requires=['pynetbox', 'proxmoxer'],
     include_package_data=True,
-- 
GitLab