photoprism/internal/entity/country.go
Michael Mayer 4e06deda76 Backend: Rename "models" package to "entity"
Signed-off-by: Michael Mayer <michael@liquidbytes.net>
2019-12-11 16:55:18 +01:00

48 lines
965 B
Go

package entity
import (
"github.com/gosimple/slug"
"github.com/jinzhu/gorm"
)
type Country struct {
ID string `gorm:"primary_key"`
CountrySlug string
CountryName string
CountryDescription string `gorm:"type:text;"`
CountryNotes string `gorm:"type:text;"`
CountryPhoto *Photo
CountryPhotoID uint
New bool `gorm:"-"`
}
// Create a new country
func NewCountry(countryCode string, countryName string) *Country {
if countryCode == "" {
countryCode = "zz"
}
if countryName == "" {
countryName = "Unknown"
}
countrySlug := slug.MakeLang(countryName, "en")
result := &Country{
ID: countryCode,
CountryName: countryName,
CountrySlug: countrySlug,
}
return result
}
func (m *Country) FirstOrCreate(db *gorm.DB) *Country {
db.FirstOrCreate(m, "id = ?", m.ID)
return m
}
func (m *Country) AfterCreate(scope *gorm.Scope) error {
return scope.SetColumn("New", true)
}