From 0c147768ba20f46e69261e8ee3aba492d61d3bdb Mon Sep 17 00:00:00 2001
From: Jon <jonathon.lockwood@networkedweb.com>
Date: Mon, 9 Jun 2025 09:09:15 +0930
Subject: [PATCH] feat(config_management): add ConfigGroupSoftware to
 history/notes migration

ref: #807 closes #794
---
 app/core/signal/centurion_model_migrate.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/core/signal/centurion_model_migrate.py b/app/core/signal/centurion_model_migrate.py
index 1abcd36f..1c31c51c 100644
--- a/app/core/signal/centurion_model_migrate.py
+++ b/app/core/signal/centurion_model_migrate.py
@@ -41,6 +41,12 @@ def centurion_model_migrate(sender, **kwargs):
             'history_model_name': 'ConfigGroupHostsHistory',
             'notes_model_name': None
         },
+        {
+            'app_label': 'config_management',
+            'model_name': 'ConfigGroupSoftware',
+            'history_model_name': 'ConfigGroupSoftwareHistory',
+            'notes_model_name': None
+        },
         {
             'app_label': 'devops',
             'model_name': 'FeatureFlag',
@@ -103,10 +109,14 @@ def centurion_model_migrate(sender, **kwargs):
 
                         try:
 
+                            entry_model = entry.model
+                            if hasattr(entry, 'child_model'):
+                                entry_model = entry.child_model
+
                             migrated_history = audit_history.objects.create(
                                 organization = entry.organization,
                                 content_type = entry.content_type,
-                                model = entry.model,
+                                model = entry_model,
                                 before = entry.before,
                                 after = entry.after,
                                 action = entry.action,
@@ -124,7 +134,7 @@ def centurion_model_migrate(sender, **kwargs):
                             print(f'        Removed {history_model_name}={id} from database.')
 
                         except Exception as e:
-                            print(f"Exception {e.__class__.__name__} occured:\n\s\s\s\s{e}")
+                            print(f"Exception {e.__class__.__name__} occured:"+"\n    "+f'{e}')
 
 
                 except LookupError as e: