☰

IMDB List Importer

Import list of titles or people in the imdb list

  1. // ==UserScript==
  2. // @name            IMDB List Importer
  3. // @namespace       Neinei0k_imdb
  4. // @include         https://www.imdb.com/list/*
  5. // @version         10.3
  6. // @license         GNU General Public License v3.0 or later
  7. // @description	Import list of titles or people in the imdb list
  8. // ==/UserScript==
  9.  
  10. var request_data_add_item = {
  11.     "query": "mutation AddConstToList($listId: ID!, $constId: ID!, $includeListItemMetadata: Boolean!, $refTagQueryParam: String, $originalTitleText: Boolean) {\n  addItemToList(input: {listId: $listId, item: {itemElementId: $constId}}) {\n    listId\n    modifiedItem {\n      ...ListItemMetadata\n      listItem @include(if: $includeListItemMetadata) {\n        ... on Title {\n          ...TitleListItemMetadata\n        }\n        ... on Name {\n          ...NameListItemMetadata\n        }\n        ... on Image {\n          ...ImageListItemMetadata\n        }\n        ... on Video {\n          ...VideoListItemMetadata\n        }\n      }\n    }\n  }\n}\n\nfragment ListItemMetadata on ListItemNode {\n  itemId\n  createdDate\n  description {\n    originalText {\n      markdown\n      plaidHtml(showLineBreak: true)\n      plainText\n    }\n  }\n}\n\nfragment TitleListItemMetadata on Title {\n  ...BaseTitleCard\n  plot {\n    plotText {\n      plainText\n    }\n  }\n  latestTrailer {\n    id\n  }\n  series {\n    series {\n      id\n      originalTitleText {\n        text\n      }\n      releaseYear {\n        endYear\n        year\n      }\n      titleText {\n        text\n      }\n    }\n  }\n}\n\nfragment BaseTitleCard on Title {\n  id\n  titleText {\n    text\n  }\n  titleType {\n    id\n    text\n    canHaveEpisodes\n    displayableProperty {\n      value {\n        plainText\n      }\n    }\n  }\n  originalTitleText {\n    text\n  }\n  primaryImage {\n    id\n    width\n    height\n    url\n    caption {\n      plainText\n    }\n  }\n  releaseYear {\n    year\n    endYear\n  }\n  ratingsSummary {\n    aggregateRating\n    voteCount\n  }\n  runtime {\n    seconds\n  }\n  certificate {\n    rating\n  }\n  canRate {\n    isRatable\n  }\n  titleGenres {\n    genres(limit: 3) {\n      genre {\n        text\n      }\n    }\n  }\n  canHaveEpisodes\n}\n\nfragment NameListItemMetadata on Name {\n  id\n  primaryImage {\n    url\n    caption {\n      plainText\n    }\n    width\n    height\n  }\n  nameText {\n    text\n  }\n  primaryProfessions {\n    category {\n      text\n    }\n  }\n  knownFor(first: 1) {\n    edges {\n      node {\n        summary {\n          yearRange {\n            year\n            endYear\n          }\n        }\n        title {\n          id\n          originalTitleText {\n            text\n          }\n          titleText {\n            text\n          }\n          titleType {\n            canHaveEpisodes\n          }\n        }\n      }\n    }\n  }\n  bio {\n    displayableArticle {\n      body {\n        plaidHtml(\n          queryParams: $refTagQueryParam\n          showOriginalTitleText: $originalTitleText\n        )\n      }\n    }\n  }\n}\n\nfragment ImageListItemMetadata on Image {\n  id\n  url\n  height\n  width\n  caption {\n    plainText\n  }\n  names(limit: 4) {\n    id\n    nameText {\n      text\n    }\n  }\n  titles(limit: 1) {\n    id\n    titleText {\n      text\n    }\n    originalTitleText {\n      text\n    }\n    releaseYear {\n      year\n      endYear\n    }\n  }\n}\n\nfragment VideoListItemMetadata on Video {\n  id\n  thumbnail {\n    url\n    width\n    height\n  }\n  name {\n    value\n    language\n  }\n  description {\n    value\n    language\n  }\n  runtime {\n    unit\n    value\n  }\n  primaryTitle {\n    id\n    originalTitleText {\n      text\n    }\n    titleText {\n      text\n    }\n    titleType {\n      canHaveEpisodes\n    }\n    releaseYear {\n      year\n      endYear\n    }\n  }\n}",
  12.     "operationName": "AddConstToList",
  13.     "variables": {
  14.         "listId": "",
  15.         "constId": "",
  16.         "includeListItemMetadata": true,
  17.         "refTagQueryParam": "lsedt_add_items",
  18.         "originalTitleText": false
  19.     }
  20. }
  21.  
  22. var request_data_add_description = {
  23.     "query": "mutation EditListItemDescription($listId: ID!, $itemId: ID!, $itemDescription: String!) {\n  editListItemDescription(\n    input: {listId: $listId, itemId: $itemId, itemDescription: $itemDescription}\n  ) {\n    formattedItemDescription {\n      originalText {\n        markdown\n        plaidHtml(showLineBreak: true)\n        plainText\n      }\n    }\n  }\n}",
  24.     "operationName": "EditListItemDescription",
  25.     "variables": {
  26.         "listId": "",
  27.         "itemId": "",
  28.         "itemDescription": ""
  29.     }
  30. }
  31.  
  32. var request_data_reorder_item = {
  33.     "query": "mutation reorderListItems($input: ReorderListInput!) {\n  reorderList(input: $input) {\n    listId\n  }\n}",
  34.     "operationName": "reorderListItems",
  35.     "variables": {
  36.         "input": {
  37.             "newPositions": [
  38.                 /*{
  39.                     "position": -1,
  40.                     "itemId": ""
  41.                 }*/
  42.             ],
  43.             "listId": ""
  44.         }
  45.     }
  46. }
  47.  
  48. if (/^https:\/\/(www.)?imdb.com\/list\/ls[0-9]+\/edit/.test(document.location)) {
  49.     var elements = createHTMLForm();
  50. }
  51.  
  52. function log(level, message) {
  53. 	console.log("(IMDB List Importer) " + level + ": " + message);
  54. }
  55.  
  56. function setStatus(message) {
  57. 	elements.status.textContent = message;
  58. }
  59.  
  60. function createHTMLForm() {
  61. 	let elements = {};
  62.  
  63.   try {
  64.     let root = createRoot();
  65.     elements.text = createTextField(root);
  66.  
  67.     if (isFileAPISupported()) {
  68.       elements.file = createFileInput(root);
  69.       elements.isFromFile = createFromFileCheckbox(root);
  70.     } else {
  71.       createFileAPINotSupportedMessage(root);
  72.     }
  73.  
  74.     elements.isCSV = createCSVCheckbox(root);
  75.     elements.isUnique = createUniqueCheckbox(root);
  76.     
  77.     elements.isReverse = createReverseCheckbox(root);
  78.     elements.insert = createInsertRadio(root);
  79.     elements.insertOther = createInsertOtherInput(root);
  80.     
  81.     elements.status = createStatusBar(root);
  82.     
  83.     createImportButton(root);
  84.   } catch (message) {
  85.    	log("Error", message);
  86.   }
  87.  
  88. 	return elements;
  89. }
  90.  
  91. function isFileAPISupported() {
  92. 	return window.File && window.FileReader && window.FileList && window.Blob;
  93. }
  94.  
  95. function createRoot() {
  96. 	let container = document.querySelector('section.ipc-page-section--base');
  97. 	if (container === null) {
  98. 		throw "section.section.ipc-page-section--base element not found";
  99. 	}
  100. 	let root = document.createElement('div');
  101. 	root.setAttribute('class', 'search-bar ipc-list-card--base ipc-list-card--border-line');
  102. 	root.style.height = 'initial';
  103.   root.style.marginTop = '30px';
  104.   root.style.marginBottom = '30px';
  105.   root.style.padding = '10px';
  106. 	container.insertBefore(root, container.children[1]);
  107.  
  108. 	return root;
  109. }
  110.  
  111. function createTextField(root) {
  112. 	let text = document.createElement('textarea');
  113. 	text.style = "background-color: white; width: 100%; height: 100px; overflow: initial;";
  114. 	root.appendChild(text);
  115. 	root.appendChild(document.createElement('br'));
  116.  
  117. 	return text;
  118. }
  119.  
  120. function createFileInput(root) {
  121. 	let file = document.createElement('input');
  122. 	file.type = 'file';
  123. 	file.disabled = true;
  124.   file.style.marginBottom = '10px';
  125. 	root.appendChild(file);
  126. 	root.appendChild(document.createElement('br'));
  127.  
  128. 	return file;
  129. }
  130.  
  131. function createFromFileCheckbox(root) {
  132. 	let isFromFile = createCheckbox("Import from file (otherwise import from text)");
  133. 	root.appendChild(isFromFile.label);
  134. 	root.appendChild(document.createElement('br'));
  135.  
  136. 	isFromFile.checkbox.addEventListener('change', fromFileOrTextChangeHandler, false);
  137. 	
  138. 	return isFromFile.checkbox;
  139. }
  140.  
  141. function createCheckbox(textContent) {
  142. 	let checkbox = document.createElement('input');
  143. 	checkbox.type = 'checkbox';
  144. 	checkbox.style = 'width: initial;';
  145.  
  146. 	let text = document.createElement('span');
  147. 	text.style = 'font-weight: normal;';
  148. 	text.textContent = textContent;
  149.  
  150. 	let label = document.createElement('label');
  151. 	label.appendChild(checkbox);
  152. 	label.appendChild(text);
  153.  
  154. 	return {label: label, checkbox: checkbox};
  155. }
  156.  
  157. function createRadio(name, value, textContent) {
  158.   let radio = document.createElement('input');
  159. 	radio.type = 'radio';
  160. 	radio.style = 'width: initial;';
  161.   radio.name = name;
  162.   radio.value = value;
  163.  
  164. 	let text = document.createElement('span');
  165. 	text.style = 'font-weight: normal;';
  166. 	text.textContent = textContent;
  167.  
  168. 	let label = document.createElement('label');
  169. 	label.appendChild(radio);
  170. 	label.appendChild(text);
  171.  
  172. 	return {label: label, radio: radio};
  173. }
  174.  
  175. function fromFileOrTextChangeHandler(event) {
  176. 	let isChecked = event.target.checked;
  177. 	elements.text.disabled = isChecked;
  178. 	elements.file.disabled = !isChecked;
  179. }
  180.  
  181. function createFileAPINotSupportedMessage(root) {
  182. 	let notSupported = document.createElement('div');
  183. 	notSupported.style = 'font-weight: normal;';
  184.   notSupported.style.marginTop = '10px';
  185.   notSupported.style.marginBottom = '10px';
  186. 	notSupported.textContent = "Your browser does not support File API for reading local files.";
  187. 	root.appendChild(notSupported);
  188. }
  189.  
  190. function createCSVCheckbox(root) {
  191. 	let isCSV = createCheckbox("Data from .csv file (otherwise extract ids from text)");
  192. 	isCSV.checkbox.checked = true;
  193. 	root.appendChild(isCSV.label);
  194. 	root.appendChild(document.createElement('br'));
  195.  
  196. 	return isCSV.checkbox;
  197. }
  198.  
  199. function createUniqueCheckbox(root) {
  200. 	let isUnique = createCheckbox("Add only unique elements");
  201. 	root.appendChild(isUnique.label);
  202. 	root.appendChild(document.createElement('br'));
  203.  
  204. 	return isUnique.checkbox;
  205. }
  206.  
  207. function createReverseCheckbox(root) {
  208.   let isReverse = createCheckbox("Reverse Items on Insertion");
  209.   root.appendChild(document.createElement('br'));
  210. 	root.appendChild(isReverse.label);
  211. 	root.appendChild(document.createElement('br'));
  212.  
  213. 	return isReverse.checkbox;
  214. }
  215.  
  216. function createInsertRadio(root) {
  217.   let insertBegin = createRadio("imdb_list_importer_insert", "1", "Insert in the Beginning");
  218.   let insertEnd = createRadio("imdb_list_importer_insert", "-1", "Insert in the End");
  219.   let insertOther = createRadio("imdb_list_importer_insert", "0", "Insert in Other Position");
  220.   
  221.   insertEnd.radio.checked = true;
  222.   
  223.   root.appendChild(insertBegin.label);
  224.   root.appendChild(document.createElement('br'));
  225.   root.appendChild(insertEnd.label);
  226.   root.appendChild(document.createElement('br'));
  227.   root.appendChild(insertOther.label);
  228.   root.appendChild(document.createElement('br'));
  229.   
  230.   insertBegin.radio.addEventListener('change', isOtherHandler, false);
  231.   insertEnd.radio.addEventListener('change', isOtherHandler, false);
  232.   insertOther.radio.addEventListener('change', isOtherHandler, false);
  233.   
  234.   return {'begin': insertBegin.radio, 'end': insertEnd.radio, 'other': insertOther.radio};
  235. }
  236.  
  237. function createInsertOtherInput(root) {
  238.   let insertOtherInput = document.createElement('input');
  239.   insertOtherInput.type = 'text';
  240.   insertOtherInput.disabled = true;
  241.   root.appendChild(insertOtherInput);
  242.   root.appendChild(document.createElement('br'));
  243.   root.appendChild(document.createElement('br'));
  244.   
  245.   return insertOtherInput;
  246. }
  247.  
  248. function isOtherHandler(event) {
  249.   let isDisable = event.target.value != "0";
  250.   elements.insertOther.disabled = isDisable;
  251. }
  252.  
  253.  
  254. function createStatusBar(root) {
  255. 	let status = document.createElement('div');
  256. 	status.textContent = "Set-up parameters. Insert text or choose file. Press 'Import List' button.";
  257.   status.style.marginTop = '10px';
  258.   status.style.marginBottom = '10px';
  259. 	root.appendChild(status);
  260.  
  261. 	return status;
  262. }
  263.  
  264. function createImportButton(root) {
  265. 	let importList = document.createElement('button');
  266. 	importList.class = 'btn';
  267. 	importList.textContent = "Import List";
  268. 	root.appendChild(importList);
  269.  
  270. 	importList.addEventListener('click', importListClickHandler, false);
  271. }
  272.  
  273. function importListClickHandler(event) {
  274. 	if (elements.hasOwnProperty('isFromFile') && elements.isFromFile.checked) {
  275. 		readFile();
  276. 	} else {
  277. 		importList(extractItems(elements.text.value));
  278. 	}
  279. }
  280.  
  281. function readFile() {
  282. 	let file = elements.file.files[0];
  283. 	if (file !== undefined) {
  284. 		log("Info", "Reading file " + file.name);
  285. 		setStatus("Reading file " + file.name);
  286. 		let fileReader = new FileReader();
  287. 		fileReader.onload = fileOnloadHandler;
  288. 		fileReader.readAsText(file);
  289. 	} else {
  290. 		setStatus("Error: File is not selected");
  291. 	}
  292. }
  293.  
  294. function fileOnloadHandler(event) {
  295. 	if (event.target.error === null) {
  296. 		importList(extractItems(event.target.result));
  297. 	} else {
  298. 		log("Error", e.target.error);
  299. 		setStatus("Error: " + e.target.error);
  300. 	}
  301. }
  302.  
  303. function extractItems(text) {
  304. 	try {
  305. 		let itemRegExp = getRegExpForItems();
  306.  
  307. 		if (elements.isCSV.checked) {
  308. 			return extractItemsFromCSV(itemRegExp, text);
  309. 		} else {
  310. 			return extractItemsFromText(itemRegExp, text);
  311. 		}
  312. 	} catch (message) {
  313. 		log("Error", message);
  314. 		setStatus("Error: " + message);
  315. 		return [];
  316. 	}
  317. }
  318.  
  319. function getRegExpForItems() {
  320. 	return "[a-z]{2}[0-9]{7,8}";
  321. }
  322.  
  323. function extractItemsFromCSV(re, text) {
  324. 	let table = parseCSV(text);
  325. 	let fields = findFieldNumbers(table);
  326.  
  327.   if (fields.description !== -1) {
  328. 		log("Info", "Found csv file fields Const(" + fields.const + ") and Description(" + fields.description + ")");
  329.   } else {
  330.     log("Info", "Found csv file field Const(" + fields.const + "). Description field is not found.");
  331.   }
  332.  
  333. 	re = new RegExp("^" + re + "$");
  334. 	let items = [];
  335. 	// Add elements to the list
  336. 	for (let i = 1; i < table.length; i++) {
  337. 		let row = table[i];
  338. 		if (re.exec(row[fields.const]) === null) {
  339. 			throw "Invalid 'const' field format on line " + (i+1);
  340. 		}
  341. 		if (elements.isUnique.checked) {
  342. 			let exists = items.findIndex(function(v){
  343. 				return v.const === row[fields.const];
  344. 			});
  345. 			if (exists !== -1) continue;
  346. 		}
  347. 		items.push({const: row[fields.const], description: (fields.description == -1 ? "" : row[fields.description])});
  348. 	}
  349.  
  350. 	return items;
  351. }
  352.  
  353. function parseCSV(text) {
  354. 	let lines = text.split(/\r|\n/);
  355. 	let table = [];
  356. 	for (let i=0; i < lines.length; i++) {
  357. 		if (isEmpty(lines[i])) {
  358. 			continue;
  359. 		}
  360. 		let isInsideString = false;
  361. 		let row = [""];
  362. 		for (let j=0; j < lines[i].length; j++) {
  363. 			if (!isInsideString && lines[i][j] === ',') {
  364. 				row.push("");
  365. 			} else if (lines[i][j] === '"') {
  366. 				isInsideString = !isInsideString;
  367. 			} else {
  368. 				row[row.length-1] += lines[i][j];
  369. 			}
  370. 		}
  371. 		table.push(row);
  372. 		if (isInsideString) {
  373. 			throw "Wrong number of \" on line " + (i+1);
  374. 		}
  375. 		if (row.length != table[0].length) {
  376. 			throw "Wrong number of fields on line " + (i+1) + ". Expected " + table[0].length + " but found " + row.length + ".";
  377. 		}
  378. 	}
  379.  
  380. 	return table;
  381. }
  382.  
  383. function isEmpty(str) {
  384. 	return str.trim().length === 0;
  385. }
  386.  
  387. function findFieldNumbers(table) {
  388. 	let fieldNames = table[0];
  389. 	let fieldNumbers = {'const': -1, 'description': -1};
  390.  
  391. 	for (let i = 0; i < fieldNames.length; i++) {
  392. 		let fieldName = fieldNames[i].toLowerCase().trim();
  393. 		if (fieldName === 'const') {
  394. 			fieldNumbers.const = i;
  395. 		} else if (fieldName === 'description') {
  396. 			fieldNumbers.description = i;
  397. 		}
  398. 	}
  399.  
  400. 	if (fieldNumbers.const === -1) {
  401. 		throw "Field 'const' not found."
  402. 	}
  403. 	return fieldNumbers;
  404. }
  405.  
  406. function extractItemsFromText(re, text) {
  407. 	re = new RegExp(re);
  408. 	let items = [];
  409. 	let e;
  410. 	while ((e = re.exec(text)) !== null) {
  411. 		let flag = '';
  412. 		if (elements.isUnique.checked)
  413. 			flag = 'g';
  414. 		text = text.replace(new RegExp(e[0], flag), '');
  415. 		items.push({const: e[0], description: ""});
  416. 	}
  417. 	return items;
  418. }
  419.  
  420. async function importList(list) {
  421. 	if (list.length === 0)
  422. 		return;
  423.  
  424. 	let msg = "Elements to add: ";
  425. 	for (let i = 0;  i < list.length; i++)
  426. 		msg += list[i].const + ",";
  427. 	log("Info", msg);
  428.  
  429. 	let list_id = /ls[0-9]{1,}/.exec(location.href)[0];
  430.   
  431.   if (elements.isReverse.checked) {
  432.    	list.reverse();
  433.   }
  434.  
  435.   let items = [];
  436.   for (let i = 0; i < list.length; ++i) {
  437.     log("Info", `Adding element ${String(i+1)}: ${list[i].const}...`);
  438.  
  439.     request_data_add_item.variables.listId = list_id;
  440.   	request_data_add_item.variables.constId = list[i].const;
  441.     let response = await sendRequest(request_data_add_item);
  442.     
  443.     let listItemId = response.data.addItemToList.modifiedItem.itemId;
  444.     log("Info", `${list[i].const} added as ${listItemId}`);
  445.     items.push(listItemId);
  446.       
  447. 		if (list[i].description.length !== 0) {
  448.       log("Info", `Adding description to ${listItemId}...`);
  449.       request_data_add_description.variables.listId = list_id;
  450.     	request_data_add_description.variables.itemId = listItemId;
  451.     	request_data_add_description.variables.itemDescription = list[i].description;
  452.       await sendRequest(request_data_add_description);
  453.     }
  454.     
  455.     setStatus(`Ready ${String(i+1)} of ${list.length}.`);  
  456.   }
  457.   
  458.   let insertPosition = -1;
  459.   if (elements.insert.begin.checked) {
  460.    	insertPosition = 1;
  461.   } else if (elements.insert.other.checked) {
  462.    	insertPosition = Number(elements.insertOther.value);
  463.     if (isNaN(insertPosition) || insertPosition < 1) {
  464.      	insertPosition = -1; 
  465.     }
  466.   }
  467.   if (insertPosition != -1) {
  468.     request_data_reorder_item.variables.input.newPositions = [];
  469.     request_data_reorder_item.variables.input.listId = list_id;
  470.  
  471.     for (let i = items.length - 1; i >= 0; i--) {
  472.       request_data_reorder_item.variables.input.newPositions.push({
  473.         "position": insertPosition,
  474.         "itemId": items[i]
  475.       });
  476.     }
  477.  
  478.     log("Info", `Moving items to position ${insertPosition}...`);
  479.     await sendRequest(request_data_reorder_item);
  480.   }
  481.   
  482.   location.reload();
  483. }
  484.  
  485. function sendRequest(data) {
  486.   return fetch("https://api.graphql.imdb.com/", {
  487.     "credentials": "include",
  488.     "headers": {
  489.         "Accept": "application/graphql+json, application/json",
  490.         "content-type": "application/json",
  491.     },
  492.     "referrer": "https://www.imdb.com/",
  493.     "body": JSON.stringify(data),
  494.     "method": "POST",
  495.     "mode": "cors"
  496. 	}).then((response) => {
  497.   	if (!response.ok) {
  498.       throw new Error(`Request failed with status code ${response.status}`);
  499.     }
  500.  
  501.     return response.json();
  502.   });
  503. }

欢迎访问Greasy Fork镜像

土豆服务器，请按需使用

镜像地址随时可能被墙，建议加群获取最新地址

加入Q群

QingJ © 2025

镜像随时可能失效，请加Q群300939539或关注我们的公众号极客氢云获取最新地址