def sync_collection(self, m: Media, dry_run=False):
if not self.sync_collection_enabled:
return
if m.is_collected:
return
logger.info(f"To be added to collection: {m}")
if not dry_run:
m.add_to_collection()
def submit_collection(self):
if self.queue_size() == 0:
return
try:
result = self.trakt_sync_collection(self.collection)
result = self.remove_empty_values(result.copy())
if result:
logger.info(f"Updated Trakt collection: {result}")
finally:
self.collection.clear()
def addList(self, username, listname, trakt_list=None):
if trakt_list is not None:
self.lists.append(TraktList.from_trakt_list(listname, trakt_list))
logger.info("Downloaded List {}".format(listname))
return
try:
self.lists.append(TraktList(username, listname))
logger.info("Downloaded List {}".format(listname))
except (NotFoundException, OAuthException):
logger.warning("Failed to get list {} by user {}".format(
listname, username))
def addPlexItem(self, trakt_item, plex_item):
rank = self.trakt_items.get((trakt_item.media_type, trakt_item.trakt))
if rank is not None:
self.plex_items.append((rank, plex_item))
if isinstance(plex_item, Episode):
logger.info('Show [{} ({})]: {} added to list {}'.format(
plex_item.show().title,
plex_item.show().year, plex_item.seasonEpisode, self.name))
else:
logger.info('Movie [{} ({})]: added to list {}'.format(
plex_item.title, plex_item.year, self.name))
def sync_watched(self, m: Media, dry_run=False):
if not self.sync_watched_status_enabled:
return
if m.watched_on_plex is m.watched_on_trakt:
return
if m.watched_on_plex:
logger.info(f"Marking as watched in Trakt: {m}")
if not dry_run:
m.mark_watched_trakt()
elif m.watched_on_trakt:
logger.info(f"Marking as watched in Plex: {m}")
if not dry_run:
m.mark_watched_plex()
def sync_ratings(self, m: Media, dry_run=False):
if not self.sync_ratings_enabled:
return
if m.plex_rating is m.trakt_rating:
return
# Plex rating takes precedence over Trakt rating
if m.plex_rating is not None:
logger.info(f"Rating {m} with {m.plex_rating} on Trakt")
if not dry_run:
m.trakt_rate()
elif m.trakt_rating is not None:
logger.info(f"Rating {m} with {m.trakt_rating} on Plex")
if not dry_run:
m.plex_rate()
def clear_collections(confirm, dry_run):
"""
Clear Movies and Shows collections in Trakt
"""
if not confirm and not dry_run:
click.echo('You need to pass --confirm or --dry-run option to proceed')
return
trakt = factory.trakt_api()
for movie in trakt.movie_collection:
logger.info(f"Deleting: {movie}")
if not dry_run:
trakt.remove_from_library(movie)
for show in trakt.show_collection:
logger.info(f"Deleting: {show}")
if not dry_run:
trakt.remove_from_library(show)
def sync(sync_option: str, library: str, show: str, movie: str, batch_size: int, dry_run: bool, no_progress_bar: bool):
"""
Perform sync between Plex and Trakt
"""
git_version = git_version_info()
if git_version:
logger.info(f"PlexTraktSync [{git_version}]")
ensure_login()
logger.info(f"Syncing with Plex {CONFIG['PLEX_USERNAME']} and Trakt {CONFIG['TRAKT_USERNAME']}")
movies = sync_option in ["all", "movies"]
tv = sync_option in ["all", "tv"]
plex = factory.plex_api()
trakt = factory.trakt_api(batch_size=batch_size)
mf = factory.media_factory(batch_size=batch_size)
pb = factory.progressbar(not no_progress_bar)
w = Walker(plex, mf, movies=movies, shows=tv, progressbar=pb)
if library:
logger.info(f"Filtering Library: {library}")
w.add_library(library)
if show:
w.add_show(show)
logger.info(f"Syncing Show: {show}")
if movie:
w.add_movie(movie)
logger.info(f"Syncing Movie: {movie}")
if not w.is_valid():
click.echo("Nothing to sync, this is likely due conflicting options given.")
return
if dry_run:
print("Enabled dry-run mode: not making actual changes")
w.walk_details(print=tqdm.write)
with measure_time("Completed full sync"):
sync_all(walker=w, trakt=trakt, plex=plex, dry_run=dry_run)
def _get_plex_server():
CONFIG = factory.config()
plex_token = CONFIG["PLEX_TOKEN"]
plex_baseurl = CONFIG["PLEX_BASEURL"]
plex_fallbackurl = CONFIG["PLEX_FALLBACKURL"]
if plex_token == '-':
plex_token = ""
server = None
plexapi.X_PLEX_PLATFORM = PLEX_PLATFORM
plexapi.BASE_HEADERS['X-Plex-Platform'] = plexapi.X_PLEX_PLATFORM
session = factory.session()
PlexServer = partial(plexapi.server.PlexServer, session=session)
# if connection fails, it will try :
# 1. url expected by new ssl certificate
# 2. url without ssl
# 3. fallback url (localhost)
try:
server = PlexServer(token=plex_token, baseurl=plex_baseurl)
except plexapi.server.requests.exceptions.SSLError as e:
m = "Plex connection error: {}, fallback url {} didn't respond either.".format(
str(e), plex_fallbackurl)
excep_msg = str(e.__context__)
if "doesn't match '*." in excep_msg:
hash_pos = excep_msg.find("*.") + 2
new_hash = excep_msg[hash_pos:hash_pos + 32]
end_pos = plex_baseurl.find(".plex.direct")
new_plex_baseurl = plex_baseurl[:end_pos -
32] + new_hash + plex_baseurl[
end_pos:]
try: # 1
server = PlexServer(token=plex_token, baseurl=new_plex_baseurl)
# save new url to .env
CONFIG["PLEX_TOKEN"] = plex_token
CONFIG["PLEX_BASEURL"] = new_plex_baseurl
CONFIG["PLEX_FALLBACKURL"] = plex_fallbackurl
CONFIG.save()
logger.info(
"Plex server url changed to {}".format(new_plex_baseurl))
except Exception:
pass
if server is None and plex_baseurl[:5] == "https":
new_plex_baseurl = plex_baseurl.replace("https", "http")
try: # 2
server = PlexServer(token=plex_token, baseurl=new_plex_baseurl)
logger.warning(
"Switched to Plex unsecure connection because of SSLError."
)
except Exception:
pass
except Exception as e:
m = "Plex connection error: {}, fallback url {} didn't respond either.".format(
str(e), plex_fallbackurl)
pass
if server is None:
try: # 3
server = PlexServer(token=plex_token, baseurl=plex_fallbackurl)
logger.warning("No response from {}, fallback to {}".format(
plex_baseurl, plex_fallbackurl))
except Exception:
logger.error(m)
print(m)
exit(1)
return server