from tmdbv3api import TMDb
from tmdbv3api import Movie
from PIL import Image
from PIL import ImageFile
import getopt
import requests
import shutil
import os
import sys
import re
from re import search

def main(argv):
    work_folder=''
    try:
        opts, args = getopt.getopt(argv,"h:f",["folder="])
    except getopt.GetoptError:
        print ('download_poster.py -f <work_folder>')
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print ('download_poster.py -f <work_folder>')
            sys.exit()
        elif opt in ("-f", "--folder"):
            work_folder = arg
    if (work_folder==""):
        print ("Se necesita definir el directorio de trabajo")
        sys.exit()

    print ('El directorio de trabajo es '+work_folder)
    tmdb = TMDb()
    # En tmp.api_key ponemos el que consigamos en https://developers.themoviedb.org/3
    tmdb.api_key = 'SUSTITUIR_POR_API_KEY'
    tmdb.language = 'es'
    tmdb.debug = False
    movie = Movie()
    video_extensions = [".mkv",".avi",".mp4",".iso"]
    ImageFile.LOAD_TRUNCATED_IMAGES = True

    def store_image (file,poster_path):
        if (type(poster_path) == str and type(file) == str):
            image_url = "https://image.tmdb.org/t/p/w185/"+poster_path
            resp = requests.get(image_url, stream=True)
            local_file = open(work_folder+"tmp.jpg", 'wb')
            resp.raw.decode_content = True
            shutil.copyfileobj(resp.raw, local_file)
            del resp
            try:
                img = Image.open(work_folder+'tmp.jpg', 'r')
                img_w, img_h = img.size
                background = Image.new('RGBA', (258,258), (255,255,255,0))
                bg_w, bg_h = background.size
                offset = ((bg_w - img_w) // 2, (bg_h - img_h) // 2)
                try:
                    background.paste(img, offset)
                    background.save(work_folder+file+".png", optimize = True, compress_level = 9)
                    os.remove(work_folder+"tmp.jpg")
                except:
                    os.rename(work_folder+"tmp.jpg",work_folder+file+".jpg")
            except:
                print ("Error al bajar imagen")