photoprism/internal/entity/country.go

72 lines
1.5 KiB
Go
Raw Normal View History

package entity
2018-09-19 09:20:57 +02:00
import (
"github.com/gosimple/slug"
2018-09-19 09:20:57 +02:00
"github.com/jinzhu/gorm"
"github.com/photoprism/photoprism/internal/maps"
2018-09-19 09:20:57 +02:00
)
var altCountryNames = map[string]string{
"United States of America": "USA",
"United States": "USA",
"": "Unknown",
}
2018-09-19 09:20:57 +02:00
type Country struct {
ID string `gorm:"primary_key"`
CountrySlug string
2018-09-19 09:20:57 +02:00
CountryName string
CountryDescription string `gorm:"type:text;"`
CountryNotes string `gorm:"type:text;"`
CountryPhoto *Photo
CountryPhotoID uint
New bool `gorm:"-"`
2018-09-19 09:20:57 +02:00
}
var UnknownCountry = NewCountry("zz", maps.CountryNames["zz"])
func CreateUnknownCountry(db *gorm.DB) {
UnknownCountry.FirstOrCreate(db)
}
2018-11-06 19:02:03 +01:00
// Create a new country
func NewCountry(countryCode string, countryName string) *Country {
if countryCode == "" {
countryCode = "zz"
}
if altName, ok := altCountryNames[countryName]; ok {
countryName = altName
}
countrySlug := slug.MakeLang(countryName, "en")
result := &Country{
ID: countryCode,
CountryName: countryName,
CountrySlug: countrySlug,
}
return result
}
func (m *Country) FirstOrCreate(db *gorm.DB) *Country {
if err := db.FirstOrCreate(m, "id = ?", m.ID).Error; err != nil {
log.Errorf("country: %s", err)
}
return m
}
func (m *Country) AfterCreate(scope *gorm.Scope) error {
return scope.SetColumn("New", true)
}
func (m *Country) Code() string {
return m.ID
}
func (m *Country) Name() string {
return m.CountryName
}