photoprism/internal/entity/photo_yaml.go
Michael Mayer 3fec93ee4e Load existing photo if uid was found in YAML sidecar file #357
Signed-off-by: Michael Mayer <michael@liquidbytes.net>
2020-07-07 12:20:38 +02:00

62 lines
1.2 KiB
Go

package entity
import (
"io/ioutil"
"os"
"path/filepath"
"github.com/photoprism/photoprism/pkg/fs"
"gopkg.in/yaml.v2"
)
// Yaml returns photo data as YAML string.
func (m *Photo) Yaml() ([]byte, error) {
out, err := yaml.Marshal(m)
if err != nil {
return []byte{}, err
}
return out, err
}
// SaveAsYaml saves photo data as YAML file.
func (m *Photo) SaveAsYaml(fileName string) error {
data, err := m.Yaml()
if err != nil {
return err
}
// Make sure directory exists.
if err := os.MkdirAll(filepath.Dir(fileName), os.ModePerm); err != nil {
return err
}
// Write YAML data to file.
if err := ioutil.WriteFile(fileName, data, os.ModePerm); err != nil {
return err
}
return nil
}
// LoadFromYaml photo data from a YAML file.
func (m *Photo) LoadFromYaml(fileName string) error {
data, err := ioutil.ReadFile(fileName)
if err != nil {
return err
}
if err := yaml.Unmarshal(data, m); err != nil {
return err
}
return nil
}
// YamlFileName returns the YAML backup file name.
func (m *Photo) YamlFileName(originalsPath, sidecarPath string) string {
return fs.FileName(filepath.Join(originalsPath, m.PhotoPath, m.PhotoName), sidecarPath, originalsPath, fs.YamlExt, false)
}