add button to create new class without images closes #14

logic/models/data.go

@@ -459,6 +459,45 @@ func handleDataUpload(handle *Handle) {
 		return c.SendJSON(model.Id)
 	})
 
+	// Create New class
+	type CreateNewEmptyClass struct {
+		Id   string `json:"id" validate:"required"`
+		Name string `json:"name" validate:"required"`
+	}
+	PostAuthJson(handle, "/models/data/class/new", User_Normal, func(c *Context, obj *CreateNewEmptyClass) *Error {
+		model, err := GetBaseModel(c.Db, obj.Id)
+		if err == ModelNotFoundError {
+			return c.JsonBadRequest("Model not found")
+		}
+
+		if model.ModelType != 2 && model.Status != CONFIRM_PRE_TRAINING || model.ModelType == 2 && model.Status != CONFIRM_PRE_TRAINING && model.Status != READY {
+			return c.JsonBadRequest("Model not in the correct status for adding new class")
+		}
+
+		var baseOrder struct {
+			Order int `db:"class_order"`
+		}
+		err = GetDBOnce(c, &baseOrder, "model_classes where model_id=$1 order by class_order desc;", model.Id)
+		if err != nil {
+			return c.E500M("Could not create class", err)
+		}
+
+		id, err := model_classes.CreateClass(c.Db, model.Id, baseOrder.Order+1, obj.Name)
+		if err == model_classes.ClassAlreadyExists {
+			return c.JsonBadRequest("Class Already exists")
+		} else if err != nil {
+			return c.E500M("Could not create class", err)
+		}
+
+		var modelClass model_classes.ModelClass
+		err = GetDBOnce(c, &modelClass, "model_classes where id=$1;", id)
+		if err != nil {
+			return c.E500M("Failed to get class information but class was creted", err)
+		}
+
+		return c.SendJSON(modelClass)
+	})
+
 	// ------
 	// ------ CLASS DATA UPLOAD
 	// ------