made StatusRenderer less strict to allow usage of custom states (#6008)

inventree · Nov 30, 2023 · b343ef3 · b343ef3
1 parent 0cd66fd
commit b343ef3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/frontend/src/components/render/StatusRenderer.tsx b/src/frontend/src/components/render/StatusRenderer.tsx
@@ -69,7 +69,7 @@ export const StatusRenderer = ({
   options
 }: {
   status: string;
-  type: ModelType;
+  type: ModelType | string;
   options?: renderStatusLabelOptionsInterface;
 }) => {
   const statusCodeList = useServerApiState.getState().status;

diff --git a/src/frontend/src/states/ApiState.tsx b/src/frontend/src/states/ApiState.tsx
@@ -9,7 +9,7 @@ import { ApiPaths } from '../enums/ApiEndpoints';
 import { ModelType } from '../enums/ModelType';
 import { ServerAPIProps } from './states';
 
-type StatusLookup = Record<ModelType, StatusCodeListInterface>;
+type StatusLookup = Record<ModelType | string, StatusCodeListInterface>;
 
 interface ServerApiStateProps {
   server: ServerAPIProps;
@@ -35,7 +35,8 @@ export const useServerApiState = create<ServerApiStateProps>()(
         await api.get(apiUrl(ApiPaths.global_status)).then((response) => {
           const newStatusLookup: StatusLookup = {} as StatusLookup;
           for (const key in response.data) {
-            newStatusLookup[statusCodeList[key]] = response.data[key].values;
+            newStatusLookup[statusCodeList[key] || key] =
+              response.data[key].values;
           }
           set({ status: newStatusLookup });
         });