photoprism/internal/entity/camera.go

74 lines
1.6 KiB
Go
Raw Normal View History

package entity
import (
"fmt"
"strings"
"time"
2018-09-13 20:54:34 +02:00
"github.com/gosimple/slug"
"github.com/jinzhu/gorm"
"github.com/photoprism/photoprism/internal/mutex"
)
// Camera model and make (as extracted from UpdateExif metadata)
type Camera struct {
ID uint `gorm:"primary_key"`
CameraSlug string `gorm:"type:varbinary(128);unique_index;"`
2018-09-19 09:20:57 +02:00
CameraModel string
CameraMake string
2018-09-19 09:20:57 +02:00
CameraType string
CameraOwner string
CameraDescription string `gorm:"type:text;"`
CameraNotes string `gorm:"type:text;"`
CreatedAt time.Time
UpdatedAt time.Time
DeletedAt *time.Time `sql:"index"`
}
func NewCamera(modelName string, makeName string) *Camera {
makeName = strings.TrimSpace(makeName)
if modelName == "" {
modelName = "Unknown"
} else if strings.HasPrefix(modelName, makeName) {
modelName = strings.TrimSpace(modelName[len(makeName):])
}
var cameraSlug string
if makeName != "" {
cameraSlug = slug.Make(makeName + " " + modelName)
} else {
cameraSlug = slug.Make(modelName)
}
2018-09-13 20:54:34 +02:00
result := &Camera{
CameraModel: modelName,
CameraMake: makeName,
2018-09-17 18:40:57 +02:00
CameraSlug: cameraSlug,
}
return result
}
func (m *Camera) FirstOrCreate(db *gorm.DB) *Camera {
mutex.Db.Lock()
defer mutex.Db.Unlock()
if err := db.FirstOrCreate(m, "camera_model = ? AND camera_make = ?", m.CameraModel, m.CameraMake).Error; err != nil {
log.Errorf("camera: %s", err)
}
return m
}
func (m *Camera) String() string {
if m.CameraMake != "" && m.CameraModel != "" {
return fmt.Sprintf("%s %s", m.CameraMake, m.CameraModel)
} else if m.CameraModel != "" {
return m.CameraModel
}
return ""
}