Fixes running in netbox-docker with errors on version strings

netbox_diode_plugin/api/serializers.py

@@ -19,10 +19,16 @@
 
 from netbox_diode_plugin.models import Setting
 
-if version.parse(version.parse(settings.VERSION).base_version) >= version.parse("4.1"):
+try:
+    if version.parse(version.parse(settings.VERSION).base_version) >= version.parse(
+        "4.1"
+    ):
+        from core.models import ObjectChange
+    else:
+        from extras.models import ObjectChange
+except version.InvalidVersion:
     from core.models import ObjectChange
-else:
-    from extras.models import ObjectChange
+
 from ipam.api.serializers import IPAddressSerializer, PrefixSerializer
 from rest_framework import serializers
 from utilities.api import get_serializer_for_model

netbox_diode_plugin/api/views.py

@@ -7,10 +7,13 @@
 from django.conf import settings
 from packaging import version
 
-if version.parse(settings.VERSION).major >= 4:
+try:
+    if version.parse(settings.VERSION).major >= 4:
+        from core.models import ObjectType as NetBoxType
+    else:
+        from django.contrib.contenttypes.models import ContentType as NetBoxType
+except version.InvalidVersion:
     from core.models import ObjectType as NetBoxType
-else:
-    from django.contrib.contenttypes.models import ContentType as NetBoxType
 from django.core.exceptions import FieldError
 from django.db import transaction
 from django.db.models import Q

netbox_diode_plugin/tables.py

@@ -8,10 +8,13 @@
 from django.conf import settings
 from packaging import version
 
-if version.parse(settings.VERSION).major >= 4:
+try:
+    if version.parse(settings.VERSION).major >= 4:
+        from core.models import ObjectType as NetBoxType
+    else:
+        from django.contrib.contenttypes.models import ContentType as NetBoxType
+except version.InvalidVersion:
     from core.models import ObjectType as NetBoxType
-else:
-    from django.contrib.contenttypes.models import ContentType as NetBoxType
 
 from netbox.tables import BaseTable, columns
 from utilities.object_types import object_type_identifier, object_type_name