package models import ( "net/http" "os" "path" . "git.andr3h3nriqu3s.com/andr3/fyp/logic/db_types" . "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils" ) func DeleteModel(c BasePack, id string) (err error) { c.GetLogger().Warnf("Removing model with id: %s", id) _, err = c.GetDb().Exec("delete from models where id=$1;", id) if err != nil { return } model_path := path.Join("./savedData", id) c.GetLogger().Warnf("Removing folder of model with id: %s at %s", id, model_path) err = os.RemoveAll(model_path) if err != nil { return } return } func deleteModelJSON(c *Context, id string) *Error { err := DeleteModel(c, id) if err != nil { return c.E500M("Failed to delete models", err) } return c.SendJSON(id) } func handleDelete(handle *Handle) { type DeleteModel struct { Id string `json:"id" validate:"required"` Name *string `json:"name,omitempty"` } DeleteAuthJson(handle, "/models/delete", User_Normal, func(c *Context, dat *DeleteModel) *Error { var model struct { Id string Name string Status int } err := GetDBOnce(c, &model, "models where id=$1 and user_id=$2;", dat.Id, c.User.Id) if err == NotFoundError { return c.SendJSONStatus(http.StatusNotFound, "Model not found!") } else if err != nil { return c.E500M("Faield to get model", err) } switch model.Status { case FAILED_TRAINING: fallthrough case FAILED_PREPARING_ZIP_FILE: fallthrough case FAILED_PREPARING_TRAINING: fallthrough case FAILED_PREPARING: return deleteModelJSON(c, dat.Id) case READY: fallthrough case READY_RETRAIN_FAILED: fallthrough case READY_ALTERATION_FAILED: fallthrough case CONFIRM_PRE_TRAINING: if dat.Name == nil { return c.JsonBadRequest("Provided name does not match the model name") } if *dat.Name != model.Name { return c.JsonBadRequest("Provided name does not match the model name") } return deleteModelJSON(c, dat.Id) default: c.Logger.Warn("Do not know how to handle model in status", "status", model.Status) return c.JsonBadRequest("Model in invalid status") } }) }