From 7ffaa7543fef690e51270b49d84baa1861ec5520 Mon Sep 17 00:00:00 2001
From: Blacktwin <asimmons28501@gmail.com>
Date: Tue, 3 Apr 2018 11:57:51 -0400
Subject: [PATCH] cleanup

---
 utility/plex_api_poster_pull.py | 47 +++++++++++++++------------------
 1 file changed, 22 insertions(+), 25 deletions(-)

diff --git a/utility/plex_api_poster_pull.py b/utility/plex_api_poster_pull.py
index c6fe95d..053a0bc 100644
--- a/utility/plex_api_poster_pull.py
+++ b/utility/plex_api_poster_pull.py
@@ -1,19 +1,22 @@
-'''
-Pull Movie and TV Show poster images from Plex. 
-Save to Movie and TV Show directories in scripts working directory.
+"""
+Description: Pull Movie and TV Show poster images from Plex.  Save to Movie and TV Show directories in scripts working directory.
+Author: Blacktwin
+Requires: plexapi
 
-'''
+ Example:
+    python plex_api_poster_pull.py
+
+"""
 
 
 from plexapi.server import PlexServer
-# pip install plexapi
 import re
 import os
 import urllib
 
 
 PLEX_URL = 'http://localhost:32400'
-PLEX_TOKEN = 'xxxxxxx'
+PLEX_TOKEN = ''
 plex = PlexServer(PLEX_URL, PLEX_TOKEN)
 
 library_name = ['Movies','TV Shows'] # You library names
@@ -27,30 +30,24 @@ show_path = '{}/TV Shows'.format(os.path.dirname(__file__))
 if not os.path.isdir(show_path):
     os.mkdir(show_path)
 
-child_lst = []
 
 # Get all movies or shows from LIBRARY_NAME
 for library in library_name:
     for child in plex.library.section(library).all():
-        child_lst.append(child)
-
-
-for video in child_lst:
-    # Clean names of special characters
-    name = re.sub('\W+',' ', video.title)
-    # Pull URL for poster
-    thumb_url = '{}{}?X-Plex-Token={}'.format(PLEX_URL, video.thumb, PLEX_TOKEN)
-    if video.type == 'movie':
-        image_path = u'{}/{}.jpg'.format(movie_path, name)
+        # Clean names of special characters
+        name = re.sub('\W+',' ', child.title)
+        # Add (year) to name
+        name = '{} ({})'.format(name, child.year)
+        # Pull URL for poster
+        thumb_url = '{}{}?X-Plex-Token={}'.format(PLEX_URL, child.thumb, PLEX_TOKEN)
+        # Select path based on media_type
+        if child.type == 'movie':
+            image_path = u'{}/{}.jpg'.format(movie_path, name)
+        elif child.type == 'show':
+            image_path = u'{}/{}.jpg'.format(show_path, name)
+        # Check if file already exists
         if os.path.isfile(image_path):
             print("ERROR, %s already exist" % image_path)
         else:
-            # Save to Movie directory
-            urllib.urlretrieve(thumb_url, image_path)
-    elif video.type == 'show':
-        image_path = u'{}/{}.jpg'.format(show_path, name)
-        if os.path.isfile(image_path):
-            print("ERROR, %s already exist" % image_path)
-        else:
-            # Save to TV Show directory
+            # Save to directory
             urllib.urlretrieve(thumb_url, image_path)