Greasy Fork is available in English.
Import ratings for movies, TV series and episodes from a csv file.
// ==UserScript==
// @name IMDB Ratings Importer
// @namespace Neinei0k_imdb
// @include https://www.imdb.com/user/ur*/ratings*
// @grant none
// @version 2.01
// @license GNU General Public License v3.0 or later
// @description Import ratings for movies, TV series and episodes from a csv file.
// ==/UserScript==
let request_data_add_rating = {
"query": "mutation UpdateTitleRating($rating: Int!, $titleId: ID!) {\n rateTitle(input: {rating: $rating, titleId: $titleId}) {\n rating {\n value\n }\n }\n}",
"operationName": "UpdateTitleRating",
"variables": {
"rating": 0,
"titleId": ""
}
}
let elements = createHTMLForm();
function log(level, message) {
console.log("(IMDB Ratings Importer) " + level + ": " + message);
}
function setStatus(message) {
elements.status.textContent = message;
}
function createHTMLForm() {
let elements = {};
try {
let root = createRoot();
elements.text = createTextField(root);
if (isFileAPISupported()) {
elements.file = createFileInput(root);
elements.isFromFile = createFromFileCheckbox(root);
} else {
createFileAPINotSupportedMessage(root);
}
elements.status = createStatusBar(root);
createImportButton(root);
} catch (message) {
log("Error", message);
}
return elements;
}
function isFileAPISupported() {
return window.File && window.FileReader && window.FileList && window.Blob;
}
function createRoot() {
let container = document.querySelector('.ipc-page-section--base');
if (container === null) {
throw ".ipc-page-section--base element not found";
}
let nextChild = container.children[0];
let root = document.createElement('div');
root.setAttribute('class', 'aux-content-widget-2 ipc-list-card--base ipc-list-card--border-line');
root.style.height = 'initial';
root.style.marginTop = '30px';
root.style.marginBottom = '30px';
root.style.padding = '10px';
container.insertBefore(root, nextChild);
return root;
}
function createTextField(root) {
let text = document.createElement('textarea');
text.style = "background-color: white; width: 100%; height: 100px; overflow: initial;";
root.appendChild(text);
root.appendChild(document.createElement('br'));
return text;
}
function createFileInput(root) {
let file = document.createElement('input');
file.type = 'file';
file.disabled = true;
file.style.marginBottom = '10px';
root.appendChild(file);
root.appendChild(document.createElement('br'));
return file;
}
function createFromFileCheckbox(root) {
let isFromFile = createCheckbox("Import from file (otherwise import from text)");
root.appendChild(isFromFile.label);
root.appendChild(document.createElement('br'));
isFromFile.checkbox.addEventListener('change', fromFileOrTextChangeHandler, false);
return isFromFile.checkbox;
}
function createCheckbox(textContent) {
let checkbox = document.createElement('input');
checkbox.type = 'checkbox';
checkbox.style = 'width: initial;';
let text = document.createElement('span');
text.style = 'font-weight: normal;';
text.textContent = textContent;
let label = document.createElement('label');
label.appendChild(checkbox);
label.appendChild(text);
return {label: label, checkbox: checkbox};
}
function fromFileOrTextChangeHandler(event) {
let isChecked = event.target.checked;
elements.text.disabled = isChecked;
elements.file.disabled = !isChecked;
}
function createFileAPINotSupportedMessage(root) {
let notSupported = document.createElement('div');
notSupported.style = 'font-weight: normal;';
notSupported.style.marginTop = '10px';
notSupported.style.marginBottom = '10px';
notSupported.textContent = "Your browser does not support File API for reading local files.";
root.appendChild(notSupported);
}
function createStatusBar(root) {
let status = document.createElement('div');
status.textContent = "Insert text or choose file. Press 'Import Ratings' button.";
status.style.marginTop = '10px';
status.style.marginBottom = '10px';
root.appendChild(status);
return status;
}
function createImportButton(root) {
let importList = document.createElement('button');
importList.class = 'btn';
importList.textContent = "Import Ratings";
root.appendChild(importList);
importList.addEventListener('click', importRatingsClickHandler, false);
}
function importRatingsClickHandler(event) {
if (elements.hasOwnProperty('isFromFile') && elements.isFromFile.checked) {
readFile();
} else {
importRatings(extractItems(elements.text.value));
}
}
function readFile() {
let file = elements.file.files[0];
if (file !== undefined) {
log("Info", "Reading file " + file.name);
setStatus("Reading file " + file.name);
let fileReader = new FileReader();
fileReader.onload = fileOnloadHandler;
fileReader.readAsText(file);
} else {
setStatus("Error: File is not selected");
}
}
function fileOnloadHandler(event) {
if (event.target.error === null) {
importRatings(extractItems(event.target.result));
} else {
log("Error", e.target.error);
setStatus("Error: " + e.target.error);
}
}
function extractItems(text) {
try {
return extractItemsFromCSV(text);
} catch (message) {
log("Error", message);
setStatus("Error: " + message);
return [];
}
}
function extractItemsFromCSV(text) {
let table = parseCSV(text);
let fields = findFieldNumbers(table);
log("Info", "Found csv file fields Const(" + fields.const + ") and Your Rating(" + fields.rating + ")");
re = new RegExp("^tt[0-9]{7,8}$");
let items = [];
// Add elements to the list
for (let i = 1; i < table.length; i++) {
let fconst = table[i][fields.const];
let frating = table[i][fields.rating];
if (re.exec(fconst) === null) {
throw "Invalid 'const' field format on line " + (i+1);
}
if (frating === "") {
continue;
}
frating = parseInt(frating);
if (isNaN(frating)) {
throw "Invalid 'your rating' field format on line " + (i+1);
}
items.push({const: fconst, rating: frating});
}
return items;
}
function parseCSV(text) {
let lines = text.split(/\r|\n/);
let table = [];
for (let i=0; i < lines.length; i++) {
if (isEmpty(lines[i])) {
continue;
}
let isInsideString = false;
let row = [""];
for (let j=0; j < lines[i].length; j++) {
if (!isInsideString && lines[i][j] === ',') {
row.push("");
} else if (lines[i][j] === '"') {
isInsideString = !isInsideString;
} else {
row[row.length-1] += lines[i][j];
}
}
table.push(row);
if (isInsideString) {
throw "Wrong number of \" on line " + (i+1);
}
if (row.length != table[0].length) {
throw "Wrong number of fields on line " + (i+1) + ". Expected " + table[0].length + " but found " + row.length + ".";
}
}
return table;
}
function isEmpty(str) {
return str.trim().length === 0;
}
function findFieldNumbers(table) {
let fieldNames = table[0];
let fieldNumbers = {'const': -1, 'rating': -1};
for (let i = 0; i < fieldNames.length; i++) {
let fieldName = fieldNames[i].toLowerCase().trim();
if (fieldName === 'const') {
fieldNumbers.const = i;
} else if (fieldName === 'your rating') {
fieldNumbers.rating = i;
}
}
if (fieldNumbers.const === -1) {
throw "Field 'const' not found.";
} else if (fieldNumbers.rating === -1) {
throw "Field 'your rating' not found.";
}
return fieldNumbers;
}
async function importRatings(list) {
if (list.length === 0)
return;
let l = {};
l.list = list;
l.ready = 0;
for (let i = 0; i < list.length; ++i) {
log("Info", `Setting rating ${list[i].rating} for ${list[i].const}...`);
request_data_add_rating.variables.rating = list[i].rating;
request_data_add_rating.variables.titleId = list[i].const;
await sendRequest(request_data_add_rating);
setStatus(`Ready ${i+1} of ${list.length}.`);
}
}
function sendRequest(data) {
return fetch("https://api.graphql.imdb.com/", {
"credentials": "include",
"headers": {
"Accept": "application/graphql+json, application/json",
"content-type": "application/json",
},
"referrer": "https://www.imdb.com/",
"body": JSON.stringify(data),
"method": "POST",
"mode": "cors"
}).then((response) => {
if (!response.ok) {
throw new Error(`Request failed with status code ${response.status}`);
}
return response.json();
});
}