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, )