photoprism/internal/models/label.go
2019-06-05 14:05:21 +02:00

43 lines
944 B
Go

package models
import (
"strings"
"github.com/gosimple/slug"
"github.com/jinzhu/gorm"
)
// Labels for photo, album and location categorization
type Label struct {
Model
LabelSlug string `gorm:"type:varchar(128);index;"`
LabelName string `gorm:"type:varchar(128);index;"`
LabelPriority int
LabelDescription string `gorm:"type:text;"`
LabelNotes string `gorm:"type:text;"`
LabelCategories []*Label `gorm:"many2many:categories;association_jointable_foreignkey:category_id"`
}
func NewLabel(labelName string, labelPriority int) *Label {
labelName = strings.TrimSpace(labelName)
if labelName == "" {
labelName = "Unknown"
}
labelSlug := slug.Make(labelName)
result := &Label{
LabelName: labelName,
LabelSlug: labelSlug,
LabelPriority: labelPriority,
}
return result
}
func (m *Label) FirstOrCreate(db *gorm.DB) *Label {
db.FirstOrCreate(m, "label_slug = ?", m.LabelSlug)
return m
}