photoprism/internal/entity/photo_label.go

98 lines
2.7 KiB
Go
Raw Normal View History

package entity
import (
"github.com/photoprism/photoprism/internal/classify"
)
type PhotoLabels []PhotoLabel
// PhotoLabel represents the many-to-many relation between Photo and label.
// Labels are weighted by uncertainty (100 - confidence)
type PhotoLabel struct {
PhotoID uint `gorm:"primary_key;auto_increment:false"`
LabelID uint `gorm:"primary_key;auto_increment:false;index"`
LabelSrc string `gorm:"type:VARBINARY(8);"`
Uncertainty int `gorm:"type:SMALLINT"`
Photo *Photo `gorm:"PRELOAD:false"`
Label *Label `gorm:"PRELOAD:true"`
}
// TableName returns the entity database table name.
func (PhotoLabel) TableName() string {
return "photos_labels"
}
// NewPhotoLabel registers a new PhotoLabel relation with an uncertainty and a source of label
func NewPhotoLabel(photoID, labelID uint, uncertainty int, source string) *PhotoLabel {
result := &PhotoLabel{
PhotoID: photoID,
LabelID: labelID,
Uncertainty: uncertainty,
LabelSrc: source,
}
return result
}
// Updates multiple columns in the database.
func (m *PhotoLabel) Updates(values interface{}) error {
return UnscopedDb().Model(m).UpdateColumns(values).Error
}
// Updates a column in the database.
func (m *PhotoLabel) Update(attr string, value interface{}) error {
return UnscopedDb().Model(m).UpdateColumn(attr, value).Error
}
// Save saves the entity in the database.
func (m *PhotoLabel) Save() error {
if m.Photo != nil {
m.Photo = nil
}
if m.Label != nil {
m.Label.SetName(m.Label.LabelName)
}
return Db().Save(m).Error
}
// Create inserts a new row to the database.
func (m *PhotoLabel) Create() error {
return Db().Create(m).Error
}
// FirstOrCreatePhotoLabel returns the existing row, inserts a new row or nil in case of errors.
func FirstOrCreatePhotoLabel(m *PhotoLabel) *PhotoLabel {
result := PhotoLabel{}
if err := Db().Where("photo_id = ? AND label_id = ?", m.PhotoID, m.LabelID).First(&result).Error; err == nil {
return &result
} else if createErr := m.Create(); createErr == nil {
return m
} else if err := Db().Where("photo_id = ? AND label_id = ?", m.PhotoID, m.LabelID).First(&result).Error; err == nil {
return &result
} else {
2020-12-14 13:31:18 +01:00
log.Errorf("photo-label: %s (find or create)", createErr)
}
return nil
}
// ClassifyLabel returns the label as classify.Label
func (m *PhotoLabel) ClassifyLabel() classify.Label {
if m.Label == nil {
log.Errorf("photo-label: classify label is nil (photo id %d, label id %d) - bug?", m.PhotoID, m.LabelID)
return classify.Label{}
}
result := classify.Label{
2020-04-16 23:16:55 +02:00
Name: m.Label.LabelName,
Source: m.LabelSrc,
Uncertainty: m.Uncertainty,
2020-04-16 23:16:55 +02:00
Priority: m.Label.LabelPriority,
}
return result
}