package entity import ( "github.com/photoprism/photoprism/internal/classify" ) // 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 PhotoLabel table identifier "photos_labels" 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 err := m.Create(); err != nil { // Already exists. return nil } return m } // 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{ Name: m.Label.LabelName, Source: m.LabelSrc, Uncertainty: m.Uncertainty, Priority: m.Label.LabelPriority, } return result }