photoprism/internal/query/subjects.go

120 lines
3.2 KiB
Go
Raw Normal View History

package query
import (
2021-08-19 23:12:51 +02:00
"fmt"
"github.com/photoprism/photoprism/pkg/txt"
"github.com/photoprism/photoprism/internal/entity"
)
// People returns the sorted names of the first 2000 people.
func People() (people entity.People, err error) {
err = UnscopedDb().
Table(entity.Subject{}.TableName()).
Select("subj_uid, subj_name, subj_alias, subj_favorite").
Where("deleted_at IS NULL AND subj_type = ?", entity.SubjPerson).
Order("subj_favorite DESC, subj_name").
Limit(2000).Offset(0).
Scan(&people).Error
return people, err
}
// PeopleCount returns the total number of people in the index.
func PeopleCount() (count int, err error) {
err = Db().
Table(entity.Subject{}.TableName()).
Where("deleted_at IS NULL").
Where("subj_type = ?", entity.SubjPerson).
Count(&count).Error
return count, err
}
// Subjects returns subjects from the index.
func Subjects(limit, offset int) (result entity.Subjects, err error) {
stmt := Db()
stmt = stmt.Order("subj_name").Limit(limit).Offset(offset)
err = stmt.Find(&result).Error
return result, err
}
2021-08-19 23:12:51 +02:00
2021-08-28 15:26:26 +02:00
// SubjectMap returns a map of subjects indexed by UID.
func SubjectMap() (result map[string]entity.Subject, err error) {
result = make(map[string]entity.Subject)
var subj entity.Subjects
stmt := Db()
2021-08-28 15:26:26 +02:00
if err = stmt.Find(&subj).Error; err != nil {
return result, err
}
for _, s := range subj {
result[s.SubjUID] = s
2021-08-28 15:26:26 +02:00
}
return result, err
}
// RemoveOrphanSubjects permanently deletes dangling marker subjects from the index.
func RemoveOrphanSubjects() (removed int64, err error) {
res := UnscopedDb().
Where("subj_src = ?", entity.SrcMarker).
Where(fmt.Sprintf("subj_uid NOT IN (SELECT subj_uid FROM %s)", entity.Face{}.TableName())).
Where(fmt.Sprintf("subj_uid NOT IN (SELECT subj_uid FROM %s)", entity.Marker{}.TableName())).
Delete(&entity.Subject{})
return res.RowsAffected, res.Error
2021-08-19 23:12:51 +02:00
}
// CreateMarkerSubjects adds and references known marker subjects.
func CreateMarkerSubjects() (affected int64, err error) {
var markers entity.Markers
if err := Db().
Where("subj_uid = '' AND marker_name <> '' AND subj_src <> ?", entity.SrcAuto).
Where("marker_invalid = 0 AND marker_type = ?", entity.MarkerFace).
Order("marker_name").
Find(&markers).Error; err != nil {
return affected, err
} else if len(markers) == 0 {
return affected, nil
}
var name string
var subj *entity.Subject
for _, m := range markers {
if name == m.MarkerName && subj != nil {
// Do nothing.
} else if subj = entity.NewSubject(m.MarkerName, entity.SubjPerson, entity.SrcMarker); subj == nil {
log.Errorf("faces: subject should not be nil - bug?")
continue
} else if subj = entity.FirstOrCreateSubject(subj); subj == nil {
log.Errorf("faces: failed adding subject %s", txt.Quote(m.MarkerName))
continue
} else {
affected++
}
name = m.MarkerName
if err := m.Updates(entity.Values{"SubjUID": subj.SubjUID, "MarkerReview": false}); err != nil {
return affected, err
}
if m.FaceID == "" {
continue
} else if err := Db().Model(&entity.Face{}).Where("id = ? AND subj_uid = ''", m.FaceID).Update("SubjUID", subj.SubjUID).Error; err != nil {
return affected, err
}
}
return affected, err
}