├── .gitignore ├── LICENSE ├── README.md ├── csv-examples ├── import-barebones.csv └── import.csv ├── export_trakt.py ├── helpers.py ├── import_trakt.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | /.idea/ 2 | *.csv 3 | *.ini 4 | *.pyc 5 | *.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | GNU GENERAL PUBLIC LICENSE 2 | Version 3, 29 June 2007 3 | 4 | Copyright (C) 2007 Free Software Foundation, Inc. 5 | Everyone is permitted to copy and distribute verbatim copies 6 | of this license document, but changing it is not allowed. 7 | 8 | Preamble 9 | 10 | The GNU General Public License is a free, copyleft license for 11 | software and other kinds of works. 12 | 13 | The licenses for most software and other practical works are designed 14 | to take away your freedom to share and change the works. By contrast, 15 | the GNU General Public License is intended to guarantee your freedom to 16 | share and change all versions of a program--to make sure it remains free 17 | software for all its users. We, the Free Software Foundation, use the 18 | GNU General Public License for most of our software; it applies also to 19 | any other work released this way by its authors. You can apply it to 20 | your programs, too. 21 | 22 | When we speak of free software, we are referring to freedom, not 23 | price. Our General Public Licenses are designed to make sure that you 24 | have the freedom to distribute copies of free software (and charge for 25 | them if you wish), that you receive source code or can get it if you 26 | want it, that you can change the software or use pieces of it in new 27 | free programs, and that you know you can do these things. 28 | 29 | To protect your rights, we need to prevent others from denying you 30 | these rights or asking you to surrender the rights. Therefore, you have 31 | certain responsibilities if you distribute copies of the software, or if 32 | you modify it: responsibilities to respect the freedom of others. 33 | 34 | For example, if you distribute copies of such a program, whether 35 | gratis or for a fee, you must pass on to the recipients the same 36 | freedoms that you received. You must make sure that they, too, receive 37 | or can get the source code. And you must show them these terms so they 38 | know their rights. 39 | 40 | Developers that use the GNU GPL protect your rights with two steps: 41 | (1) assert copyright on the software, and (2) offer you this License 42 | giving you legal permission to copy, distribute and/or modify it. 43 | 44 | For the developers' and authors' protection, the GPL clearly explains 45 | that there is no warranty for this free software. For both users' and 46 | authors' sake, the GPL requires that modified versions be marked as 47 | changed, so that their problems will not be attributed erroneously to 48 | authors of previous versions. 49 | 50 | Some devices are designed to deny users access to install or run 51 | modified versions of the software inside them, although the manufacturer 52 | can do so. This is fundamentally incompatible with the aim of 53 | protecting users' freedom to change the software. The systematic 54 | pattern of such abuse occurs in the area of products for individuals to 55 | use, which is precisely where it is most unacceptable. Therefore, we 56 | have designed this version of the GPL to prohibit the practice for those 57 | products. If such problems arise substantially in other domains, we 58 | stand ready to extend this provision to those domains in future versions 59 | of the GPL, as needed to protect the freedom of users. 60 | 61 | Finally, every program is threatened constantly by software patents. 62 | States should not allow patents to restrict development and use of 63 | software on general-purpose computers, but in those that do, we wish to 64 | avoid the special danger that patents applied to a free program could 65 | make it effectively proprietary. To prevent this, the GPL assures that 66 | patents cannot be used to render the program non-free. 67 | 68 | The precise terms and conditions for copying, distribution and 69 | modification follow. 70 | 71 | TERMS AND CONDITIONS 72 | 73 | 0. Definitions. 74 | 75 | "This License" refers to version 3 of the GNU General Public License. 76 | 77 | "Copyright" also means copyright-like laws that apply to other kinds of 78 | works, such as semiconductor masks. 79 | 80 | "The Program" refers to any copyrightable work licensed under this 81 | License. Each licensee is addressed as "you". "Licensees" and 82 | "recipients" may be individuals or organizations. 83 | 84 | To "modify" a work means to copy from or adapt all or part of the work 85 | in a fashion requiring copyright permission, other than the making of an 86 | exact copy. The resulting work is called a "modified version" of the 87 | earlier work or a work "based on" the earlier work. 88 | 89 | A "covered work" means either the unmodified Program or a work based 90 | on the Program. 91 | 92 | To "propagate" a work means to do anything with it that, without 93 | permission, would make you directly or secondarily liable for 94 | infringement under applicable copyright law, except executing it on a 95 | computer or modifying a private copy. Propagation includes copying, 96 | distribution (with or without modification), making available to the 97 | public, and in some countries other activities as well. 98 | 99 | To "convey" a work means any kind of propagation that enables other 100 | parties to make or receive copies. Mere interaction with a user through 101 | a computer network, with no transfer of a copy, is not conveying. 102 | 103 | An interactive user interface displays "Appropriate Legal Notices" 104 | to the extent that it includes a convenient and prominently visible 105 | feature that (1) displays an appropriate copyright notice, and (2) 106 | tells the user that there is no warranty for the work (except to the 107 | extent that warranties are provided), that licensees may convey the 108 | work under this License, and how to view a copy of this License. If 109 | the interface presents a list of user commands or options, such as a 110 | menu, a prominent item in the list meets this criterion. 111 | 112 | 1. Source Code. 113 | 114 | The "source code" for a work means the preferred form of the work 115 | for making modifications to it. "Object code" means any non-source 116 | form of a work. 117 | 118 | A "Standard Interface" means an interface that either is an official 119 | standard defined by a recognized standards body, or, in the case of 120 | interfaces specified for a particular programming language, one that 121 | is widely used among developers working in that language. 122 | 123 | The "System Libraries" of an executable work include anything, other 124 | than the work as a whole, that (a) is included in the normal form of 125 | packaging a Major Component, but which is not part of that Major 126 | Component, and (b) serves only to enable use of the work with that 127 | Major Component, or to implement a Standard Interface for which an 128 | implementation is available to the public in source code form. A 129 | "Major Component", in this context, means a major essential component 130 | (kernel, window system, and so on) of the specific operating system 131 | (if any) on which the executable work runs, or a compiler used to 132 | produce the work, or an object code interpreter used to run it. 133 | 134 | The "Corresponding Source" for a work in object code form means all 135 | the source code needed to generate, install, and (for an executable 136 | work) run the object code and to modify the work, including scripts to 137 | control those activities. However, it does not include the work's 138 | System Libraries, or general-purpose tools or generally available free 139 | programs which are used unmodified in performing those activities but 140 | which are not part of the work. For example, Corresponding Source 141 | includes interface definition files associated with source files for 142 | the work, and the source code for shared libraries and dynamically 143 | linked subprograms that the work is specifically designed to require, 144 | such as by intimate data communication or control flow between those 145 | subprograms and other parts of the work. 146 | 147 | The Corresponding Source need not include anything that users 148 | can regenerate automatically from other parts of the Corresponding 149 | Source. 150 | 151 | The Corresponding Source for a work in source code form is that 152 | same work. 153 | 154 | 2. Basic Permissions. 155 | 156 | All rights granted under this License are granted for the term of 157 | copyright on the Program, and are irrevocable provided the stated 158 | conditions are met. This License explicitly affirms your unlimited 159 | permission to run the unmodified Program. The output from running a 160 | covered work is covered by this License only if the output, given its 161 | content, constitutes a covered work. This License acknowledges your 162 | rights of fair use or other equivalent, as provided by copyright law. 163 | 164 | You may make, run and propagate covered works that you do not 165 | convey, without conditions so long as your license otherwise remains 166 | in force. You may convey covered works to others for the sole purpose 167 | of having them make modifications exclusively for you, or provide you 168 | with facilities for running those works, provided that you comply with 169 | the terms of this License in conveying all material for which you do 170 | not control copyright. Those thus making or running the covered works 171 | for you must do so exclusively on your behalf, under your direction 172 | and control, on terms that prohibit them from making any copies of 173 | your copyrighted material outside their relationship with you. 174 | 175 | Conveying under any other circumstances is permitted solely under 176 | the conditions stated below. Sublicensing is not allowed; section 10 177 | makes it unnecessary. 178 | 179 | 3. Protecting Users' Legal Rights From Anti-Circumvention Law. 180 | 181 | No covered work shall be deemed part of an effective technological 182 | measure under any applicable law fulfilling obligations under article 183 | 11 of the WIPO copyright treaty adopted on 20 December 1996, or 184 | similar laws prohibiting or restricting circumvention of such 185 | measures. 186 | 187 | When you convey a covered work, you waive any legal power to forbid 188 | circumvention of technological measures to the extent such circumvention 189 | is effected by exercising rights under this License with respect to 190 | the covered work, and you disclaim any intention to limit operation or 191 | modification of the work as a means of enforcing, against the work's 192 | users, your or third parties' legal rights to forbid circumvention of 193 | technological measures. 194 | 195 | 4. Conveying Verbatim Copies. 196 | 197 | You may convey verbatim copies of the Program's source code as you 198 | receive it, in any medium, provided that you conspicuously and 199 | appropriately publish on each copy an appropriate copyright notice; 200 | keep intact all notices stating that this License and any 201 | non-permissive terms added in accord with section 7 apply to the code; 202 | keep intact all notices of the absence of any warranty; and give all 203 | recipients a copy of this License along with the Program. 204 | 205 | You may charge any price or no price for each copy that you convey, 206 | and you may offer support or warranty protection for a fee. 207 | 208 | 5. Conveying Modified Source Versions. 209 | 210 | You may convey a work based on the Program, or the modifications to 211 | produce it from the Program, in the form of source code under the 212 | terms of section 4, provided that you also meet all of these conditions: 213 | 214 | a) The work must carry prominent notices stating that you modified 215 | it, and giving a relevant date. 216 | 217 | b) The work must carry prominent notices stating that it is 218 | released under this License and any conditions added under section 219 | 7. This requirement modifies the requirement in section 4 to 220 | "keep intact all notices". 221 | 222 | c) You must license the entire work, as a whole, under this 223 | License to anyone who comes into possession of a copy. This 224 | License will therefore apply, along with any applicable section 7 225 | additional terms, to the whole of the work, and all its parts, 226 | regardless of how they are packaged. This License gives no 227 | permission to license the work in any other way, but it does not 228 | invalidate such permission if you have separately received it. 229 | 230 | d) If the work has interactive user interfaces, each must display 231 | Appropriate Legal Notices; however, if the Program has interactive 232 | interfaces that do not display Appropriate Legal Notices, your 233 | work need not make them do so. 234 | 235 | A compilation of a covered work with other separate and independent 236 | works, which are not by their nature extensions of the covered work, 237 | and which are not combined with it such as to form a larger program, 238 | in or on a volume of a storage or distribution medium, is called an 239 | "aggregate" if the compilation and its resulting copyright are not 240 | used to limit the access or legal rights of the compilation's users 241 | beyond what the individual works permit. Inclusion of a covered work 242 | in an aggregate does not cause this License to apply to the other 243 | parts of the aggregate. 244 | 245 | 6. Conveying Non-Source Forms. 246 | 247 | You may convey a covered work in object code form under the terms 248 | of sections 4 and 5, provided that you also convey the 249 | machine-readable Corresponding Source under the terms of this License, 250 | in one of these ways: 251 | 252 | a) Convey the object code in, or embodied in, a physical product 253 | (including a physical distribution medium), accompanied by the 254 | Corresponding Source fixed on a durable physical medium 255 | customarily used for software interchange. 256 | 257 | b) Convey the object code in, or embodied in, a physical product 258 | (including a physical distribution medium), accompanied by a 259 | written offer, valid for at least three years and valid for as 260 | long as you offer spare parts or customer support for that product 261 | model, to give anyone who possesses the object code either (1) a 262 | copy of the Corresponding Source for all the software in the 263 | product that is covered by this License, on a durable physical 264 | medium customarily used for software interchange, for a price no 265 | more than your reasonable cost of physically performing this 266 | conveying of source, or (2) access to copy the 267 | Corresponding Source from a network server at no charge. 268 | 269 | c) Convey individual copies of the object code with a copy of the 270 | written offer to provide the Corresponding Source. This 271 | alternative is allowed only occasionally and noncommercially, and 272 | only if you received the object code with such an offer, in accord 273 | with subsection 6b. 274 | 275 | d) Convey the object code by offering access from a designated 276 | place (gratis or for a charge), and offer equivalent access to the 277 | Corresponding Source in the same way through the same place at no 278 | further charge. You need not require recipients to copy the 279 | Corresponding Source along with the object code. If the place to 280 | copy the object code is a network server, the Corresponding Source 281 | may be on a different server (operated by you or a third party) 282 | that supports equivalent copying facilities, provided you maintain 283 | clear directions next to the object code saying where to find the 284 | Corresponding Source. Regardless of what server hosts the 285 | Corresponding Source, you remain obligated to ensure that it is 286 | available for as long as needed to satisfy these requirements. 287 | 288 | e) Convey the object code using peer-to-peer transmission, provided 289 | you inform other peers where the object code and Corresponding 290 | Source of the work are being offered to the general public at no 291 | charge under subsection 6d. 292 | 293 | A separable portion of the object code, whose source code is excluded 294 | from the Corresponding Source as a System Library, need not be 295 | included in conveying the object code work. 296 | 297 | A "User Product" is either (1) a "consumer product", which means any 298 | tangible personal property which is normally used for personal, family, 299 | or household purposes, or (2) anything designed or sold for incorporation 300 | into a dwelling. In determining whether a product is a consumer product, 301 | doubtful cases shall be resolved in favor of coverage. For a particular 302 | product received by a particular user, "normally used" refers to a 303 | typical or common use of that class of product, regardless of the status 304 | of the particular user or of the way in which the particular user 305 | actually uses, or expects or is expected to use, the product. A product 306 | is a consumer product regardless of whether the product has substantial 307 | commercial, industrial or non-consumer uses, unless such uses represent 308 | the only significant mode of use of the product. 309 | 310 | "Installation Information" for a User Product means any methods, 311 | procedures, authorization keys, or other information required to install 312 | and execute modified versions of a covered work in that User Product from 313 | a modified version of its Corresponding Source. The information must 314 | suffice to ensure that the continued functioning of the modified object 315 | code is in no case prevented or interfered with solely because 316 | modification has been made. 317 | 318 | If you convey an object code work under this section in, or with, or 319 | specifically for use in, a User Product, and the conveying occurs as 320 | part of a transaction in which the right of possession and use of the 321 | User Product is transferred to the recipient in perpetuity or for a 322 | fixed term (regardless of how the transaction is characterized), the 323 | Corresponding Source conveyed under this section must be accompanied 324 | by the Installation Information. But this requirement does not apply 325 | if neither you nor any third party retains the ability to install 326 | modified object code on the User Product (for example, the work has 327 | been installed in ROM). 328 | 329 | The requirement to provide Installation Information does not include a 330 | requirement to continue to provide support service, warranty, or updates 331 | for a work that has been modified or installed by the recipient, or for 332 | the User Product in which it has been modified or installed. Access to a 333 | network may be denied when the modification itself materially and 334 | adversely affects the operation of the network or violates the rules and 335 | protocols for communication across the network. 336 | 337 | Corresponding Source conveyed, and Installation Information provided, 338 | in accord with this section must be in a format that is publicly 339 | documented (and with an implementation available to the public in 340 | source code form), and must require no special password or key for 341 | unpacking, reading or copying. 342 | 343 | 7. Additional Terms. 344 | 345 | "Additional permissions" are terms that supplement the terms of this 346 | License by making exceptions from one or more of its conditions. 347 | Additional permissions that are applicable to the entire Program shall 348 | be treated as though they were included in this License, to the extent 349 | that they are valid under applicable law. If additional permissions 350 | apply only to part of the Program, that part may be used separately 351 | under those permissions, but the entire Program remains governed by 352 | this License without regard to the additional permissions. 353 | 354 | When you convey a copy of a covered work, you may at your option 355 | remove any additional permissions from that copy, or from any part of 356 | it. (Additional permissions may be written to require their own 357 | removal in certain cases when you modify the work.) You may place 358 | additional permissions on material, added by you to a covered work, 359 | for which you have or can give appropriate copyright permission. 360 | 361 | Notwithstanding any other provision of this License, for material you 362 | add to a covered work, you may (if authorized by the copyright holders of 363 | that material) supplement the terms of this License with terms: 364 | 365 | a) Disclaiming warranty or limiting liability differently from the 366 | terms of sections 15 and 16 of this License; or 367 | 368 | b) Requiring preservation of specified reasonable legal notices or 369 | author attributions in that material or in the Appropriate Legal 370 | Notices displayed by works containing it; or 371 | 372 | c) Prohibiting misrepresentation of the origin of that material, or 373 | requiring that modified versions of such material be marked in 374 | reasonable ways as different from the original version; or 375 | 376 | d) Limiting the use for publicity purposes of names of licensors or 377 | authors of the material; or 378 | 379 | e) Declining to grant rights under trademark law for use of some 380 | trade names, trademarks, or service marks; or 381 | 382 | f) Requiring indemnification of licensors and authors of that 383 | material by anyone who conveys the material (or modified versions of 384 | it) with contractual assumptions of liability to the recipient, for 385 | any liability that these contractual assumptions directly impose on 386 | those licensors and authors. 387 | 388 | All other non-permissive additional terms are considered "further 389 | restrictions" within the meaning of section 10. If the Program as you 390 | received it, or any part of it, contains a notice stating that it is 391 | governed by this License along with a term that is a further 392 | restriction, you may remove that term. If a license document contains 393 | a further restriction but permits relicensing or conveying under this 394 | License, you may add to a covered work material governed by the terms 395 | of that license document, provided that the further restriction does 396 | not survive such relicensing or conveying. 397 | 398 | If you add terms to a covered work in accord with this section, you 399 | must place, in the relevant source files, a statement of the 400 | additional terms that apply to those files, or a notice indicating 401 | where to find the applicable terms. 402 | 403 | Additional terms, permissive or non-permissive, may be stated in the 404 | form of a separately written license, or stated as exceptions; 405 | the above requirements apply either way. 406 | 407 | 8. Termination. 408 | 409 | You may not propagate or modify a covered work except as expressly 410 | provided under this License. Any attempt otherwise to propagate or 411 | modify it is void, and will automatically terminate your rights under 412 | this License (including any patent licenses granted under the third 413 | paragraph of section 11). 414 | 415 | However, if you cease all violation of this License, then your 416 | license from a particular copyright holder is reinstated (a) 417 | provisionally, unless and until the copyright holder explicitly and 418 | finally terminates your license, and (b) permanently, if the copyright 419 | holder fails to notify you of the violation by some reasonable means 420 | prior to 60 days after the cessation. 421 | 422 | Moreover, your license from a particular copyright holder is 423 | reinstated permanently if the copyright holder notifies you of the 424 | violation by some reasonable means, this is the first time you have 425 | received notice of violation of this License (for any work) from that 426 | copyright holder, and you cure the violation prior to 30 days after 427 | your receipt of the notice. 428 | 429 | Termination of your rights under this section does not terminate the 430 | licenses of parties who have received copies or rights from you under 431 | this License. If your rights have been terminated and not permanently 432 | reinstated, you do not qualify to receive new licenses for the same 433 | material under section 10. 434 | 435 | 9. Acceptance Not Required for Having Copies. 436 | 437 | You are not required to accept this License in order to receive or 438 | run a copy of the Program. Ancillary propagation of a covered work 439 | occurring solely as a consequence of using peer-to-peer transmission 440 | to receive a copy likewise does not require acceptance. However, 441 | nothing other than this License grants you permission to propagate or 442 | modify any covered work. These actions infringe copyright if you do 443 | not accept this License. Therefore, by modifying or propagating a 444 | covered work, you indicate your acceptance of this License to do so. 445 | 446 | 10. Automatic Licensing of Downstream Recipients. 447 | 448 | Each time you convey a covered work, the recipient automatically 449 | receives a license from the original licensors, to run, modify and 450 | propagate that work, subject to this License. You are not responsible 451 | for enforcing compliance by third parties with this License. 452 | 453 | An "entity transaction" is a transaction transferring control of an 454 | organization, or substantially all assets of one, or subdividing an 455 | organization, or merging organizations. If propagation of a covered 456 | work results from an entity transaction, each party to that 457 | transaction who receives a copy of the work also receives whatever 458 | licenses to the work the party's predecessor in interest had or could 459 | give under the previous paragraph, plus a right to possession of the 460 | Corresponding Source of the work from the predecessor in interest, if 461 | the predecessor has it or can get it with reasonable efforts. 462 | 463 | You may not impose any further restrictions on the exercise of the 464 | rights granted or affirmed under this License. For example, you may 465 | not impose a license fee, royalty, or other charge for exercise of 466 | rights granted under this License, and you may not initiate litigation 467 | (including a cross-claim or counterclaim in a lawsuit) alleging that 468 | any patent claim is infringed by making, using, selling, offering for 469 | sale, or importing the Program or any portion of it. 470 | 471 | 11. Patents. 472 | 473 | A "contributor" is a copyright holder who authorizes use under this 474 | License of the Program or a work on which the Program is based. The 475 | work thus licensed is called the contributor's "contributor version". 476 | 477 | A contributor's "essential patent claims" are all patent claims 478 | owned or controlled by the contributor, whether already acquired or 479 | hereafter acquired, that would be infringed by some manner, permitted 480 | by this License, of making, using, or selling its contributor version, 481 | but do not include claims that would be infringed only as a 482 | consequence of further modification of the contributor version. For 483 | purposes of this definition, "control" includes the right to grant 484 | patent sublicenses in a manner consistent with the requirements of 485 | this License. 486 | 487 | Each contributor grants you a non-exclusive, worldwide, royalty-free 488 | patent license under the contributor's essential patent claims, to 489 | make, use, sell, offer for sale, import and otherwise run, modify and 490 | propagate the contents of its contributor version. 491 | 492 | In the following three paragraphs, a "patent license" is any express 493 | agreement or commitment, however denominated, not to enforce a patent 494 | (such as an express permission to practice a patent or covenant not to 495 | sue for patent infringement). To "grant" such a patent license to a 496 | party means to make such an agreement or commitment not to enforce a 497 | patent against the party. 498 | 499 | If you convey a covered work, knowingly relying on a patent license, 500 | and the Corresponding Source of the work is not available for anyone 501 | to copy, free of charge and under the terms of this License, through a 502 | publicly available network server or other readily accessible means, 503 | then you must either (1) cause the Corresponding Source to be so 504 | available, or (2) arrange to deprive yourself of the benefit of the 505 | patent license for this particular work, or (3) arrange, in a manner 506 | consistent with the requirements of this License, to extend the patent 507 | license to downstream recipients. "Knowingly relying" means you have 508 | actual knowledge that, but for the patent license, your conveying the 509 | covered work in a country, or your recipient's use of the covered work 510 | in a country, would infringe one or more identifiable patents in that 511 | country that you have reason to believe are valid. 512 | 513 | If, pursuant to or in connection with a single transaction or 514 | arrangement, you convey, or propagate by procuring conveyance of, a 515 | covered work, and grant a patent license to some of the parties 516 | receiving the covered work authorizing them to use, propagate, modify 517 | or convey a specific copy of the covered work, then the patent license 518 | you grant is automatically extended to all recipients of the covered 519 | work and works based on it. 520 | 521 | A patent license is "discriminatory" if it does not include within 522 | the scope of its coverage, prohibits the exercise of, or is 523 | conditioned on the non-exercise of one or more of the rights that are 524 | specifically granted under this License. You may not convey a covered 525 | work if you are a party to an arrangement with a third party that is 526 | in the business of distributing software, under which you make payment 527 | to the third party based on the extent of your activity of conveying 528 | the work, and under which the third party grants, to any of the 529 | parties who would receive the covered work from you, a discriminatory 530 | patent license (a) in connection with copies of the covered work 531 | conveyed by you (or copies made from those copies), or (b) primarily 532 | for and in connection with specific products or compilations that 533 | contain the covered work, unless you entered into that arrangement, 534 | or that patent license was granted, prior to 28 March 2007. 535 | 536 | Nothing in this License shall be construed as excluding or limiting 537 | any implied license or other defenses to infringement that may 538 | otherwise be available to you under applicable patent law. 539 | 540 | 12. No Surrender of Others' Freedom. 541 | 542 | If conditions are imposed on you (whether by court order, agreement or 543 | otherwise) that contradict the conditions of this License, they do not 544 | excuse you from the conditions of this License. If you cannot convey a 545 | covered work so as to satisfy simultaneously your obligations under this 546 | License and any other pertinent obligations, then as a consequence you may 547 | not convey it at all. For example, if you agree to terms that obligate you 548 | to collect a royalty for further conveying from those to whom you convey 549 | the Program, the only way you could satisfy both those terms and this 550 | License would be to refrain entirely from conveying the Program. 551 | 552 | 13. Use with the GNU Affero General Public License. 553 | 554 | Notwithstanding any other provision of this License, you have 555 | permission to link or combine any covered work with a work licensed 556 | under version 3 of the GNU Affero General Public License into a single 557 | combined work, and to convey the resulting work. The terms of this 558 | License will continue to apply to the part which is the covered work, 559 | but the special requirements of the GNU Affero General Public License, 560 | section 13, concerning interaction through a network will apply to the 561 | combination as such. 562 | 563 | 14. Revised Versions of this License. 564 | 565 | The Free Software Foundation may publish revised and/or new versions of 566 | the GNU General Public License from time to time. Such new versions will 567 | be similar in spirit to the present version, but may differ in detail to 568 | address new problems or concerns. 569 | 570 | Each version is given a distinguishing version number. If the 571 | Program specifies that a certain numbered version of the GNU General 572 | Public License "or any later version" applies to it, you have the 573 | option of following the terms and conditions either of that numbered 574 | version or of any later version published by the Free Software 575 | Foundation. If the Program does not specify a version number of the 576 | GNU General Public License, you may choose any version ever published 577 | by the Free Software Foundation. 578 | 579 | If the Program specifies that a proxy can decide which future 580 | versions of the GNU General Public License can be used, that proxy's 581 | public statement of acceptance of a version permanently authorizes you 582 | to choose that version for the Program. 583 | 584 | Later license versions may give you additional or different 585 | permissions. However, no additional obligations are imposed on any 586 | author or copyright holder as a result of your choosing to follow a 587 | later version. 588 | 589 | 15. Disclaimer of Warranty. 590 | 591 | THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY 592 | APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT 593 | HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY 594 | OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, 595 | THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 596 | PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM 597 | IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF 598 | ALL NECESSARY SERVICING, REPAIR OR CORRECTION. 599 | 600 | 16. Limitation of Liability. 601 | 602 | IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING 603 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS 604 | THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY 605 | GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE 606 | USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF 607 | DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD 608 | PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS), 609 | EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF 610 | SUCH DAMAGES. 611 | 612 | 17. Interpretation of Sections 15 and 16. 613 | 614 | If the disclaimer of warranty and limitation of liability provided 615 | above cannot be given local legal effect according to their terms, 616 | reviewing courts shall apply local law that most closely approximates 617 | an absolute waiver of all civil liability in connection with the 618 | Program, unless a warranty or assumption of liability accompanies a 619 | copy of the Program in return for a fee. 620 | 621 | END OF TERMS AND CONDITIONS 622 | 623 | How to Apply These Terms to Your New Programs 624 | 625 | If you develop a new program, and you want it to be of the greatest 626 | possible use to the public, the best way to achieve this is to make it 627 | free software which everyone can redistribute and change under these terms. 628 | 629 | To do so, attach the following notices to the program. It is safest 630 | to attach them to the start of each source file to most effectively 631 | state the exclusion of warranty; and each file should have at least 632 | the "copyright" line and a pointer to where the full notice is found. 633 | 634 | 635 | Copyright (C) 636 | 637 | This program is free software: you can redistribute it and/or modify 638 | it under the terms of the GNU General Public License as published by 639 | the Free Software Foundation, either version 3 of the License, or 640 | (at your option) any later version. 641 | 642 | This program is distributed in the hope that it will be useful, 643 | but WITHOUT ANY WARRANTY; without even the implied warranty of 644 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 645 | GNU General Public License for more details. 646 | 647 | You should have received a copy of the GNU General Public License 648 | along with this program. If not, see . 649 | 650 | Also add information on how to contact you by electronic and paper mail. 651 | 652 | If the program does terminal interaction, make it output a short 653 | notice like this when it starts in an interactive mode: 654 | 655 | Copyright (C) 656 | This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'. 657 | This is free software, and you are welcome to redistribute it 658 | under certain conditions; type `show c' for details. 659 | 660 | The hypothetical commands `show w' and `show c' should show the appropriate 661 | parts of the General Public License. Of course, your program's commands 662 | might be different; for a GUI interface, you would use an "about box". 663 | 664 | You should also get your employer (if you work as a programmer) or school, 665 | if any, to sign a "copyright disclaimer" for the program, if necessary. 666 | For more information on this, and how to apply and follow the GNU GPL, see 667 | . 668 | 669 | The GNU General Public License does not permit incorporating your program 670 | into proprietary programs. If your program is a subroutine library, you 671 | may consider it more useful to permit linking proprietary applications with 672 | the library. If this is what you want to do, use the GNU Lesser General 673 | Public License instead of this License. But first, please read 674 | . 675 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # trakt-csv-scripts 2 | 3 | Python scripts that use the [Trakt API](https://trakt.docs.apiary.io/) to export movie and episodes from trakt lists to a csv file, and import them back to any Trakt account. 4 | This project is essentially an updated version of the csv import/export functionality of [xbgmsharp/trakt](https://github.com/xbgmsharp/trakt). 5 | 6 | ## Installation 7 | 8 | You must use Python 3.x. 9 | 10 | To install requirements: 11 | 12 | ```console 13 | $ pip install -r requirements.txt 14 | ``` 15 | 16 | ## Usage 17 | 18 | ### Setup 19 | 20 | * Create an [Trakt.tv application](https://trakt.tv/oauth/applications) to have your own ``client_id`` and ``client_secret``, https://trakt.tv/oauth/applications. 21 | You only need to fill up the ``Name`` with a ``Description`` and ``Redirect uri`` to `urn:ietf:wg:oauth:2.0:oob`, leave the rest empty and click on ``SAVE APP``. 22 | * Run the script to create a default config file ``config.ini`` 23 | ``` 24 | $ python export_trakt.py 25 | ``` 26 | * Edit the config file ``config.ini``. You **must** specify 27 | * ``client_id`` 28 | * ``client_secret`` 29 | * ``username`` 30 | * any other settings appropriate to your environment, eg: URL, proxy, etc... 31 | * Run the script to authenticate against Trakt.tv API using the PIN method. 32 | ``` 33 | $ python export_trakt.py 34 | ``` 35 | * In the CLI, you will be prompted to open a link into a browser and paste the pincode back to the script. You should only need to do this once. 36 | 37 | ### Import 38 | 39 | For help, run 40 | ```console 41 | $ python import_trakt.py -h 42 | ``` 43 | #### Import movies from csv file to history with csv ``watched_at`` time 44 | The following command will import all movies in ``m.csv`` to the default history list in Trakt with the ``watched_at`` and the ``trakt`` id columns specified in ``m.csv``. 45 | ```console 46 | $ python import_trakt.py -c config.ini -f trakt -i m.csv -l history -t movies -w 47 | ``` 48 | #### Import episodes from csv file to history with custom ``watched_at`` time 49 | The following command will import all episodes in ``e.csv`` to the default history list in Trakt with the ``trakt`` id columns specified in ``e.csv``, with a ``watched_at`` date set to ``2014-01-01T00:00:00.000Z``. 50 | ```console 51 | $ python import_trakt.py -c config.ini -f trakt -i e.csv -l history -t episodes -s 2014-01-01T00:00:00.000Z 52 | ``` 53 | #### Import movies from csv file to user list 54 | The following command will import all movies in ``m.csv`` to a custom user list using the ``trakt`` id columns specified in ``m.csv``. 55 | ```console 56 | $ python import_trakt.py -c config.ini -f trakt -i m_f.csv -t movies -u 57 | ``` 58 | 59 | The user will have to provide the id of the user list they want to import to. 60 | ```console 61 | Found 2 user list(s) 62 | 63 | id | name 64 | 21358741 | test2 65 | 21369802 | test 66 | 67 | Type in the id matching with the name of the list you want to import item(s) to. 68 | Input: 69 | ``` 70 | 71 | ### Export 72 | 73 | For help, run 74 | ```console 75 | $ python export_trakt.py -h 76 | ``` 77 | #### Export movies from history list to csv 78 | The following command will export all movies in Trakt's default history list to ``m.csv``. 79 | ```console 80 | $ python export_trakt.py -c config.ini -t movies -o m.csv -l history 81 | ``` 82 | #### Export episodes from history list to csv 83 | The following command will export all episodes in Trakt's default history list to ``e.csv``. 84 | ```console 85 | $ python export_trakt.py -c config.ini -t episodes -o e.csv -l history 86 | ``` 87 | #### Export user list to csv 88 | The following command will export all movies from a custom user list to ``test2.csv``. 89 | ```console 90 | $ python export_trakt.py -c config.ini -t movies -o test2.csv -u 91 | ``` 92 | 93 | The user will have to provide the id of the user list they want to export, or ``all`` for all user lists. This will export them in to csv files with the name of the user list. 94 | ```console 95 | Found 2 user list(s) 96 | 97 | id | name 98 | 21358741 | test2 99 | 21369802 | test 100 | 101 | Type in the id matching with the name of the list you want to export, or 'all' for all lists. 102 | Input: 103 | ``` -------------------------------------------------------------------------------- /csv-examples/import-barebones.csv: -------------------------------------------------------------------------------- 1 | listed_at,trakt,imdb,tmdb 2 | 2021-06-23T21:04:47.000Z,356328,tt12801262,508943 3 | 2021-06-23T21:05:22.000Z,218008,tt3228774,337404 4 | 2021-06-23T21:05:57.000Z,430545,tt6654210,581726 5 | -------------------------------------------------------------------------------- /csv-examples/import.csv: -------------------------------------------------------------------------------- 1 | "listed_at","title","year","trakt","imdb","tmdb" 2 | "2021-06-23T21:04:47.000Z","Luca","2021","356328","tt12801262","508943" 3 | "2021-06-23T21:05:22.000Z","Cruella","2021","218008","tt3228774","337404" 4 | "2021-06-23T21:05:57.000Z","Infinite","2021","430545","tt6654210","581726" 5 | -------------------------------------------------------------------------------- /export_trakt.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | # 4 | # (c) Copyright 2016-2020 xbgmsharp 5 | # 6 | # Purpose: 7 | # Export Movies or TVShows IDs from Trakt.tv 8 | # 9 | # Requirement on Ubuntu/Debian Linux system 10 | # apt-get install python3-dateutil python3-simplejson python3-requests python3-openssl jq 11 | # 12 | # Requirement on Windows on Python 3 13 | # \Scripts\pip3.exe install requests simplejson 14 | # 15 | 16 | import sys, os 17 | # https://urllib3.readthedocs.org/en/latest/security.html#disabling-warnings 18 | # http://quabr.com/27981545/surpress-insecurerequestwarning-unverified-https-request-is-being-made-in-pytho 19 | # http://docs.python-requests.org/en/v2.4.3/user/advanced/#proxies 20 | try: 21 | import simplejson as json 22 | import requests 23 | requests.packages.urllib3.disable_warnings() 24 | import csv 25 | # TODO: Use pandas instead of csv for bigger csv handling 26 | # import pandas as pd 27 | except: 28 | sys.exit( 29 | "Please use your favorite method to install the following module requests and simplejson to use this script" 30 | ) 31 | 32 | import argparse 33 | import configparser 34 | import datetime 35 | import collections 36 | import pprint 37 | import time 38 | 39 | import helpers 40 | 41 | pp = pprint.PrettyPrinter(indent=4) 42 | 43 | desc = """This program export Movies or TVShows IDs from Trakt.tv list.""" 44 | 45 | epilog = """Read a list from Trakt API. 46 | Export them into a CSV file.""" 47 | 48 | _trakt = { 49 | 'client_id': '', # Auth details for trakt API 50 | 'client_secret': '', # Auth details for trakt API 51 | 'access_token': '', # Auth details for trakt API 52 | 'refresh_token': '', # Auth details for trakt API 53 | 'baseurl': 54 | 'https://api.trakt.tv', # Sandbox environment https://api-staging.trakt.tv, 55 | 'config_parser': '', # configparser.ConfigParser object 56 | 'config_path': '', # path of config file 57 | 'username': '' # username 58 | } 59 | 60 | _headers = { 61 | 'Accept': 'application/json', # required per API 62 | 'Content-Type': 'application/json', # required per API 63 | 'User-Agent': 'Tratk importer', # User-agent 64 | 'Connection': 65 | 'Keep-Alive', # Thanks to urllib3, keep-alive is 100% automatic within a session! 66 | 'trakt-api-version': '2', # required per API 67 | 'trakt-api-key': '', # required per API 68 | 'Authorization': '', # required per API 69 | } 70 | 71 | _proxy = { 72 | 'proxy': False, # True or False, trigger proxy use 73 | 'host': 'https://127.0.0.1', # Host/IP of the proxy 74 | 'port': '3128' # Port of the proxy 75 | } 76 | 77 | _proxyDict = { 78 | "http": _proxy['host'] + ':' + _proxy['port'], 79 | "https": _proxy['host'] + ':' + _proxy['port'] 80 | } 81 | 82 | response_arr = [] 83 | 84 | 85 | def write_csv(options, results): 86 | """Write list output into a CSV file format 87 | """ 88 | if len(results) == 0: 89 | return None 90 | 91 | print("Writing to: {0}".format(options.output)) 92 | # Write result CSV, works with windows now 93 | with open(options.output, 'w', encoding='utf-8', newline='') as fp: 94 | mycsv = csv.DictWriter(fp, 95 | fieldnames=list(results[0].keys()), 96 | quoting=csv.QUOTE_ALL) 97 | mycsv.writeheader() 98 | for row in results: 99 | mycsv.writerow(row) 100 | fp.close() 101 | 102 | 103 | # TODO: Not sure if removing from list works 104 | def api_remove_from_list(options, remove_data, is_id=False): 105 | """API call for Sync / Remove from list 106 | """ 107 | time.sleep(1) 108 | url = _trakt['baseurl'] + '/sync/{list}/remove'.format(list=options.list) 109 | if not is_id: 110 | values = {options.type: remove_data} 111 | else: 112 | values = {'ids': remove_data} 113 | json_data = json.dumps(values) 114 | if options.verbose: 115 | print(url) 116 | pp.pprint(json_data) 117 | if _proxy['proxy']: 118 | r = requests.post(url, 119 | data=json_data, 120 | headers=_headers, 121 | proxies=_proxyDict, 122 | timeout=(10, 60)) 123 | else: 124 | r = requests.post(url, 125 | data=json_data, 126 | headers=_headers, 127 | timeout=(5, 60)) 128 | if r.status_code != 200: 129 | print("Error removing items from {list}: {status} [{text}]".format( 130 | list=options.list, status=r.status_code, text=r.text)) 131 | return None 132 | else: 133 | return json.loads(r.text) 134 | 135 | 136 | def process_export_data(options, export_data): 137 | """Process a set of data and create a csv""" 138 | if export_data: 139 | print("Found {count} {type}(s) from {list}".format( 140 | count=len(export_data), type=options.type, list=options.list)) 141 | 142 | if options.list == 'history': 143 | options.time = 'watched_at' 144 | elif options.list == 'watchlist': 145 | options.time = 'listed_at' 146 | elif options.list == 'collection': 147 | options.time = 'collected_at' 148 | elif options.userlist != None: 149 | options.time = 'listed_at' 150 | 151 | export_csv = [] 152 | find_dupids = [] 153 | for data in export_data: 154 | # TODO: Add support for exporting movies and episodes to the same lsit 155 | #pp.pprint(data) 156 | # If movie or show 157 | #if options.type[:-1] != "episode" and 'imdb' in data[options.type[:-1]]['ids']: 158 | #print(data) 159 | if 'movie' in data: 160 | # Since exporting from Trakt, use Trakt id for finding dupids 161 | find_dupids.append(data[options.type[:-1]]['ids']['trakt']) 162 | export_csv.append({ 163 | options.time: data[options.time], 164 | 'title': data[options.type[:-1]]['title'], 165 | 'year': data[options.type[:-1]]['year'], 166 | 'trakt': data[options.type[:-1]]['ids']['trakt'], 167 | 'imdb': data[options.type[:-1]]['ids']['imdb'], 168 | 'tmdb': data[options.type[:-1]]['ids']['tmdb'] 169 | }) 170 | # If episode 171 | elif 'episode' in data: 172 | find_dupids.append(data[options.type[:-1]]['ids']['trakt']) 173 | if not data['episode']['title']: 174 | data['episode']['title'] = "no episode title" 175 | 176 | row = { 177 | options.time: data[options.time], 178 | 'season': data[options.type[:-1]]['season'], 179 | 'episode': data[options.type[:-1]]['number'], 180 | 'episode_title': data['episode']['title'], 181 | 'show_title': '', 182 | 'trakt': data[options.type[:-1]]['ids']['trakt'], 183 | 'tvdb': data[options.type[:-1]]['ids']['tvdb'], 184 | 'tmdb': data[options.type[:-1]]['ids']['tmdb'] 185 | } 186 | if options.list == "collection": 187 | del row['show_title'] 188 | else: 189 | row['show_title'] = data['show']['title'] 190 | export_csv.append(row) 191 | # TODO: Don't know if options.clean works 192 | ## Empty list after export 193 | if options.clean: 194 | cleanup_results = {'sentids': 0, 'deleted': 0, 'not_found': 0} 195 | to_remove = [] 196 | for data in export_data: 197 | # TODO add filter 198 | #if data[options.time] == "2012-01-01T00:00:00.000Z": 199 | to_remove.append({'ids': data[options.type[:-1]]['ids']}) 200 | if len(to_remove) >= 10: # Remove by batch of 10 201 | cleanup_results['sentids'] += len(to_remove) 202 | result = api_remove_from_list(options, to_remove) 203 | if result: 204 | print("Result: {0}".format(result)) 205 | if 'deleted' in result and result['deleted'] and options.type != "shows": 206 | cleanup_results['deleted'] += result['deleted'][ 207 | options.type] 208 | if 'not_found' in result and result['not_found'] and options.type != "shows": 209 | cleanup_results['not_found'] += len( 210 | result['not_found'][options.type]) 211 | to_remove = [] 212 | # Remove the rest 213 | if len(to_remove) > 0: 214 | #print pp.pprint(data) 215 | cleanup_results['sentids'] += len(to_remove) 216 | result = api_remove_from_list(options, to_remove) 217 | if result: 218 | print("Result: {0}".format(result)) 219 | if 'deleted' in result and result['deleted'] and options.type != "shows": 220 | cleanup_results['deleted'] += result['deleted'][ 221 | options.type] 222 | if 'not_found' in result and result['not_found'] and options.type != "shows": 223 | cleanup_results['not_found'] += len( 224 | result['not_found'][options.type]) 225 | print( 226 | "Overall cleanup {sent} {type}, results deleted:{deleted}, not_found:{not_found}" 227 | .format(sent=cleanup_results['sentids'], 228 | type=options.type, 229 | deleted=cleanup_results['deleted'], 230 | not_found=cleanup_results['not_found'])) 231 | 232 | ## Find duplicate and remove duplicate 233 | dup_ids = [ 234 | item for item, count in list(collections.Counter(find_dupids).items()) 235 | if count > 1 236 | ] 237 | print("Found {dups} duplicate(s) out of {total} {entry}(s)".format( 238 | entry=options.type, dups=len(dup_ids), total=len(find_dupids))) 239 | if options.dup: 240 | if len(dup_ids) > 0: 241 | print(dup_ids) 242 | dup_results = {'sentids': 0, 'deleted': 0, 'not_found': 0} 243 | to_remove = [] 244 | for dupid in find_dupids: 245 | count = 0 246 | for data in export_data: 247 | if data[options.type[:-1]]['ids']['trakt'] == dupid: 248 | #print "{0} {1}".format(dupid, data['id']) 249 | count += 1 250 | if count > 1: 251 | print("Removing {0} {1}".format(dupid, data['id'])) 252 | to_remove.append(data['id']) 253 | dup_results['sentids'] += len(to_remove) 254 | result = api_remove_from_list(options, 255 | to_remove, 256 | is_id=True) 257 | if len(to_remove) >= 10: # Remove by batch of 10 258 | if result: 259 | print("Result: {0}".format(result)) 260 | if 'deleted' in result and result['deleted']: 261 | dup_results['deleted'] += result[ 262 | 'deleted'][options.type] 263 | if 'not_found' in result and result[ 264 | 'not_found']: 265 | dup_results['not_found'] += len( 266 | result['not_found'][options.type]) 267 | to_remove = [] 268 | ## Remove the rest 269 | if len(to_remove) > 0: 270 | dup_results['sentids'] += len(to_remove) 271 | result = api_remove_from_list(options, to_remove, is_id=True) 272 | if result: 273 | print("Result: {0}".format(result)) 274 | if 'deleted' in result and result['deleted']: 275 | dup_results['deleted'] += result['deleted'][options.type] 276 | if 'not_found' in result and result['not_found']: 277 | dup_results['not_found'] += len( 278 | result['not_found'][options.type]) 279 | to_remove = [] 280 | print( 281 | "Overall {dup} duplicate {sent} {type}, results deleted:{deleted}, not_found:{not_found}" 282 | .format(dup=len(dup_ids), 283 | sent=dup_results['sentids'], 284 | type=options.type, 285 | deleted=dup_results['deleted'], 286 | not_found=dup_results['not_found'])) 287 | # print(export_csv) 288 | ## Write export data into CSV file 289 | write_csv(options, export_csv) 290 | 291 | 292 | def main(): 293 | """ 294 | Main program loop 295 | * Read configuration file and validate 296 | * Authenticate if require 297 | * Export data from Trakt.tv 298 | * Cleanup list from Trakt.tv 299 | * Write to CSV 300 | """ 301 | ## Parse inputs if any 302 | parser = argparse.ArgumentParser(description=desc, epilog=epilog) 303 | list_group = parser.add_mutually_exclusive_group(required=True) 304 | parser.add_argument('-v', action='version', version='%(prog)s 0.3') 305 | parser.add_argument( 306 | '-c', 307 | '--config', 308 | help='allow to overwrite default config filename, default %(default)s', 309 | action='store', 310 | type=str, 311 | dest='config', 312 | default='config.ini') 313 | parser.add_argument( 314 | '-o', 315 | '--output', 316 | help='allow to overwrite default output filename, default %(default)s', 317 | nargs='?', 318 | type=str, 319 | const='export.csv', 320 | default=None) 321 | parser.add_argument('-t', 322 | '--type', 323 | help='allow to overwrite type, default %(default)s', 324 | choices=['movies', 'shows', 'episodes'], 325 | dest='type', 326 | default='movies') 327 | list_group.add_argument( 328 | '-l', 329 | '--list', 330 | help='allow to overwrite default list, default %(default)s', 331 | choices=['watchlist', 'collection', 'history'], 332 | dest='list', 333 | default='history') 334 | list_group.add_argument( 335 | '-u', 336 | '--userlist', 337 | help='allow to export a user custom list, default %(default)s', 338 | dest='userlist', 339 | default=False, 340 | action='store_true') 341 | parser.add_argument('-C', 342 | '--clean', 343 | help='empty list after export, default %(default)s', 344 | default=False, 345 | action='store_true', 346 | dest='clean') 347 | parser.add_argument( 348 | '-D', 349 | '--duplicate', 350 | help='remove duplicate from list after export, default %(default)s', 351 | default=False, 352 | action='store_true', 353 | dest='dup') 354 | #parser.add_argument('-d', '--dryrun', 355 | # help='do not update the account, default %(default)s', 356 | # default=True, action='store_true', dest='dryrun') 357 | parser.add_argument( 358 | '-V', 359 | '--verbose', 360 | help='print additional verbose information, default %(default)s', 361 | default=False, 362 | action='store_true', 363 | dest='verbose') 364 | options = parser.parse_args() 365 | 366 | ## Display debug information 367 | if options.verbose: 368 | print("Options: %s" % options) 369 | 370 | if options.userlist: 371 | options.list = "user list" 372 | 373 | if not options.output: 374 | options.output = 'export_{type}_{list}.csv'.format(type=options.type, 375 | list=options.list) 376 | 377 | ## Read configuration and validate 378 | helpers.read_config(_trakt, options) 379 | 380 | ## Try refreshing to get new access token. If it doesn't work, user needs to authenticate again. 381 | helpers.api_auth_refresh(_trakt, _headers, options) 382 | 383 | ## Display debug information 384 | if options.verbose: 385 | print("trakt: {}".format(_trakt)) 386 | print("Authorization header: {}".format(_headers['Authorization'])) 387 | 388 | export_data = [] 389 | ## Get Trakt user lists (custom lists) 390 | if options.userlist: 391 | export_data = helpers.api_get_userlists(_trakt, _headers, _proxyDict, 392 | options, 1) 393 | #print("export data") 394 | #print(export_data) 395 | if export_data: 396 | print("Found {0} user list(s)".format(len(export_data))) 397 | print("") 398 | #pp.pprint(export_data) 399 | # TODO: add export all user lists functionality 400 | print("id | name") 401 | for data in export_data: 402 | print("{id} | {name}".format(name=data['name'], 403 | id=data['ids']['trakt'])) 404 | #print("{id} | {name} | {items}".format( 405 | # name=data['name'], id=data['ids']['trakt'], items=data['item_count'], own=data['user']['username'])) 406 | print("") 407 | print( 408 | "Type in the id matching with the name of the list you want to export, or 'all' for all lists." 409 | ) 410 | options.listid = str(input('Input: ')) 411 | if options.listid == "all": 412 | for data in export_data: 413 | options.listid = data['ids']['trakt'] 414 | options.list = "{username}'s user list with id: {id}, name: '{name}'".format( 415 | username=data['user']['username'], 416 | id=data['ids']['trakt'], 417 | name=data['name']) 418 | global response_arr ## Cleanup global.... 419 | response_arr = [] 420 | export_data = helpers.api_get_userlist_items( 421 | _trakt, _headers, _proxyDict, options, 1) 422 | options.output = data['name'] + ".csv" 423 | process_export_data(options, export_data) 424 | else: 425 | response_arr = [] 426 | user_list = helpers.api_get_userlist(_trakt, _headers, 427 | _proxyDict, options, 1)[0] 428 | # print(user_list) 429 | options.list = "{username}'s user list with id: {id}, name: '{name}'".format( 430 | username=user_list['user']['username'], 431 | id=user_list['ids']['trakt'], 432 | name=user_list['name']) 433 | export_data = helpers.api_get_userlist_items( 434 | _trakt, _headers, _proxyDict, options, 1) 435 | #pp.pprint(export_data) 436 | process_export_data(options, export_data) 437 | else: 438 | print("Error, no user lists found".format(type=options.type, 439 | list=options.userlist)) 440 | sys.exit(1) 441 | else: 442 | export_data = helpers.api_get_list(_trakt, _headers, _proxyDict, 443 | options, 1) 444 | if export_data: 445 | process_export_data(options, export_data) 446 | else: 447 | print("Error, no item(s) found for {type} from {list}".format( 448 | type=options.type, list=options.list)) 449 | sys.exit(1) 450 | 451 | 452 | if __name__ == '__main__': 453 | main() 454 | -------------------------------------------------------------------------------- /helpers.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | # 4 | # (c) Copyright 2016-2020 xbgmsharp 5 | # 6 | # Purpose: 7 | # Helpers for 8 | # 9 | # Requirement on Ubuntu/Debian Linux system 10 | # apt-get install python3-dateutil python3-simplejson python3-requests python3-openssl jq 11 | # 12 | # Requirement on Windows on Python 3 13 | # C:\Python3\Scripts\easy_install3.exe simplejson requests 14 | # 15 | 16 | import sys, os 17 | # https://urllib3.readthedocs.org/en/latest/security.html#disabling-warnings 18 | # http://quabr.com/27981545/surpress-insecurerequestwarning-unverified-https-request-is-being-made-in-pytho 19 | # http://docs.python-requests.org/en/v2.4.3/user/advanced/#proxies 20 | try: 21 | import simplejson as json 22 | import requests 23 | requests.packages.urllib3.disable_warnings() 24 | import csv 25 | #from ratelimit import limits 26 | import time 27 | except: 28 | sys.exit( 29 | "Please use your favorite method to install the following module requests and simplejson to use this script" 30 | ) 31 | 32 | import argparse 33 | import configparser 34 | import datetime 35 | import collections 36 | import pprint 37 | 38 | pp = pprint.PrettyPrinter(indent=4) 39 | 40 | desc = """This program contains helpers for import_trakt.py and export_trakt.py""" 41 | 42 | epilog = """Read a list of ID from 'imdb', 'tmdb', 'tvdb' or 'tvrage' or 'trakt'. 43 | Import them into a list in Trakt.tv, mark as seen if need.""" 44 | 45 | _trakt = { 46 | 'client_id': '', # Auth details for trakt API 47 | 'client_secret': '', # Auth details for trakt API 48 | 'access_token': '', # Auth details for trakt API 49 | 'refresh_token': '', # Auth details for trakt API 50 | 'baseurl': 51 | 'https://api.trakt.tv', # Sandbox environment https://api-staging.trakt.tv, 52 | 'config_parser': '', # configparser.ConfigParser object 53 | 'config_path': '', # path of config file 54 | 'username': '' # username 55 | } 56 | 57 | _headers = { 58 | 'Accept': 'application/json', # required per API 59 | 'Content-Type': 'application/json', # required per API 60 | 'User-Agent': 'Tratk importer', # User-agent 61 | 'Connection': 62 | 'Keep-Alive', # Thanks to urllib3, keep-alive is 100% automatic within a session! 63 | 'trakt-api-version': '2', # required per API 64 | 'trakt-api-key': '', # required per API 65 | 'Authorization': '', # required per API 66 | } 67 | 68 | _proxy = { 69 | 'proxy': False, # True or False, trigger proxy use 70 | 'host': 'https://127.0.0.1', # Host/IP of the proxy 71 | 'port': '3128' # Port of the proxy 72 | } 73 | 74 | _proxyDict = { 75 | "http": _proxy['host'] + ':' + _proxy['port'], 76 | "https": _proxy['host'] + ':' + _proxy['port'] 77 | } 78 | 79 | # response_arr = [] 80 | 81 | 82 | def read_config(_trakt, options): 83 | """Read config file and if provided overwrite default values. 84 | If no config file exist, create one with default values. 85 | """ 86 | global work_dir 87 | work_dir = '' 88 | if getattr(sys, 'frozen', False): 89 | work_dir = os.path.dirname(sys.executable) 90 | elif __file__: 91 | work_dir = os.path.dirname(__file__) 92 | _configfile = os.path.join(work_dir, options.config) 93 | if os.path.exists(options.config): 94 | _configfile = options.config 95 | if options.verbose: 96 | print("Config file: {0}".format(_configfile)) 97 | # For recording configparser 98 | config = "" 99 | if os.path.exists(_configfile): 100 | try: 101 | config = configparser.ConfigParser() 102 | config.read(_configfile) 103 | if config.has_option('TRAKT', 'CLIENT_ID') and len( 104 | config.get('TRAKT', 'CLIENT_ID')) != 0: 105 | _trakt['client_id'] = config.get('TRAKT', 'CLIENT_ID') 106 | else: 107 | print('Error, you must specify a trakt.tv CLIENT_ID') 108 | sys.exit(1) 109 | if config.has_option('TRAKT', 'CLIENT_SECRET') and len( 110 | config.get('TRAKT', 'CLIENT_SECRET')) != 0: 111 | _trakt['client_secret'] = config.get('TRAKT', 'CLIENT_SECRET') 112 | else: 113 | print('Error, you must specify a trakt.tv CLIENT_SECRET') 114 | sys.exit(1) 115 | if config.has_option( 116 | 'TRAKT', 117 | 'USERNAME') and len(config.get('TRAKT', 'USERNAME')) != 0: 118 | _trakt['username'] = config.get('TRAKT', 'username') 119 | else: 120 | print( 121 | 'Error, you must specify a trakt.tv username for user lists to work' 122 | ) 123 | sys.exit(1) 124 | if config.has_option('TRAKT', 'ACCESS_TOKEN') and len( 125 | config.get('TRAKT', 'ACCESS_TOKEN')) != 0: 126 | _trakt['access_token'] = config.get('TRAKT', 'ACCESS_TOKEN') 127 | else: 128 | print( 129 | 'Warning, no access token found. Authentification is required' 130 | ) 131 | if config.has_option('TRAKT', 'REFRESH_TOKEN') and len( 132 | config.get('TRAKT', 'REFRESH_TOKEN')) != 0: 133 | _trakt['refresh_token'] = config.get('TRAKT', 'REFRESH_TOKEN') 134 | else: 135 | print( 136 | 'Warning, no refresh token found. Authentification is required' 137 | ) 138 | if config.has_option('TRAKT', 'BASEURL'): 139 | _trakt['baseurl'] = config.get('TRAKT', 'BASEURL') 140 | if config.has_option('SETTINGS', 'PROXY'): 141 | _proxy['proxy'] = config.getboolean('SETTINGS', 'PROXY') 142 | if _proxy['proxy'] and config.has_option( 143 | 'SETTINGS', 'PROXY_HOST') and config.has_option( 144 | 'SETTINGS', 'PROXY_PORT'): 145 | _proxy['host'] = config.get('SETTINGS', 'PROXY_HOST') 146 | _proxy['port'] = config.get('SETTINGS', 'PROXY_PORT') 147 | _proxyDict['http'] = _proxy['host'] + ':' + _proxy['port'] 148 | _proxyDict['https'] = _proxy['host'] + ':' + _proxy['port'] 149 | except: 150 | print("Error reading configuration file {0}".format(_configfile)) 151 | sys.exit(1) 152 | else: 153 | try: 154 | print('%s file was not found!' % _configfile) 155 | config = configparser.RawConfigParser() 156 | config.add_section('TRAKT') 157 | config.set('TRAKT', 'CLIENT_ID', '') 158 | config.set('TRAKT', 'CLIENT_SECRET', '') 159 | config.set('TRAKT', 'USERNAME', '') 160 | config.set('TRAKT', 'ACCESS_TOKEN', '') 161 | config.set('TRAKT', 'REFRESH_TOKEN', '') 162 | config.set('TRAKT', 'BASEURL', 'https://api.trakt.tv') 163 | config.add_section('SETTINGS') 164 | config.set('SETTINGS', 'PROXY', False) 165 | config.set('SETTINGS', 'PROXY_HOST', 'https://127.0.0.1') 166 | config.set('SETTINGS', 'PROXY_PORT', '3128') 167 | with open(_configfile, 'w') as configfile: 168 | config.write(configfile) 169 | print("Default settings wrote to file {0}".format(_configfile)) 170 | except: 171 | print("Error writing configuration file {0}".format(_configfile)) 172 | sys.exit(1) 173 | _trakt['config_parser'] = config 174 | _trakt['config_path'] = _configfile 175 | 176 | 177 | def write_csv(options, results): 178 | """Write list output into a CSV file format 179 | """ 180 | print("Writing to: {0}".format(options.output)) 181 | # Write result CSV, works with windows now 182 | with open(options.output, 'w', encoding='utf-8', newline='') as fp: 183 | mycsv = csv.DictWriter(fp, 184 | fieldnames=list(results[0].keys()), 185 | quoting=csv.QUOTE_ALL) 186 | mycsv.writeheader() 187 | for row in results: 188 | mycsv.writerow(row) 189 | fp.close() 190 | 191 | 192 | def api_auth_refresh(_trakt, _headers, options): 193 | ## Try refreshing to get new access token. If it doesn't work, user needs to authenticate again. 194 | if _trakt['refresh_token']: 195 | values = { 196 | "refresh_token": _trakt['refresh_token'], 197 | "client_id": _trakt['client_id'], 198 | "client_secret": _trakt["client_secret"], 199 | "redirect_uri": "urn:ietf:wg:oauth:2.0:oob", 200 | "grant_type": "refresh_token" 201 | } 202 | 203 | url = _trakt['baseurl'] + '/oauth/token' 204 | r = requests.post(url, data=values) 205 | print(r.status_code) 206 | if r.status_code == 200: 207 | response = r.json() 208 | _trakt['access_token'] = response["access_token"] 209 | _trakt['refresh_token'] = response["refresh_token"] 210 | _headers['Authorization'] = 'Bearer ' + response["access_token"] 211 | _headers['trakt-api-key'] = _trakt['client_id'] 212 | config = _trakt['config_parser'] 213 | config.set('TRAKT', 'ACCESS_TOKEN', response["access_token"]) 214 | config.set('TRAKT', 'REFRESH_TOKEN', response["refresh_token"]) 215 | with open(options.config, 'w') as configfile: 216 | config.write(configfile) 217 | print('Saved as "access_token" in file {0}: {1}'.format( 218 | options.config, response["access_token"])) 219 | print('Saved as "refresh_token" in file {0}: {1}'.format( 220 | options.config, response["refresh_token"])) 221 | else: 222 | print( 223 | "Refreshing access_token failed. Get new refresh_token and access_token by manually authenticating again" 224 | ) 225 | api_auth(_trakt, _headers, options) 226 | else: 227 | print( 228 | "No refresh_token found in config file. Get new refresh_token and access_token by manually authenticating again" 229 | ) 230 | api_auth(_trakt, _headers, options) 231 | 232 | 233 | def api_auth(_trakt, _headers, options): 234 | """API call for authentification OAUTH""" 235 | print( 236 | "Manual authentification. Open the link in a browser and paste the pincode when prompted" 237 | ) 238 | print(("https://trakt.tv/oauth/authorize?response_type=code&" 239 | "client_id={0}&redirect_uri=urn:ietf:wg:oauth:2.0:oob".format( 240 | _trakt["client_id"]))) 241 | pincode = str(input('Input:')) 242 | url = _trakt['baseurl'] + '/oauth/token' 243 | values = { 244 | "code": pincode, 245 | "client_id": _trakt["client_id"], 246 | "client_secret": _trakt["client_secret"], 247 | "redirect_uri": "urn:ietf:wg:oauth:2.0:oob", 248 | "grant_type": "authorization_code" 249 | } 250 | 251 | request = requests.post(url, data=values) 252 | response = request.json() 253 | _headers['Authorization'] = 'Bearer ' + response["access_token"] 254 | _headers['trakt-api-key'] = _trakt['client_id'] 255 | config = _trakt['config_parser'] 256 | config.set('TRAKT', 'ACCESS_TOKEN', response["access_token"]) 257 | config.set('TRAKT', 'REFRESH_TOKEN', response["refresh_token"]) 258 | with open(options.config, 'w') as configfile: 259 | config.write(configfile) 260 | print('Saved as "access_token" in file {0}: {1}'.format( 261 | options.config, response["access_token"])) 262 | print('Saved as "refresh_token" in file {0}: {1}'.format( 263 | options.config, response["refresh_token"])) 264 | 265 | 266 | def api_get_request(_headers, _proxyDict, options, url, page): 267 | """Uses Trakt API to sends a request to the URL given and returns the results as a response array, starting with page. 268 | """ 269 | response_arr = [] 270 | if options.verbose: 271 | print(url) 272 | if _proxy['proxy']: 273 | r = requests.get(url + "&page={0}".format(page), 274 | headers=_headers, 275 | proxies=_proxyDict, 276 | timeout=(10, 60)) 277 | else: 278 | r = requests.get(url + "&page={0}".format(page), headers=_headers, timeout=(5, 60)) 279 | 280 | if r.status_code != 200: 281 | print( 282 | "Error fetching GET response for {list}: {status} [{text}]".format( 283 | list=options.list, status=r.status_code, text=r.text)) 284 | return None 285 | else: 286 | json_dict = json.loads(r.text) 287 | if type(json_dict) is list: 288 | response_arr += json_dict 289 | else: 290 | response_arr.append(json_dict) 291 | 292 | # if 'X-Pagination-Page-Count'in r.headers and r.headers['X-Pagination-Page-Count'] != "0": 293 | if 'X-Pagination-Page-Count' in r.headers and r.headers[ 294 | 'X-Pagination-Page-Count']: 295 | if int(r.headers['X-Pagination-Page-Count']) == 0: 296 | print("No pages found in response, trakt list may be empty") 297 | return response_arr 298 | 299 | print("Fetched page {page} of {PageCount} pages for {list}".format( 300 | page=page, 301 | PageCount=r.headers['X-Pagination-Page-Count'], 302 | list=options.list)) 303 | if page != int(r.headers['X-Pagination-Page-Count']): 304 | response_arr += api_get_request(_headers, _proxyDict, options, url, page + 1) 305 | 306 | return response_arr 307 | 308 | 309 | def api_get_list(_trakt, _headers, _proxyDict, options, page): 310 | """Get items of default list (e.g history) by type starting from page 311 | """ 312 | url = _trakt[ 313 | 'baseurl'] + '/sync/{list}/{type}?limit={limit}'.format( 314 | list=options.list, type=options.type, page=page, limit=1000) 315 | return api_get_request(_headers, _proxyDict, options, url, page) 316 | 317 | 318 | def api_get_userlists(_trakt, _headers, _proxyDict, options, page): 319 | """Get list of all user lists 320 | """ 321 | url = _trakt['baseurl'] + '/users/{user}/lists?limit={limit}'.format( 322 | user=_trakt['username'], page=page, limit=1000) 323 | return api_get_request(_headers, _proxyDict, options, url, page) 324 | 325 | 326 | def api_get_userlist(_trakt, _headers, _proxyDict, options, page): 327 | """Get items of user list by type 328 | """ 329 | url = _trakt['baseurl'] + '/users/{user}/lists/{list_id}?limit={limit}'.format( 330 | user=_trakt['username'], 331 | list_id=options.listid, 332 | type=options.type, 333 | page=page, 334 | limit=1000) 335 | return api_get_request(_headers, _proxyDict, options, url, page) 336 | 337 | 338 | def api_get_userlist_items(_trakt, _headers, _proxyDict, options, page): 339 | """Get items of user list by type 340 | """ 341 | url = _trakt[ 342 | 'baseurl'] + '/users/{user}/lists/{list_id}/items/{type}&limit={limit}'.format( 343 | user=_trakt['username'], 344 | list_id=options.listid, 345 | type=options.type, 346 | page=page, 347 | limit=1000) 348 | return api_get_request(_headers, _proxyDict, options, url, page) 349 | -------------------------------------------------------------------------------- /import_trakt.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | # 4 | # (c) Copyright 2016-2020 xbgmsharp 5 | # 6 | # Purpose: 7 | # Import Movies or TVShows IDs into Trakt.tv 8 | # 9 | # Requirement on Ubuntu/Debian Linux system 10 | # apt-get install python3-dateutil python3-simplejson python3-requests python3-openssl jq 11 | # 12 | # Requirement on Windows on Python 3 13 | # C:\Python3\Scripts\easy_install3.exe simplejson requests 14 | # 15 | 16 | import sys, os 17 | # https://urllib3.readthedocs.org/en/latest/security.html#disabling-warnings 18 | # http://quabr.com/27981545/surpress-insecurerequestwarning-unverified-https-request-is-being-made-in-pytho 19 | # http://docs.python-requests.org/en/v2.4.3/user/advanced/#proxies 20 | try: 21 | import simplejson as json 22 | import requests 23 | requests.packages.urllib3.disable_warnings() 24 | import csv 25 | #from ratelimit import limits 26 | import time 27 | except: 28 | sys.exit( 29 | "Please use your favorite method to install the following module requests and simplejson to use this script" 30 | ) 31 | 32 | import argparse 33 | import configparser 34 | from datetime import datetime 35 | import collections 36 | import pprint 37 | 38 | import helpers 39 | 40 | pp = pprint.PrettyPrinter(indent=4) 41 | 42 | desc = """This program import Movies or TVShows IDs into Trakt.tv.""" 43 | 44 | epilog = """Read a list of ID from 'imdb', 'tmdb', 'tvdb' or 'tvrage' or 'trakt'. 45 | Import them into a list in Trakt.tv, mark as seen if need.""" 46 | 47 | _trakt = { 48 | 'client_id': '', # Auth details for trakt API 49 | 'client_secret': '', # Auth details for trakt API 50 | 'access_token': '', # Auth details for trakt API 51 | 'refresh_token': '', # Auth details for trakt API 52 | 'baseurl': 53 | 'https://api.trakt.tv', # Sandbox environment https://api-staging.trakt.tv, 54 | 'config_parser': '', # configparser.ConfigParser object 55 | 'config_path': '', # path of config file 56 | 'username': '' # username 57 | } 58 | 59 | _headers = { 60 | 'Accept': 'application/json', # required per API 61 | 'Content-Type': 'application/json', # required per API 62 | 'User-Agent': 'Tratk importer', # User-agent 63 | 'Connection': 64 | 'Keep-Alive', # Thanks to urllib3, keep-alive is 100% automatic within a session! 65 | 'trakt-api-version': '2', # required per API 66 | 'trakt-api-key': '', # required per API 67 | 'Authorization': '', # required per API 68 | } 69 | 70 | _proxy = { 71 | 'proxy': False, # True or False, trigger proxy use 72 | 'host': 'https://127.0.0.1', # Host/IP of the proxy 73 | 'port': '3128' # Port of the proxy 74 | } 75 | 76 | _proxyDict = { 77 | "http": _proxy['host'] + ':' + _proxy['port'], 78 | "https": _proxy['host'] + ':' + _proxy['port'] 79 | } 80 | 81 | response_arr = [] 82 | 83 | 84 | def read_csv(options): 85 | """Read CSV of Movies or TVShows IDs and return a dict""" 86 | reader = csv.DictReader(options.input, delimiter=',') 87 | print("Done reading " + _trakt['config_path']) 88 | return list(reader) 89 | 90 | 91 | # def api_search_by_id(options, id): 92 | # """API call for Search / ID Lookup / Get ID lookup results""" 93 | # url = _trakt['baseurl'] + '/search?id_type={0}&id={1}'.format(options.format, id) 94 | # if options.verbose: 95 | # print(url) 96 | # if _proxy['proxy']: 97 | # r = requests.get(url, headers=_headers, proxies=_proxyDict, timeout=(10, 60)) 98 | # else: 99 | # r = requests.get(url, headers=_headers, timeout=(5, 60)) 100 | # if r.status_code != 200: 101 | # print("Error Get ID lookup results: {0} [{1}]".format(r.status_code, r.text)) 102 | # return None 103 | # else: 104 | # return json.loads(r.text) 105 | 106 | 107 | # @limits(calls=1, period=1) 108 | def api_add_to_list(options, import_data): 109 | """API call for Sync / Add items to list""" 110 | 111 | # Rate limit for API 112 | time.sleep(1) 113 | if options.userlist: 114 | url = _trakt[ 115 | 'baseurl'] + '/users/{username}/lists/{list_id}/items'.format( 116 | username=_trakt['username'], list_id=options.listid) 117 | else: 118 | url = _trakt['baseurl'] + '/sync/{list}'.format(list=options.list) 119 | #values = '{ "movies": [ { "ids": { "imdb": "tt0000111" } }, { "ids": { , "imdb": "tt1502712" } } ] }' 120 | #values = '{ "movies": [ { "watched_at": "2014-01-01T00:00:00.000Z", "ids": { "imdb": "tt0000111" } }, { "watched_at": "2013-01-01T00:00:00.000Z", "ids": { "imdb": "tt1502712" } } ] }' 121 | if options.type == 'episodes': 122 | values = {'episodes': import_data} 123 | else: 124 | values = {options.type: import_data} 125 | 126 | json_data = json.dumps(values) 127 | if options.verbose: 128 | print("Sending to URL: {0}".format(url)) 129 | pp.pprint(json_data) 130 | 131 | if _proxy['proxy']: 132 | #print(url) 133 | r = requests.post(url, 134 | data=json_data, 135 | headers=_headers, 136 | proxies=_proxyDict, 137 | timeout=(10, 60)) 138 | else: 139 | r = requests.post(url, 140 | data=json_data, 141 | headers=_headers, 142 | timeout=(5, 60)) 143 | 144 | if r.status_code != 201: 145 | print("Error Adding items to {list}: {status} [{text}]".format( 146 | list=options.list, status=r.status_code, text=r.text)) 147 | return None 148 | else: 149 | return json.loads(r.text) 150 | 151 | 152 | # TODO: Not sure if removing from list works 153 | def api_remove_from_list(options, remove_data): 154 | """API call for Sync / Remove from list""" 155 | url = _trakt['baseurl'] + '/sync/{list}/remove'.format(list=options.list) 156 | if options.type == 'episodes': 157 | values = {'shows': remove_data} 158 | else: 159 | values = {options.type: remove_data} 160 | json_data = json.dumps(values) 161 | if options.verbose: 162 | print(url) 163 | pp.pprint(json_data) 164 | if _proxy['proxy']: 165 | r = requests.post(url, 166 | data=json_data, 167 | headers=_headers, 168 | proxies=_proxyDict, 169 | timeout=(10, 60)) 170 | else: 171 | r = requests.post(url, 172 | data=json_data, 173 | headers=_headers, 174 | timeout=(5, 60)) 175 | if r.status_code != 200: 176 | print("Error removing items from {list}: {status} [{text}]".format( 177 | list=options.list, status=r.status_code, text=r.text)) 178 | return None 179 | else: 180 | return json.loads(r.text) 181 | 182 | 183 | # TODO: Not sure if cleaning up list works 184 | def cleanup_list(options): 185 | """Empty list prior to import""" 186 | if options.userlist: 187 | export_data = helpers.api_get_userlist(_trakt, _headers, _proxyDict, 188 | options, 1) 189 | else: 190 | export_data = helpers.api_get_list(_trakt, _headers, _proxyDict, 191 | options, 1) 192 | print(export_data) 193 | if export_data: 194 | print("Found {0} Item-Count".format(len(export_data))) 195 | else: 196 | print("Error, Cleanup no item return for {type} from the {list} list". 197 | format(type=options.type, list=options.list)) 198 | sys.exit(1) 199 | results = {'sentids': 0, 'deleted': 0, 'not_found': 0} 200 | to_remove = [] 201 | for data in export_data: 202 | to_remove.append({'ids': data[options.type[:-1]]['ids']}) 203 | if len(to_remove) >= 10: 204 | results['sentids'] += len(to_remove) 205 | result = api_remove_from_list(options, to_remove) 206 | if result: 207 | print("Result: {0}".format(result)) 208 | if 'deleted' in result and result['deleted']: 209 | results['deleted'] += result['deleted'][options.type] 210 | if 'not_found' in result and result['not_found']: 211 | results['not_found'] += len( 212 | result['not_found'][options.type]) 213 | to_remove = [] 214 | # Remove the rest 215 | if len(to_remove) > 0: 216 | #pp.pprint(data) 217 | results['sentids'] += len(to_remove) 218 | result = api_remove_from_list(options, to_remove) 219 | if result: 220 | print("Result: {0}".format(result)) 221 | if 'deleted' in result and result['deleted']: 222 | results['deleted'] += result['deleted'][options.type] 223 | if 'not_found' in result and result['not_found']: 224 | results['not_found'] += len(result['not_found'][options.type]) 225 | print( 226 | "Overall cleanup {sent} {type}, results deleted:{deleted}, not_found:{not_found}" 227 | .format(sent=results['sentids'], 228 | type=options.type, 229 | deleted=results['deleted'], 230 | not_found=results['not_found'])) 231 | 232 | 233 | def main(): 234 | """ 235 | Main program loop 236 | * Read configuration file and validate 237 | * Read CSV file 238 | * Authenticate if require 239 | * Cleanup list from Trakt.tv 240 | * Inject data into Trakt.tv 241 | """ 242 | # Parse inputs if any 243 | parser = argparse.ArgumentParser(description=desc, epilog=epilog) 244 | list_group = parser.add_mutually_exclusive_group(required=True) 245 | time_group = parser.add_mutually_exclusive_group(required=False) 246 | parser.add_argument('-v', action='version', version='%(prog)s 0.1') 247 | parser.add_argument( 248 | '-c', 249 | '--config', 250 | help='allow to overwrite default config filename, default %(default)s', 251 | action='store', 252 | type=str, 253 | dest='config', 254 | default='config.ini') 255 | parser.add_argument('-i', 256 | '--input', 257 | help='CSV file to import, default %(default)s', 258 | nargs='?', 259 | type=argparse.FileType('r'), 260 | default=None, 261 | required=True) 262 | time_group.add_argument( 263 | '-w', 264 | '--watched_at', 265 | help= 266 | 'import watched_at date from CSV, the format must be UTC datetime. NOTE: Only works with history, not with watchlist/userlist. default %(default)s', 267 | default=False, 268 | action='store_true', 269 | dest='watched_at') 270 | now = datetime.now() 271 | time_group.add_argument( 272 | '-s', 273 | '--seen', 274 | help= 275 | 'use custom time for watched_at if importing to history, default %(default)s. NOTE: Only works with history, not with watchlist/userlist. Use specific time if provided, default is current time.', 276 | nargs='?', 277 | const=now.strftime('%Y-%m-%dT%H:%M:%S.000Z'), 278 | action='store', 279 | type=str, 280 | dest='seen', 281 | default=False) 282 | parser.add_argument( 283 | '-f', 284 | '--format', 285 | help='allow to overwrite default ID type format, default %(default)s', 286 | choices=['imdb', 'tmdb', 'tvdb', 'tvrage', 'trakt'], 287 | dest='format', 288 | default='trakt') 289 | parser.add_argument('-t', 290 | '--type', 291 | help='allow to overwrite type, default %(default)s', 292 | choices=['movies', 'shows', 'episodes'], 293 | dest='type', 294 | default='movies') 295 | list_group.add_argument( 296 | '-l', 297 | '--list', 298 | help='allow to overwrite default list, default %(default)s', 299 | choices=['watchlist', 'collection', 'history'], 300 | dest='list', 301 | default='watchlist') 302 | list_group.add_argument( 303 | '-u', 304 | '--userlist', 305 | help='allow to add item(s) to a user custom list, default %(default)s', 306 | dest='userlist', 307 | default=False, 308 | action='store_true') 309 | parser.add_argument('-C', 310 | '--clean', 311 | help='empty list prior to import, default %(default)s', 312 | default=False, 313 | action='store_true', 314 | dest='clean') 315 | #parser.add_argument('-d', '--dryrun', 316 | # help='do not update the account, default %(default)s', 317 | # default=True, action='store_true', dest='dryrun') 318 | parser.add_argument( 319 | '-V', 320 | '--verbose', 321 | help='print additional verbose information, default %(default)s', 322 | default=False, 323 | action='store_true', 324 | dest='verbose') 325 | options = parser.parse_args() 326 | 327 | # Display debug information 328 | if options.verbose: 329 | print("Options: %s" % options) 330 | 331 | if options.seen and options.list != "history": 332 | print( 333 | "Error, you can only mark seen {0} when adding into the history list" 334 | .format(options.type)) 335 | sys.exit(1) 336 | 337 | if options.seen: 338 | try: 339 | datetime.strptime(options.seen, '%Y-%m-%dT%H:%M:%S.000Z') 340 | except: 341 | sys.exit( 342 | "Error, invalid format, it's must be UTC datetime, eg: '2016-01-01T00:00:00.000Z'" 343 | ) 344 | 345 | ## Read configuration and validate 346 | helpers.read_config(_trakt, options) 347 | 348 | ## Try refreshing to get new access token. If it doesn't work, user needs to authenticate again. 349 | helpers.api_auth_refresh(_trakt, _headers, options) 350 | 351 | # Display debug information 352 | if options.verbose: 353 | print("API Trakt: {}".format(_trakt)) 354 | print("Authorization header: {}".format(_headers['Authorization'])) 355 | 356 | # Handle userlist 357 | if options.userlist: 358 | print(_trakt['access_token']) 359 | export_data = helpers.api_get_userlists(_trakt, _headers, _proxyDict, 360 | options, 1) 361 | if export_data: 362 | print("") 363 | print("Found {0} user list(s)".format(len(export_data))) 364 | print("") 365 | #pp.pprint(export_data) 366 | print("id | name") 367 | for data in export_data: 368 | print("{id} | {name}".format(name=data['name'], 369 | id=data['ids']['trakt'])) 370 | #print("{id} | {name} | {items}".format( 371 | # name=data['name'], id=data['ids']['trakt'], items=data['item_count'], own=data['user']['username'])) 372 | print("") 373 | print( 374 | "Type in the id matching with the name of the list you want to import item(s) to." 375 | ) 376 | options.listid = str(input('Input: ')) 377 | print( 378 | "Importing to {username}'s user list with id: {id}, name: '{name}'" 379 | .format(username=data['user']['username'], 380 | id=data['ids']['trakt'], 381 | name=data['name'])) 382 | response_arr = [] 383 | else: 384 | print("Error, no user lists found") 385 | sys.exit(1) 386 | # else: 387 | # export_data = helpers.api_get_list(_trakt, _headers, _proxyDict, options, 1) 388 | # if not export_data: 389 | # print("Error, no item(s) found for {type} from {list}".format( 390 | # type=options.type, list=options.list)) 391 | # sys.exit(1) 392 | 393 | # Empty list prior to import 394 | if options.clean: 395 | cleanup_list(options) 396 | 397 | # Read CSV list of IDs 398 | read_ids = read_csv(options) 399 | 400 | # if IDs make the list into trakt format 401 | data = [] 402 | results = {'sentids': 0, 'added': 0, 'existing': 0, 'not_found': 0} 403 | 404 | if options.list == 'history': 405 | options.time_key = 'watched_at' 406 | elif options.list == 'watchlist': 407 | options.time_key = 'listed_at' 408 | elif options.list == 'collection': 409 | options.time_key = 'collected_at' 410 | elif options.userlist != None: 411 | options.time_key = 'listed_at' 412 | 413 | if read_ids: 414 | print("Found {0} items to import".format(len(read_ids))) 415 | 416 | for myid in read_ids: 417 | # If id (row) exists and is not blank (has a format) 418 | if myid and myid[options.format]: 419 | # Record time format in csv we're importing from. 420 | # NOTE: Trakt API does not allow for custom times for listed_at and collected_at. 421 | # Therefore, options.time_key doesn't do anything for lists other than history. 422 | # However, this allows for any type of list to be imported in to default lists. 423 | if 'watched_at' in myid: 424 | options.csv_time = 'watched_at' 425 | elif 'listed_at' in myid: 426 | options.csv_time = 'listed_at' 427 | elif 'collected_at' in myid: 428 | options.csv_time = 'collected_at' 429 | else: 430 | options.csv_time = 'listed_at' 431 | if options.verbose: 432 | pp.pprint(myid) 433 | row_title = "No title in csv" 434 | if options.seen: 435 | row_time = options.seen 436 | elif options.watched_at: 437 | row_time = myid[options.csv_time] 438 | else: 439 | row_time = "No time option" 440 | 441 | # If format is not "imdb" it must be cast to an integer 442 | if not options.format == "imdb" and not myid[ 443 | options.format].startswith('tt'): 444 | myid[options.format] = int(myid[options.format]) 445 | if 'title' in myid: 446 | row_title = "title: " + myid['title'] 447 | if 'show_title' in myid and 'episode_title' in myid: 448 | row_title = "title: " + myid['show_title'] + ", episode_title: " + myid['episode_title'] 449 | 450 | if (options.type == "movies" or options.type == "shows" or options.type == "episodes"): 451 | data.append({ 452 | 'ids': { 453 | options.format: myid[options.format] 454 | }, 455 | options.time_key: row_time}) 456 | else: 457 | data.append( 458 | {'ids': { 459 | options.format: myid[options.format] 460 | }}) 461 | print( 462 | "Importing record, {title}, id: {id}, {csv_time}: {time}". 463 | format(title=row_title, 464 | id=myid[options.format], 465 | csv_time=options.csv_time, 466 | time=row_time)) 467 | # Import batch of 10 IDs 468 | if len(data) >= 10: 469 | #pp.pprint(json.dumps(data)) 470 | results['sentids'] += len(data) 471 | result = api_add_to_list(options, data) 472 | if result: 473 | # print("Result: {0}".format(result)) 474 | if 'added' in result and result['added']: 475 | results['added'] += result['added'][options.type] 476 | if 'existing' in result and result['existing']: 477 | results['existing'] += result['existing'][ 478 | options.type] 479 | if 'not_found' in result and result['not_found']: 480 | results['not_found'] += len( 481 | result['not_found'][options.type]) 482 | data = [] 483 | # Import the rest 484 | if len(data) > 0: 485 | #pp.pprint(data) 486 | results['sentids'] += len(data) 487 | result = api_add_to_list(options, data) 488 | if result: 489 | # pp.pprint("Result: {0}".format(result)) 490 | if 'added' in result and result['added']: 491 | results['added'] += result['added'][options.type] 492 | if 'existing' in result and result['existing']: 493 | results['existing'] += result['existing'][options.type] 494 | if 'not_found' in result and result['not_found']: 495 | results['not_found'] += len( 496 | result['not_found'][options.type]) 497 | else: 498 | # TODO: Read STDIN to ID 499 | print("No items found, nothing to do.") 500 | sys.exit(0) 501 | 502 | print( 503 | "Overall imported {sent} {type}, results added:{added}, existing:{existing}, not_found:{not_found}" 504 | .format(sent=results['sentids'], 505 | type=options.type, 506 | added=results['added'], 507 | existing=results['existing'], 508 | not_found=results['not_found'])) 509 | 510 | 511 | if __name__ == '__main__': 512 | main() 513 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | simplejson==3.17.2 2 | requests==2.24.0 3 | --------------------------------------------------------------------------------