From 354c5dc1c5757bf8d532f447ca7f2770e75bafb4 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@netboxlabs.com>
Date: Thu, 29 May 2025 09:37:07 -0400
Subject: [PATCH] Add migrators for MAC address assignments

---
 .../migrations/0200_populate_mac_addresses.py | 40 +++++++++++++++++++
 .../migrations/0048_populate_mac_addresses.py | 40 +++++++++++++++++++
 2 files changed, 80 insertions(+)

diff --git a/netbox/dcim/migrations/0200_populate_mac_addresses.py b/netbox/dcim/migrations/0200_populate_mac_addresses.py
index 6ffb71731..48d2c746b 100644
--- a/netbox/dcim/migrations/0200_populate_mac_addresses.py
+++ b/netbox/dcim/migrations/0200_populate_mac_addresses.py
@@ -1,4 +1,6 @@
 import django.db.models.deletion
+from django.apps import apps
+from django.contrib.contenttypes.models import ContentType
 from django.db import migrations, models
 
 
@@ -51,3 +53,41 @@ class Migration(migrations.Migration):
             name='mac_address',
         ),
     ]
+
+
+def oc_interface_primary_mac_address(objectchange, revert):
+    MACAddress = apps.get_model('dcim', 'MACAddress')
+    interface_ct = ContentType.objects.get_by_natural_key('dcim', 'interface')
+
+    if not revert:
+        before, after = objectchange.prechange_data, objectchange.postchange_data
+    else:
+        before, after = objectchange.postchange_data, objectchange.prechange_data
+
+    if after.get('mac_address') != before.get('mac_address'):
+        # Create & assign the new MACAddress (if any)
+        if after.get('mac_address'):
+            mac = MACAddress.objects.create(
+                mac_address=after['mac_address'],
+                assigned_object_type=interface_ct,
+                assigned_object_id=objectchange.changed_object_id,
+            )
+            after['primary_mac_address'] = mac.pk
+        else:
+            after['primary_mac_address'] = None
+        # Delete the old MACAddress (if any)
+        if before.get('mac_address'):
+            MACAddress.objects.filter(
+                mac_address=before['mac_address'],
+                assigned_object_type=interface_ct,
+                assigned_object_id=objectchange.changed_object_id,
+            ).delete()
+        before['primary_mac_address'] = None
+
+    before.pop('mac_address')
+    after.pop('mac_address')
+
+
+objectchange_migrators = {
+    'dcim.interface': oc_interface_primary_mac_address,
+}
diff --git a/netbox/virtualization/migrations/0048_populate_mac_addresses.py b/netbox/virtualization/migrations/0048_populate_mac_addresses.py
index fe485d52e..6de748fe8 100644
--- a/netbox/virtualization/migrations/0048_populate_mac_addresses.py
+++ b/netbox/virtualization/migrations/0048_populate_mac_addresses.py
@@ -1,4 +1,6 @@
 import django.db.models.deletion
+from django.apps import apps
+from django.contrib.contenttypes.models import ContentType
 from django.db import migrations, models
 
 
@@ -50,3 +52,41 @@ class Migration(migrations.Migration):
             name='mac_address',
         ),
     ]
+
+
+def oc_vminterface_primary_mac_address(objectchange, revert):
+    MACAddress = apps.get_model('dcim', 'MACAddress')
+    vminterface_ct = ContentType.objects.get_by_natural_key('virtualization', 'vminterface')
+
+    if not revert:
+        before, after = objectchange.prechange_data, objectchange.postchange_data
+    else:
+        before, after = objectchange.postchange_data, objectchange.prechange_data
+
+    if after.get('mac_address') != before.get('mac_address'):
+        # Create & assign the new MACAddress (if any)
+        if after.get('mac_address'):
+            mac = MACAddress.objects.create(
+                mac_address=after['mac_address'],
+                assigned_object_type=vminterface_ct,
+                assigned_object_id=objectchange.changed_object_id,
+            )
+            after['primary_mac_address'] = mac.pk
+        else:
+            after['primary_mac_address'] = None
+        # Delete the old MACAddress (if any)
+        if before.get('mac_address'):
+            MACAddress.objects.filter(
+                mac_address=before['mac_address'],
+                assigned_object_type=vminterface_ct,
+                assigned_object_id=objectchange.changed_object_id,
+            ).delete()
+        before['primary_mac_address'] = None
+
+    before.pop('mac_address')
+    after.pop('mac_address')
+
+
+objectchange_migrators = {
+    'virtualization.vminterface': oc_vminterface_primary_mac_address,
+}