-
Notifications
You must be signed in to change notification settings - Fork 45
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
1,111 additions
and
913 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,11 +1,11 @@ | ||
| sudo: false | ||
| language: python | ||
| python: | ||
| - '3.6' | ||
| - '3.6' | ||
| install: | ||
| - pip install -U flake8 pipenv setuptools wheel pip | ||
| - pip install -r requirements.txt | ||
| - pip install -U flake8 pipenv setuptools wheel pip | ||
| - pip install -r requirements.txt | ||
| before_script: | ||
| - flake8 --ignore=W391 progressbar tests | ||
| - flake8 --ignore=W391 . | ||
| script: | ||
| - python PlexAniSync.py | ||
| - python PlexAniSync.py |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,165 +1,162 @@ | ||
| # coding=utf-8 | ||
| import collections | ||
| import configparser | ||
| import coloredlogs | ||
| import json | ||
| import locale | ||
| import logging | ||
| import logging.handlers | ||
| import os | ||
| import re | ||
| import requests | ||
| import sys | ||
| from guessit import guessit | ||
| from plexapi.myplex import MyPlexAccount | ||
| from plexapi.server import PlexServer | ||
|
|
||
| import anilist | ||
| import plexmodule | ||
|
|
||
| __version__ = "1.2.6" | ||
|
|
||
| # Logger settings | ||
| log_filename = 'PlexAniSync.log' | ||
| logger = logging.getLogger('PlexAniSync') | ||
|
|
||
| # Add the rotating log message handler to the standard log | ||
| handler = logging.handlers.RotatingFileHandler(log_filename, maxBytes=10000000, backupCount=5,encoding='utf-8') | ||
| handler.setLevel(logging.INFO) | ||
| logger.addHandler(handler) | ||
|
|
||
| # Debug log | ||
| logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(levelname)s %(message)s', handlers=[logging.FileHandler('PlexAniSync-DEBUG.log', 'w', 'utf-8')]) | ||
|
|
||
| # Install colored logs | ||
| coloredlogs.install(fmt='%(asctime)s %(message)s', logger=logger) | ||
|
|
||
|
|
||
| # Enable this if you want to also log all messages coming from imported libraries | ||
| # coloredlogs.install(level='DEBUG') | ||
|
|
||
| ## Settings section ## | ||
|
|
||
| def read_settings(settings_file): | ||
| if not os.path.isfile(settings_file): | ||
| logger.critical( | ||
| '[CONFIG] Settings file file not found: %s' % (settings_file)) | ||
| sys.exit() | ||
| settings = configparser.ConfigParser() | ||
| settings.read(settings_file) | ||
| return settings | ||
|
|
||
|
|
||
| if len(sys.argv) > 1: | ||
| settings_file = sys.argv[1] | ||
| logger.warning( | ||
| 'Found settings file parameter and using: %s' % | ||
| (settings_file)) | ||
| else: | ||
| settings_file = 'settings.ini' | ||
|
|
||
| settings = read_settings(settings_file) | ||
| anilist_settings = settings['ANILIST'] | ||
| plex_settings = settings['PLEX'] | ||
|
|
||
| ANILIST_SKIP_UPDATE = anilist_settings['skip_list_update'].lower() | ||
| ANILIST_ACCESS_TOKEN = anilist_settings['access_token'].strip() | ||
|
|
||
| if 'plex_episode_count_priority' in anilist_settings: | ||
| ANILIST_PLEX_EPISODE_COUNT_PRIORITY = anilist_settings['plex_episode_count_priority'].lower().strip() | ||
| else: | ||
| ANILIST_PLEX_EPISODE_COUNT_PRIORITY = 'false' | ||
|
|
||
| mapping_file = 'custom_mappings.ini' | ||
| custom_mappings = [] | ||
|
|
||
|
|
||
| def read_custom_mappings(mapping_file): | ||
| if not os.path.isfile(mapping_file): | ||
| logger.info( | ||
| '[MAPPING] Custom map file not found: %s' % (mapping_file)) | ||
| else: | ||
| logger.info('[MAPPING] Custom map file found: %s' % (mapping_file)) | ||
| file = open(mapping_file, "r") | ||
| for line in file: | ||
| try: | ||
| mappingSplit = line.split('^') | ||
| series_title = mappingSplit[0] | ||
| season = mappingSplit[1] | ||
| anime_id = int(mappingSplit[2]) | ||
|
|
||
| logger.info( | ||
| "[MAPPING] Adding custom mapping | title: %s | season: %s | anilist id: %s" % | ||
| (series_title, season, anime_id)) | ||
| mapping = anilist.anilist_custom_mapping( | ||
| series_title, season, anime_id) | ||
| custom_mappings.append(mapping) | ||
| except BaseException: | ||
| logger.error( | ||
| '[MAPPING] Invalid entry found for line: %s' % | ||
| (line)) | ||
|
|
||
| ## Startup section ## | ||
|
|
||
|
|
||
| def start(): | ||
| logger.info( | ||
| "PlexAniSync - version: %s" % (__version__)) | ||
|
|
||
| if ANILIST_SKIP_UPDATE == 'true': | ||
| logger.warning( | ||
| 'AniList skip list update enabled in settings, will match but NOT update your list') | ||
|
|
||
| if ANILIST_PLEX_EPISODE_COUNT_PRIORITY == 'true': | ||
| logger.warning( | ||
| 'Plex episode watched count will take priority over AniList, this will always update AniList watched count over Plex data') | ||
|
|
||
| # Cleanup any old logs | ||
| exists = os.path.isfile("failed_matches.txt") | ||
| if exists: | ||
| try: | ||
| os.remove("failed_matches.txt") | ||
| except BaseException: | ||
| pass | ||
|
|
||
| # Anilist | ||
| anilist_username = anilist_settings['username'] | ||
| anilist.custom_mappings = custom_mappings | ||
| anilist.ANILIST_ACCESS_TOKEN = ANILIST_ACCESS_TOKEN | ||
| anilist.ANILIST_PLEX_EPISODE_COUNT_PRIORITY = ANILIST_PLEX_EPISODE_COUNT_PRIORITY | ||
| anilist.ANILIST_SKIP_UPDATE = ANILIST_SKIP_UPDATE | ||
| anilist_series = anilist.process_user_list(anilist_username) | ||
|
|
||
| # Plex | ||
| if anilist_series is None: | ||
| logger.error( | ||
| 'Unable to retrieve AniList list, check your username and access token') | ||
| else: | ||
| if not anilist_series: | ||
| logger.error('No items found on your AniList list for additional processing later on') | ||
|
|
||
| plexmodule.plex_settings = plex_settings | ||
| plex_anime_series = plexmodule.get_anime_shows() | ||
|
|
||
| if(plex_anime_series is None): | ||
| logger.error('Found no Plex shows for processing') | ||
| plex_series_watched = None | ||
| else: | ||
| plex_series_watched = plexmodule.get_watched_shows( | ||
| plex_anime_series) | ||
|
|
||
| if(plex_series_watched is None): | ||
| logger.error( | ||
| 'Found no watched shows on Plex for processing') | ||
| else: | ||
| anilist.match_to_plex( | ||
| anilist_series, | ||
| plex_anime_series, | ||
| plex_series_watched) | ||
|
|
||
| logger.info('Plex to AniList sync finished') | ||
|
|
||
|
|
||
| if __name__ == '__main__': | ||
| read_custom_mappings(mapping_file) | ||
| start() | ||
| # coding=utf-8 | ||
| import configparser | ||
| import logging.handlers | ||
| import os | ||
| import sys | ||
|
|
||
| import coloredlogs | ||
|
|
||
| import anilist | ||
| import plexmodule | ||
|
|
||
| __version__ = "1.2.6" | ||
|
|
||
| # Logger settings | ||
| log_filename = "PlexAniSync.log" | ||
| logger = logging.getLogger("PlexAniSync") | ||
|
|
||
| # Add the rotating log message handler to the standard log | ||
| handler = logging.handlers.RotatingFileHandler( | ||
| log_filename, maxBytes=10000000, backupCount=5, encoding="utf-8" | ||
| ) | ||
| handler.setLevel(logging.INFO) | ||
| logger.addHandler(handler) | ||
|
|
||
| # Debug log | ||
| logging.basicConfig( | ||
| level=logging.DEBUG, | ||
| format="%(asctime)s %(levelname)s %(message)s", | ||
| handlers=[logging.FileHandler("PlexAniSync-DEBUG.log", "w", "utf-8")], | ||
| ) | ||
|
|
||
| # Install colored logs | ||
| coloredlogs.install(fmt="%(asctime)s %(message)s", logger=logger) | ||
|
|
||
|
|
||
| # Enable this if you want to also log all messages coming from imported libraries | ||
| # coloredlogs.install(level='DEBUG') | ||
|
|
||
| ## Settings section ## | ||
|
|
||
|
|
||
| def read_settings(settings_file): | ||
| if not os.path.isfile(settings_file): | ||
| logger.critical("[CONFIG] Settings file file not found: %s" % (settings_file)) | ||
| sys.exit() | ||
| settings = configparser.ConfigParser() | ||
| settings.read(settings_file) | ||
| return settings | ||
|
|
||
|
|
||
| if len(sys.argv) > 1: | ||
| settings_file = sys.argv[1] | ||
| logger.warning("Found settings file parameter and using: %s" % (settings_file)) | ||
| else: | ||
| settings_file = "settings.ini" | ||
|
|
||
| settings = read_settings(settings_file) | ||
| anilist_settings = settings["ANILIST"] | ||
| plex_settings = settings["PLEX"] | ||
|
|
||
| ANILIST_SKIP_UPDATE = anilist_settings["skip_list_update"].lower() | ||
| ANILIST_ACCESS_TOKEN = anilist_settings["access_token"].strip() | ||
|
|
||
| if "plex_episode_count_priority" in anilist_settings: | ||
| ANILIST_PLEX_EPISODE_COUNT_PRIORITY = ( | ||
| anilist_settings["plex_episode_count_priority"].lower().strip() | ||
| ) | ||
| else: | ||
| ANILIST_PLEX_EPISODE_COUNT_PRIORITY = "false" | ||
|
|
||
| mapping_file = "custom_mappings.ini" | ||
| custom_mappings = [] | ||
|
|
||
|
|
||
| def read_custom_mappings(mapping_file): | ||
| if not os.path.isfile(mapping_file): | ||
| logger.info("[MAPPING] Custom map file not found: %s" % (mapping_file)) | ||
| else: | ||
| logger.info("[MAPPING] Custom map file found: %s" % (mapping_file)) | ||
| file = open(mapping_file, "r") | ||
| for line in file: | ||
| try: | ||
| mappingSplit = line.split("^") | ||
| series_title = mappingSplit[0] | ||
| season = mappingSplit[1] | ||
| anime_id = int(mappingSplit[2]) | ||
|
|
||
| logger.info( | ||
| "[MAPPING] Adding custom mapping | title: %s | season: %s | anilist id: %s" | ||
| % (series_title, season, anime_id) | ||
| ) | ||
| mapping = anilist.anilist_custom_mapping(series_title, season, anime_id) | ||
| custom_mappings.append(mapping) | ||
| except BaseException: | ||
| logger.error("[MAPPING] Invalid entry found for line: %s" % (line)) | ||
|
|
||
|
|
||
| ## Startup section ## | ||
|
|
||
|
|
||
| def start(): | ||
| logger.info("PlexAniSync - version: %s" % (__version__)) | ||
|
|
||
| if ANILIST_SKIP_UPDATE == "true": | ||
| logger.warning( | ||
| "AniList skip list update enabled in settings, will match but NOT update your list" | ||
| ) | ||
|
|
||
| if ANILIST_PLEX_EPISODE_COUNT_PRIORITY == "true": | ||
| logger.warning( | ||
| "Plex episode watched count will take priority over AniList, this will always update AniList watched count over Plex data" | ||
| ) | ||
|
|
||
| # Cleanup any old logs | ||
| exists = os.path.isfile("failed_matches.txt") | ||
| if exists: | ||
| try: | ||
| os.remove("failed_matches.txt") | ||
| except BaseException: | ||
| pass | ||
|
|
||
| # Anilist | ||
| anilist_username = anilist_settings["username"] | ||
| anilist.custom_mappings = custom_mappings | ||
| anilist.ANILIST_ACCESS_TOKEN = ANILIST_ACCESS_TOKEN | ||
| anilist.ANILIST_PLEX_EPISODE_COUNT_PRIORITY = ANILIST_PLEX_EPISODE_COUNT_PRIORITY | ||
| anilist.ANILIST_SKIP_UPDATE = ANILIST_SKIP_UPDATE | ||
| anilist_series = anilist.process_user_list(anilist_username) | ||
|
|
||
| # Plex | ||
| if anilist_series is None: | ||
| logger.error( | ||
| "Unable to retrieve AniList list, check your username and access token" | ||
| ) | ||
| else: | ||
| if not anilist_series: | ||
| logger.error( | ||
| "No items found on your AniList list for additional processing later on" | ||
| ) | ||
|
|
||
| plexmodule.plex_settings = plex_settings | ||
| plex_anime_series = plexmodule.get_anime_shows() | ||
|
|
||
| if plex_anime_series is None: | ||
| logger.error("Found no Plex shows for processing") | ||
| plex_series_watched = None | ||
| else: | ||
| plex_series_watched = plexmodule.get_watched_shows(plex_anime_series) | ||
|
|
||
| if plex_series_watched is None: | ||
| logger.error("Found no watched shows on Plex for processing") | ||
| else: | ||
| anilist.match_to_plex( | ||
| anilist_series, plex_anime_series, plex_series_watched | ||
| ) | ||
|
|
||
| logger.info("Plex to AniList sync finished") | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| read_custom_mappings(mapping_file) | ||
| start() |
Oops, something went wrong.