From 063b5f154af487e741dc56e9a5bc6339c7add565 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Sun, 24 Nov 2024 05:00:45 +0000
Subject: [PATCH] Include custom state values in specific API endpoints

---
 src/backend/InvenTree/generic/states/api.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/backend/InvenTree/generic/states/api.py b/src/backend/InvenTree/generic/states/api.py
index 5fd70bde93..c52be2ffb3 100644
--- a/src/backend/InvenTree/generic/states/api.py
+++ b/src/backend/InvenTree/generic/states/api.py
@@ -72,6 +72,21 @@ class StatusView(GenericAPIView):
 
         data = {'class': status_class.__name__, 'values': status_class.dict()}
 
+        # Extend with custom values
+        try:
+            custom_values = status_class.custom_values()
+            for item in custom_values:
+                if item.name not in data['values']:
+                    data['values'][item.name] = {
+                        'color': item.color,
+                        'logical_key': item.logical_key,
+                        'key': item.key,
+                        'label': item.label,
+                        'name': item.name,
+                    }
+        except Exception:
+            pass
+
         return Response(data)