forked from RickDB/PlexAniSync
-
Notifications
You must be signed in to change notification settings - Fork 0
/
PlexAniSync.py
106 lines (79 loc) · 3.13 KB
/
PlexAniSync.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
# coding=utf-8
import configparser
import logging
from logging.handlers import RotatingFileHandler
import os
import sys
import coloredlogs
from plexanisync.anilist import Anilist
from plexanisync.custom_mappings import read_custom_mappings
from plexanisync.plexmodule import PlexModule
from plexanisync._version import __version__
# Logger settings
LOG_FILENAME = "PlexAniSync.log"
logger = logging.getLogger("PlexAniSync")
# Add the rotating log message handler to the standard log
handler = RotatingFileHandler(
LOG_FILENAME, maxBytes=10_000_000, backupCount=5, encoding="utf-8"
)
handler.setLevel(logging.INFO)
logger.addHandler(handler)
# Debug log
logging.basicConfig(
level=logging.DEBUG,
format="%(asctime)s %(levelname)s %(name)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) -> configparser.ConfigParser:
if not os.path.isfile(settings_file):
logger.critical(f"[CONFIG] Settings file file not found: {settings_file}")
sys.exit(1)
settings = configparser.ConfigParser()
settings.read(settings_file, encoding="utf-8")
return settings
SETTINGS_FILE = os.getenv("SETTINGS_FILE") or "settings.ini"
if len(sys.argv) > 1:
SETTINGS_FILE = sys.argv[1]
logger.warning(f"Found settings file parameter and using: {SETTINGS_FILE}")
settings = read_settings(SETTINGS_FILE)
anilist_settings = settings["ANILIST"]
plex_settings = settings["PLEX"]
## Startup section ##
def start():
logger.info(f"PlexAniSync - version: {__version__}")
custom_mappings = read_custom_mappings()
if anilist_settings.getboolean("skip_list_update", False):
logger.warning(
"AniList skip list update enabled in settings, will match but NOT update your list"
)
if anilist_settings.getboolean("plex_episode_count_priority", False):
logger.warning(
"Plex episode watched count will take priority over AniList, this will always update AniList watched count over Plex data"
)
anilist = Anilist(anilist_settings, custom_mappings)
anilist_series = anilist.process_user_list()
# Plex
if anilist_series is None:
logger.error(
"Unable to retrieve AniList list, check your username and access token"
)
else:
plexmodule = PlexModule(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_series_watched)
logger.info("Plex to AniList sync finished")
if __name__ == "__main__":
start()