API forms return the JSON data to the onSuccess function

lib/widget/category_display.dart

@@ -67,7 +67,13 @@ class _CategoryDisplayState extends RefreshableState<CategoryDisplayWidget> {
       return;
     }
 
-    _cat.editForm(context, L10().editCategory, onSuccess: refresh);
+    _cat.editForm(
+        context,
+        L10().editCategory,
+        onSuccess: (data) async {
+          refresh();
+        }
+    );
   }
 
   _CategoryDisplayState(this.category);
@@ -211,14 +217,27 @@ class _CategoryDisplayState extends RefreshableState<CategoryDisplayWidget> {
 
     int pk = category?.pk ?? -1;
 
-    launchApiForm(
-        context,
-        L10().categoryCreate,
-        InvenTreePartCategory().URL,
-        InvenTreePartCategory().formFields(),
-        modelData: {
-          "parent": (pk > 0) ? pk : null,
+    InvenTreePartCategory().createForm(
+      context,
+      L10().categoryCreate,
+      data: {
+        "parent": (pk > 0) ? pk : null,
+      },
+      onSuccess: (data) async {
+        
+        if (data.containsKey("pk")) {
+          var new_cat = InvenTreePartCategory.fromJson(data);
+
+          Navigator.push(
+            context,
+            MaterialPageRoute(
+              builder: (context) => CategoryDisplayWidget(new_cat)
+            )
+          );
+        } else {
+          refresh();
         }
+      }
     );
   }
 

lib/widget/company_detail.dart

@@ -79,7 +79,9 @@ class _CompanyDetailState extends RefreshableState<CompanyDetailWidget> {
         "currency": {},
       },
       modelData: company.jsondata,
-      onSuccess: refresh
+      onSuccess: (data) async {
+        refresh();
+      }
     );
   }
 

lib/widget/location_display.dart

@@ -94,7 +94,9 @@ class _LocationDisplayState extends RefreshableState<LocationDisplayWidget> {
         "parent": {},
       },
       modelData: _loc.jsondata,
-      onSuccess: refresh
+      onSuccess: (data) async {
+        refresh();
+      }
     );
   }
 

lib/widget/part_detail.dart

@@ -126,7 +126,9 @@ class _PartDisplayState extends RefreshableState<PartDetailWidget> {
           "virtual": {},
         },
         modelData: part.jsondata,
-        onSuccess: refresh,
+        onSuccess: (data) async {
+          refresh();
+        },
     );
   }
 

lib/widget/part_notes.dart

@@ -56,7 +56,7 @@ class _PartNotesState extends RefreshableState<PartNotesWidget> {
                 }
               },
               modelData: part.jsondata,
-              onSuccess: () async {
+              onSuccess: (data) async {
                 refresh();
               }
             );

lib/widget/stock_detail.dart

@@ -120,7 +120,9 @@ class _StockItemDisplayState extends RefreshableState<StockDetailWidget> {
         "link": {},
       },
       modelData: item.jsondata,
-      onSuccess: refresh
+      onSuccess: (data) async {
+        refresh();
+      }
     );
 
   }

lib/widget/stock_notes.dart

@@ -56,7 +56,7 @@ class _StockNotesState extends RefreshableState<StockNotesWidget> {
                       }
                     },
                     modelData: item.jsondata,
-                    onSuccess: () {
+                    onSuccess: (data) async {
                       refresh();
                     }
                 );