photoprism/internal/models/country.go

44 lines
832 B
Go
Raw Normal View History

2018-09-19 09:20:57 +02:00
package models
import (
"github.com/gosimple/slug"
2018-09-19 09:20:57 +02:00
"github.com/jinzhu/gorm"
)
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
}
2018-11-06 19:02:03 +01:00
// 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 (c *Country) FirstOrCreate(db *gorm.DB) *Country {
db.FirstOrCreate(c, "id = ?", c.ID)
return c
}