Saves poster images from Plex to same folder as the media files.

save_posters.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
Description:  Saves poster images from Plex to same folder as the media files.
Author:       /u/SwiftPanda16
Requires:     plexapi, tqdm (optional)
Usage:
    * Save posters for an entire library:
        python save_posters.py --library "Movies"

    * Save posters for a specific media type in a library:
        python save_posters.py --library "TV Shows" --libtype season

    * Save posters for a specific item:
        python save_posters.py --rating_key 1234
'''

import argparse
from pathlib import Path
from plexapi.server import PlexServer
from plexapi.utils import download


PLEX_URL = 'http://localhost:32400'
PLEX_TOKEN = 'XXXXXXXXXXXXXXXXXXXX'
# Specify the mapped docker folder paths {host: container}. Leave blank {} if non-docker.
MAPPED_FOLDERS = {
    '/mnt/movies': '/movies',
    '/mnt/tvshows': '/tv',
}

_MAPPED_FOLDERS = {Path(host): Path(container) for host, container in MAPPED_FOLDERS.items()}


def map_path(file_path):
    for host, container in _MAPPED_FOLDERS.items():
        if container in file_path.parents:
            return host / file_path.relative_to(container)
    return file_path


def save_posters_library(library, libtype=None):
    for item in library.all(libtype=libtype, includeGuids=False):
        save_posters_item(item)


def save_posters_item(item):
    if hasattr(item, 'locations'):
        file_path = Path(item.locations[0])
    else:
        file_path = Path(next(iter(item)).locations[0])
    poster_path = map_path(file_path)
    if poster_path.is_file():
        poster_path = poster_path.parent

    print(f"Downloading poster for {item.title} to {poster_path}")
    try:
        download(
            url=item.posterUrl,
            token=plex._token,
            filename='poster.jpg',
            savepath=poster_path,
            showstatus=True  # Requires `tqdm` package
        )
    except Exception as e:
        print(f"Failed to download poster for {item.title}: {e}")


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--rating_key', type=int)
    parser.add_argument('--library')
    parser.add_argument('--libtype', choices=['movie', 'show', 'season', 'artist', 'album'])
    opts = parser.parse_args()

    plex = PlexServer(PLEX_URL, PLEX_TOKEN)

    if opts.rating_key:
        item = plex.fetchItem(opts.rating_key)
        save_posters_item(item)
    elif opts.library:
        library = plex.library.section(opts.library)
        save_posters_library(library, opts.libtype)
    else:
        print("No --rating_key or --library specified. Exiting.")