From cf9d71f86cdab4327a76fac0d0bbeacd9c8f27b7 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Sat, 28 Dec 2024 01:55:54 +0000
Subject: [PATCH] Add 'custom' attribute

---
 src/backend/InvenTree/generic/states/api.py         | 3 +++
 src/backend/InvenTree/generic/states/serializers.py | 4 +++-
 src/backend/InvenTree/generic/states/states.py      | 1 +
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/backend/InvenTree/generic/states/api.py b/src/backend/InvenTree/generic/states/api.py
index 8c600e515f..092f5bbb41 100644
--- a/src/backend/InvenTree/generic/states/api.py
+++ b/src/backend/InvenTree/generic/states/api.py
@@ -83,6 +83,7 @@ class StatusView(GenericAPIView):
                         'key': item.key,
                         'label': item.label,
                         'name': item.name,
+                        'custom': True,
                     }
         except Exception:
             pass
@@ -114,12 +115,14 @@ class AllStatusViews(StatusView):
             for item in cls.custom_values():
                 label = str(item.name)
                 if label not in cls_data['values']:
+                    print('custom value:', item)
                     cls_data['values'][label] = {
                         'color': item.color,
                         'logical_key': item.logical_key,
                         'key': item.key,
                         'label': item.label,
                         'name': item.name,
+                        'custom': True,
                     }
 
             data[cls.__name__] = GenericStateClassSerializer(cls_data, many=False).data
diff --git a/src/backend/InvenTree/generic/states/serializers.py b/src/backend/InvenTree/generic/states/serializers.py
index dd62e2c0ba..52464dfaba 100644
--- a/src/backend/InvenTree/generic/states/serializers.py
+++ b/src/backend/InvenTree/generic/states/serializers.py
@@ -11,7 +11,7 @@ class GenericStateValueSerializer(serializers.Serializer):
     class Meta:
         """Meta class for GenericStateValueSerializer."""
 
-        fields = ['key', 'logical_key', 'name', 'label', 'color']
+        fields = ['key', 'logical_key', 'name', 'label', 'color', 'custom']
 
     key = serializers.IntegerField(label=_('Key'), required=True)
 
@@ -23,6 +23,8 @@ class GenericStateValueSerializer(serializers.Serializer):
 
     color = serializers.CharField(label=_('Color'), required=False)
 
+    custom = serializers.BooleanField(label=_('Custom'), required=False)
+
 
 class GenericStateClassSerializer(serializers.Serializer):
     """API serializer for generic state class information."""
diff --git a/src/backend/InvenTree/generic/states/states.py b/src/backend/InvenTree/generic/states/states.py
index af44ef95a9..3b92ea56c7 100644
--- a/src/backend/InvenTree/generic/states/states.py
+++ b/src/backend/InvenTree/generic/states/states.py
@@ -214,6 +214,7 @@ class StatusCode(BaseEnum):
                             'key': item.key,
                             'label': item.label,
                             'name': item.name,
+                            'custom': True,
                         }
             except Exception:
                 pass