From f8aa0433a3f41d6ccada7cbf403daeee2bddb5cf Mon Sep 17 00:00:00 2001 From: Stash Dev Date: Mon, 6 Jan 2020 15:18:06 -0800 Subject: [PATCH] Fix scene covers not importing correctly --- pkg/manager/task_import.go | 2 ++ pkg/models/querybuilder_scene.go | 6 ++++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/pkg/manager/task_import.go b/pkg/manager/task_import.go index 0df2891f6..634c164dc 100644 --- a/pkg/manager/task_import.go +++ b/pkg/manager/task_import.go @@ -391,6 +391,8 @@ func (t *ImportTask) ImportScenes(ctx context.Context) { if len(coverImageData) > 0 { if err = SetSceneScreenshot(mappingJSON.Checksum, coverImageData); err != nil { logger.Warnf("[scenes] <%s> failed to create cover image: %s", mappingJSON.Checksum, err.Error()) + } else { + newScene.Cover = coverImageData } } } diff --git a/pkg/models/querybuilder_scene.go b/pkg/models/querybuilder_scene.go index 9ceff6ad1..7242937b5 100644 --- a/pkg/models/querybuilder_scene.go +++ b/pkg/models/querybuilder_scene.go @@ -42,9 +42,11 @@ func (qb *SceneQueryBuilder) Create(newScene Scene, tx *sqlx.Tx) (*Scene, error) ensureTx(tx) result, err := tx.NamedExec( `INSERT INTO scenes (checksum, path, title, details, url, date, rating, size, duration, video_codec, - audio_codec, width, height, framerate, bitrate, studio_id, created_at, updated_at) + audio_codec, width, height, framerate, bitrate, studio_id, cover, + created_at, updated_at) VALUES (:checksum, :path, :title, :details, :url, :date, :rating, :size, :duration, :video_codec, - :audio_codec, :width, :height, :framerate, :bitrate, :studio_id, :created_at, :updated_at) + :audio_codec, :width, :height, :framerate, :bitrate, :studio_id, :cover, + :created_at, :updated_at) `, newScene, )