☰

IMDB Ratings Importer

Import ratings for movies, TV series and episodes from a csv file.

  1. // ==UserScript==
  2. // @name            IMDB Ratings Importer
  3. // @namespace       Neinei0k_imdb
  4. // @include         https://www.imdb.com/user/ur*/ratings*
  5. // @grant           none
  6. // @version         2.01
  7. // @license         GNU General Public License v3.0 or later
  8. // @description	    Import ratings for movies, TV series and episodes from a csv file.
  9. // ==/UserScript==
  10.  
  11. let request_data_add_rating = {
  12.     "query": "mutation UpdateTitleRating($rating: Int!, $titleId: ID!) {\n  rateTitle(input: {rating: $rating, titleId: $titleId}) {\n    rating {\n      value\n    }\n  }\n}",
  13.     "operationName": "UpdateTitleRating",
  14.     "variables": {
  15.         "rating": 0,
  16.         "titleId": ""
  17.     }
  18. }
  19.  
  20. let elements = createHTMLForm();
  21.  
  22. function log(level, message) {
  23. 	console.log("(IMDB Ratings Importer) " + level + ": " + message);
  24. }
  25.  
  26. function setStatus(message) {
  27. 	elements.status.textContent = message;
  28. }
  29.  
  30. function createHTMLForm() {
  31. 	let elements = {};
  32.  
  33.   try {
  34.     let root = createRoot();
  35.     elements.text = createTextField(root);
  36.  
  37.     if (isFileAPISupported()) {
  38.       elements.file = createFileInput(root);
  39.       elements.isFromFile = createFromFileCheckbox(root);
  40.     } else {
  41.       createFileAPINotSupportedMessage(root);
  42.     }
  43.     elements.status = createStatusBar(root);
  44.     
  45.     createImportButton(root);
  46.   } catch (message) {
  47.    	log("Error", message);
  48.   }
  49.  
  50. 	return elements;
  51. }
  52.  
  53. function isFileAPISupported() {
  54. 	return window.File && window.FileReader && window.FileList && window.Blob;
  55. }
  56.  
  57. function createRoot() {
  58. 	let container = document.querySelector('.ipc-page-section--base');
  59. 	if (container === null) {
  60. 		throw ".ipc-page-section--base element not found";
  61. 	}
  62.   let nextChild = container.children[0];
  63. 	let root = document.createElement('div');
  64. 	root.setAttribute('class', 'aux-content-widget-2 ipc-list-card--base ipc-list-card--border-line');
  65. 	root.style.height = 'initial';
  66.   root.style.marginTop = '30px';
  67.   root.style.marginBottom = '30px';
  68.   root.style.padding = '10px';
  69. 	container.insertBefore(root, nextChild);
  70.  
  71. 	return root;
  72. }
  73.  
  74. function createTextField(root) {
  75. 	let text = document.createElement('textarea');
  76. 	text.style = "background-color: white; width: 100%; height: 100px; overflow: initial;";
  77. 	root.appendChild(text);
  78. 	root.appendChild(document.createElement('br'));
  79.  
  80. 	return text;
  81. }
  82.  
  83. function createFileInput(root) {
  84. 	let file = document.createElement('input');
  85. 	file.type = 'file';
  86. 	file.disabled = true;
  87.   file.style.marginBottom = '10px';
  88. 	root.appendChild(file);
  89. 	root.appendChild(document.createElement('br'));
  90.  
  91. 	return file;
  92. }
  93.  
  94. function createFromFileCheckbox(root) {
  95. 	let isFromFile = createCheckbox("Import from file (otherwise import from text)");
  96. 	root.appendChild(isFromFile.label);
  97. 	root.appendChild(document.createElement('br'));
  98.  
  99. 	isFromFile.checkbox.addEventListener('change', fromFileOrTextChangeHandler, false);
  100. 	
  101. 	return isFromFile.checkbox;
  102. }
  103.  
  104. function createCheckbox(textContent) {
  105. 	let checkbox = document.createElement('input');
  106. 	checkbox.type = 'checkbox';
  107. 	checkbox.style = 'width: initial;';
  108.  
  109. 	let text = document.createElement('span');
  110. 	text.style = 'font-weight: normal;';
  111. 	text.textContent = textContent;
  112.  
  113. 	let label = document.createElement('label');
  114. 	label.appendChild(checkbox);
  115. 	label.appendChild(text);
  116.  
  117. 	return {label: label, checkbox: checkbox};
  118. }
  119.  
  120. function fromFileOrTextChangeHandler(event) {
  121. 	let isChecked = event.target.checked;
  122. 	elements.text.disabled = isChecked;
  123. 	elements.file.disabled = !isChecked;
  124. }
  125.  
  126. function createFileAPINotSupportedMessage(root) {
  127. 	let notSupported = document.createElement('div');
  128. 	notSupported.style = 'font-weight: normal;';
  129.   notSupported.style.marginTop = '10px';
  130.   notSupported.style.marginBottom = '10px';
  131. 	notSupported.textContent = "Your browser does not support File API for reading local files.";
  132. 	root.appendChild(notSupported);
  133. }
  134.  
  135. function createStatusBar(root) {
  136. 	let status = document.createElement('div');
  137. 	status.textContent = "Insert text or choose file. Press 'Import Ratings' button.";
  138.   status.style.marginTop = '10px';
  139.   status.style.marginBottom = '10px';
  140. 	root.appendChild(status);
  141.  
  142. 	return status;
  143. }
  144.  
  145. function createImportButton(root) {
  146. 	let importList = document.createElement('button');
  147. 	importList.class = 'btn';
  148. 	importList.textContent = "Import Ratings";
  149. 	root.appendChild(importList);
  150.  
  151. 	importList.addEventListener('click', importRatingsClickHandler, false);
  152. }
  153.  
  154. function importRatingsClickHandler(event) {
  155. 	if (elements.hasOwnProperty('isFromFile') && elements.isFromFile.checked) {
  156. 		readFile();
  157. 	} else {
  158. 		importRatings(extractItems(elements.text.value));
  159. 	}
  160. }
  161.  
  162. function readFile() {
  163. 	let file = elements.file.files[0];
  164. 	if (file !== undefined) {
  165. 		log("Info", "Reading file " + file.name);
  166. 		setStatus("Reading file " + file.name);
  167. 		let fileReader = new FileReader();
  168. 		fileReader.onload = fileOnloadHandler;
  169. 		fileReader.readAsText(file);
  170. 	} else {
  171. 		setStatus("Error: File is not selected");
  172. 	}
  173. }
  174.  
  175. function fileOnloadHandler(event) {
  176. 	if (event.target.error === null) {
  177. 		importRatings(extractItems(event.target.result));
  178. 	} else {
  179. 		log("Error", e.target.error);
  180. 		setStatus("Error: " + e.target.error);
  181. 	}
  182. }
  183.  
  184. function extractItems(text) {
  185. 	try {
  186. 		return extractItemsFromCSV(text);
  187. 	} catch (message) {
  188. 		log("Error", message);
  189. 		setStatus("Error: " + message);
  190. 		return [];
  191. 	}
  192. }
  193.  
  194. function extractItemsFromCSV(text) {
  195. 	let table = parseCSV(text);
  196. 	let fields = findFieldNumbers(table);
  197.  
  198. 	log("Info", "Found csv file fields Const(" + fields.const + ") and Your Rating(" + fields.rating + ")");
  199.  
  200. 	re = new RegExp("^tt[0-9]{7,8}$");
  201. 	let items = [];
  202. 	// Add elements to the list
  203. 	for (let i = 1; i < table.length; i++) {
  204.     let fconst = table[i][fields.const];
  205.     let frating = table[i][fields.rating];
  206. 		if (re.exec(fconst) === null) {
  207. 			throw "Invalid 'const' field format on line " + (i+1);
  208. 		}
  209.     if (frating === "") {
  210.       continue;
  211.     }
  212.     frating = parseInt(frating);
  213.     if (isNaN(frating)) {
  214.      	throw "Invalid 'your rating' field format on line " + (i+1); 
  215.     }
  216. 		items.push({const: fconst, rating: frating});
  217. 	}
  218.  
  219. 	return items;
  220. }
  221.  
  222. function parseCSV(text) {
  223. 	let lines = text.split(/\r|\n/);
  224. 	let table = [];
  225. 	for (let i=0; i < lines.length; i++) {
  226. 		if (isEmpty(lines[i])) {
  227. 			continue;
  228. 		}
  229. 		let isInsideString = false;
  230. 		let row = [""];
  231. 		for (let j=0; j < lines[i].length; j++) {
  232. 			if (!isInsideString && lines[i][j] === ',') {
  233. 				row.push("");
  234. 			} else if (lines[i][j] === '"') {
  235. 				isInsideString = !isInsideString;
  236. 			} else {
  237. 				row[row.length-1] += lines[i][j];
  238. 			}
  239. 		}
  240. 		table.push(row);
  241. 		if (isInsideString) {
  242. 			throw "Wrong number of \" on line " + (i+1);
  243. 		}
  244. 		if (row.length != table[0].length) {
  245. 			throw "Wrong number of fields on line " + (i+1) + ". Expected " + table[0].length + " but found " + row.length + ".";
  246. 		}
  247. 	}
  248.  
  249. 	return table;
  250. }
  251.  
  252. function isEmpty(str) {
  253. 	return str.trim().length === 0;
  254. }
  255.  
  256. function findFieldNumbers(table) {
  257. 	let fieldNames = table[0];
  258. 	let fieldNumbers = {'const': -1, 'rating': -1};
  259.  
  260. 	for (let i = 0; i < fieldNames.length; i++) {
  261. 		let fieldName = fieldNames[i].toLowerCase().trim();
  262. 		if (fieldName === 'const') {
  263. 			fieldNumbers.const = i;
  264. 		} else if (fieldName === 'your rating') {
  265. 			fieldNumbers.rating = i;
  266. 		}
  267. 	}
  268.  
  269. 	if (fieldNumbers.const === -1) {
  270. 		throw "Field 'const' not found.";
  271. 	} else if (fieldNumbers.rating === -1) {
  272.    	throw "Field 'your rating' not found."; 
  273.   }
  274. 	return fieldNumbers;
  275. }
  276.  
  277. async function importRatings(list) {
  278. 	if (list.length === 0)
  279. 		return;
  280.  
  281. 	let l = {};
  282. 	l.list = list;
  283. 	l.ready = 0;
  284.   
  285.   for (let i = 0; i < list.length; ++i) {
  286.     log("Info", `Setting rating ${list[i].rating} for ${list[i].const}...`);
  287.     request_data_add_rating.variables.rating = list[i].rating;
  288.     request_data_add_rating.variables.titleId = list[i].const;
  289.     await sendRequest(request_data_add_rating);
  290.     setStatus(`Ready ${i+1} of ${list.length}.`);
  291.   }
  292. }
  293.  
  294. function sendRequest(data) {
  295.   return fetch("https://api.graphql.imdb.com/", {
  296.     "credentials": "include",
  297.     "headers": {
  298.         "Accept": "application/graphql+json, application/json",
  299.         "content-type": "application/json",
  300.     },
  301.     "referrer": "https://www.imdb.com/",
  302.     "body": JSON.stringify(data),
  303.     "method": "POST",
  304.     "mode": "cors"
  305. 	}).then((response) => {
  306.   	if (!response.ok) {
  307.       throw new Error(`Request failed with status code ${response.status}`);
  308.     }
  309.  
  310.     return response.json();
  311.   });
  312. }