From 10f4168ec464153705a18377717a7b3c0cc24468 Mon Sep 17 00:00:00 2001 From: Janne Koschinski <janne@kuschku.de> Date: Fri, 25 Sep 2020 22:31:33 +0200 Subject: [PATCH] Add kind field --- src/metadata_loader.js | 6 ++++-- src/storage.js | 1 + 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/metadata_loader.js b/src/metadata_loader.js index 437e637..fb7cea2 100644 --- a/src/metadata_loader.js +++ b/src/metadata_loader.js @@ -34,12 +34,13 @@ class MetadataLoader { this.storage = storage; } - async transformData(ids, imdbResult, tmdbResult, tmdbContentRatings, tmdbTranslations) { + async transformData(ids, imdbResult, tmdbResult, tmdbContentRatings, tmdbTranslations, kind) { const [title] = await Title.upsert({ id: ids.uuid, imdb_id: ids.imdb, tmdb_id: ids.tmdb, tvdb_id: ids.tvdb, + kind, original_language: tmdbResult.original_language, runtime: imdbResult.runtime, year_start: imdbResult.startYear, @@ -179,6 +180,7 @@ class MetadataLoader { }) const [episodeTitle] = await Title.upsert({ id: mapping.episode_id, + kind: "episode", imdb_id: imdbResult.id, tmdb_id: tmdbResult.id, tvdb_id: null, @@ -514,7 +516,7 @@ class MetadataLoader { episodes: imdbEpisodes, }; - const title = await this.transformData(ids, imdbData, tmdbResult, tmdbContentRatings, tmdbTranslations); + const title = await this.transformData(ids, imdbData, tmdbResult, tmdbContentRatings, tmdbTranslations, isShow ? "show" : "movie"); return { title: title, diff --git a/src/storage.js b/src/storage.js index de5e1e7..a9b9bd3 100644 --- a/src/storage.js +++ b/src/storage.js @@ -78,6 +78,7 @@ class Backend { imdb_id: sequelize.DataTypes.TEXT, tmdb_id: sequelize.DataTypes.INTEGER, tvdb_id: sequelize.DataTypes.INTEGER, + kind: sequelize.DataTypes.TEXT, original_language: sequelize.DataTypes.TEXT, runtime: sequelize.DataTypes.INTEGER, year_start: sequelize.DataTypes.INTEGER, -- GitLab