96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package models
|
|
|
|
import (
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/photoprism/photoprism/internal/util"
|
|
uuid "github.com/satori/go.uuid"
|
|
)
|
|
|
|
// A photo can have multiple images and sidecar files
|
|
type Photo struct {
|
|
Model
|
|
PhotoUUID string `gorm:"unique_index;"`
|
|
PhotoToken string `gorm:"type:varchar(64);"`
|
|
PhotoPath string `gorm:"type:varchar(128);index;"`
|
|
PhotoName string
|
|
PhotoTitle string
|
|
PhotoTitleChanged bool
|
|
PhotoDescription string `gorm:"type:text;"`
|
|
PhotoNotes string `gorm:"type:text;"`
|
|
PhotoArtist string
|
|
PhotoFavorite bool
|
|
PhotoPrivate bool
|
|
PhotoNSFW bool
|
|
PhotoStory bool
|
|
PhotoLat float64 `gorm:"index;"`
|
|
PhotoLong float64 `gorm:"index;"`
|
|
PhotoAltitude int
|
|
PhotoFocalLength int
|
|
PhotoIso int
|
|
PhotoFNumber float64
|
|
PhotoExposure string
|
|
PhotoViews uint
|
|
Camera *Camera
|
|
CameraID uint `gorm:"index;"`
|
|
Lens *Lens
|
|
LensID uint `gorm:"index;"`
|
|
Country *Country
|
|
CountryID string `gorm:"index;"`
|
|
CountryChanged bool
|
|
Location *Location
|
|
LocationID uint
|
|
LocationChanged bool
|
|
LocationEstimated bool
|
|
TakenAt time.Time `gorm:"index;"`
|
|
TakenAtLocal time.Time
|
|
TakenAtChanged bool
|
|
TimeZone string
|
|
Labels []*PhotoLabel
|
|
Albums []*PhotoAlbum
|
|
Files []*File
|
|
}
|
|
|
|
func (m *Photo) BeforeCreate(scope *gorm.Scope) error {
|
|
if err := scope.SetColumn("PhotoUUID", uuid.NewV4().String()); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := scope.SetColumn("PhotoToken", util.RandomToken(4)); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Photo) IndexKeywords(keywords []string, db *gorm.DB) {
|
|
var keywordIds []uint
|
|
|
|
// Index title and description
|
|
keywords = append(keywords, util.Keywords(m.PhotoTitle)...)
|
|
keywords = append(keywords, util.Keywords(m.PhotoDescription)...)
|
|
last := ""
|
|
|
|
sort.Strings(keywords)
|
|
|
|
for _, w := range keywords {
|
|
if len(w) < 3 || w == last {
|
|
continue
|
|
}
|
|
|
|
last = w
|
|
kw := NewKeyword(w).FirstOrCreate(db)
|
|
|
|
if kw.Skip {
|
|
continue
|
|
}
|
|
|
|
keywordIds = append(keywordIds, kw.ID)
|
|
|
|
NewPhotoKeyword(m.ID, kw.ID).FirstOrCreate(db)
|
|
}
|
|
|
|
db.Where("photo_id = ? AND keyword_id NOT IN (?)", m.ID, keywordIds).Delete(&PhotoKeyword{})
|
|
}
|