mirror of
https://github.com/knightcrawler-stremio/knightcrawler.git
synced 2024-12-20 03:29:51 +00:00
66609c2a46
* add new indexes, and change year column to int * Change gist to gin, and change year to int * Producer changes for new gin query * Fully map the rtn response using json dump from Pydantic Also updates Rtn to 0.1.9 * Add housekeeping script to reconcile imdb ids. * Join Torrent onto the ingested torrent table Ensure that a torrent can always find the details of where it came from, and how it was parsed. * Version bump for release * missing quote on table name
55 lines
2.3 KiB
PL/PgSQL
55 lines
2.3 KiB
PL/PgSQL
DROP FUNCTION IF EXISTS kc_maintenance_reconcile_dmm_imdb_ids();
|
|
CREATE OR REPLACE FUNCTION kc_maintenance_reconcile_dmm_imdb_ids()
|
|
RETURNS INTEGER AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
imdb_rec RECORD;
|
|
rows_affected INTEGER := 0;
|
|
BEGIN
|
|
RAISE NOTICE 'Starting Reconciliation of DMM IMDB Ids...';
|
|
FOR rec IN
|
|
SELECT
|
|
it."id" as "ingestion_id",
|
|
t."infoHash",
|
|
it."category" as "ingestion_category",
|
|
f."id" as "file_Id",
|
|
f."title" as "file_Title",
|
|
(rtn_response->>'raw_title')::text as "raw_title",
|
|
(rtn_response->>'parsed_title')::text as "parsed_title",
|
|
(rtn_response->>'year')::int as "year"
|
|
FROM torrents t
|
|
JOIN ingested_torrents it ON t."ingestedTorrentId" = it."id"
|
|
JOIN files f ON t."infoHash" = f."infoHash"
|
|
WHERE t."provider" = 'DMM'
|
|
LOOP
|
|
RAISE NOTICE 'Processing record with file_Id: %', rec."file_Id";
|
|
FOR imdb_rec IN
|
|
SELECT * FROM search_imdb_meta(
|
|
rec."parsed_title",
|
|
CASE
|
|
WHEN rec."ingestion_category" = 'tv' THEN 'tvSeries'
|
|
WHEN rec."ingestion_category" = 'movies' THEN 'movie'
|
|
END,
|
|
CASE
|
|
WHEN rec."year" = 0 THEN NULL
|
|
ELSE rec."year" END,
|
|
1)
|
|
LOOP
|
|
IF imdb_rec IS NOT NULL THEN
|
|
RAISE NOTICE 'Updating file_Id: % with imdbId: %, parsed title: %, imdb title: %', rec."file_Id", imdb_rec."imdb_id", rec."parsed_title", imdb_rec."title";
|
|
UPDATE "files"
|
|
SET "imdbId" = imdb_rec."imdb_id"
|
|
WHERE "id" = rec."file_Id";
|
|
rows_affected := rows_affected + 1;
|
|
ELSE
|
|
RAISE NOTICE 'No IMDB ID found for file_Id: %, parsed title: %, imdb title: %, setting imdbId to NULL', rec."file_Id", rec."parsed_title", imdb_rec."title";
|
|
UPDATE "files"
|
|
SET "imdbId" = NULL
|
|
WHERE "id" = rec."file_Id";
|
|
END IF;
|
|
END LOOP;
|
|
END LOOP;
|
|
RAISE NOTICE 'Finished reconciliation. Total rows affected: %', rows_affected;
|
|
RETURN rows_affected;
|
|
END;
|
|
$$ LANGUAGE plpgsql; |