Add incremental filename support to poster pull utility

This commit is contained in:
Derek Quenneville 2023-09-15 14:13:34 -04:00
parent c30850673a
commit c5eec4ff2e

View File

@ -69,9 +69,12 @@ for library in library_name:
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 directory
urllib.request.urlretrieve(thumb_url, image_path)
# If the poster file to be written already exists, append an incrementing number to the filename
increment = 0
while os.path.isfile(image_path):
increment += 1
image_path = u'{}/{}_{}.jpg'.format(os.path.dirname(image_path), name, increment)
# Save to directory
urllib.request.urlretrieve(thumb_url, image_path)