55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package models
|
|
|
|
import (
|
|
"git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
|
)
|
|
|
|
// Auth level set when path is definied as 1
|
|
func handleStats(c *Context) *Error {
|
|
var b struct {
|
|
Id string `json:"id" validate:"required"`
|
|
}
|
|
|
|
if _err := c.ToJSON(&b); _err != nil {
|
|
return _err;
|
|
}
|
|
|
|
type Row struct {
|
|
Name string `db:"mc.name" json:"name"`
|
|
Training string `db:"count(mdp.id) filter (where mdp.model_mode=1)" json:"training"`
|
|
Testing string `db:"count(mdp.id) filter (where mdp.model_mode=2)" json:"testing"`
|
|
}
|
|
|
|
rows, err := GetDbMultitple[Row](c, "model_data_point as mdp inner join model_classes as mc on mc.id=mdp.class_id where mc.model_id=$1 group by mc.name order by mc.name asc;", b.Id)
|
|
if err != nil {
|
|
return c.Error500(err)
|
|
}
|
|
|
|
c.ShowMessage = false
|
|
return c.SendJSON(rows)
|
|
}
|
|
|
|
func handleList(handle *Handle) {
|
|
handle.Get("/models", func(c *Context) *Error {
|
|
if !c.CheckAuthLevel(1) {
|
|
return nil
|
|
}
|
|
|
|
type Row struct {
|
|
Name string `json:"name"`
|
|
Id string `json:"id"`
|
|
}
|
|
|
|
got, err := utils.GetDbMultitple[Row](c, "models where user_id=$1", c.User.Id)
|
|
if err != nil {
|
|
return c.Error500(nil)
|
|
}
|
|
|
|
c.ShowMessage = true
|
|
return c.SendJSON(got)
|
|
})
|
|
|
|
handle.PostAuth("/models/class/stats", 1, handleStats)
|
|
}
|