bazarr/get_movies.py

import os
import sqlite3
import requests

from get_general_settings import *
from list_subtitles import *

def update_movies():
    from get_radarr_settings import get_radarr_settings
    url_radarr = get_radarr_settings()[0]
    url_radarr_short = get_radarr_settings()[1]
    apikey_radarr = get_radarr_settings()[2]

    # Open database connection
    db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
    c = db.cursor()

    if apikey_radarr == None:
        pass
    else:
        get_profile_list()

        # Get movies data from radarr
        url_radarr_api_movies = url_radarr + "/api/movie?apikey=" + apikey_radarr
        r = requests.get(url_radarr_api_movies)

        # Get current movies in DB
        current_movies_db = c.execute('SELECT tmdbId FROM table_movies').fetchall()
        current_movies_db_list = [x[0] for x in current_movies_db]
        current_movies_radarr = []

        for movie in r.json():
            if movie['hasFile'] is True:
                try:
                    overview = unicode(movie['overview'])
                except:
                    overview = ""
                try:
                    poster_big = movie['images'][0]['url']
                    poster = os.path.splitext(poster_big)[0] + '-500' + os.path.splitext(poster_big)[1]
                except:
                    poster = ""
                try:
                    fanart = movie['images'][1]['url']
                except:
                    fanart = ""

                if 'sceneName' in movie['movieFile']:
                    sceneName = movie['movieFile']['sceneName']
                else:
                    sceneName = None

                # Add movies in radarr to current movies list
                current_movies_radarr.append(unicode(movie['tmdbId']))

                # Detect file separator
                if movie['path'][0] == "/":
                    separator = "/"
                else:
                    separator == "\\"

                # Update or insert movies list in database table
                try:
                    c.execute('''INSERT INTO table_movies(title, path, tmdbId, languages,`hearing_impaired`, radarrId, overview, poster, fanart, `audio_language`, sceneName) VALUES (?,?,?,(SELECT languages FROM table_movies WHERE tmdbId = ?),(SELECT `hearing_impaired` FROM table_movies WHERE tmdbId = ?), ?, ?, ?, ?, ?, ?)''', (movie["title"], movie["path"] + separator + movie['movieFile']['relativePath'], movie["tmdbId"], movie["tmdbId"], movie["tmdbId"], movie["id"], overview, poster, fanart, profile_id_to_language(movie['qualityProfileId']), sceneName))
                except:
                    c.execute('''UPDATE table_movies SET title = ?, path = ?, tmdbId = ?, radarrId = ?, overview = ?, poster = ?, fanart = ?, `audio_language` = ?, sceneName = ? WHERE tmdbid = ?''', (movie["title"],movie["path"] + separator + movie['movieFile']['relativePath'],movie["tmdbId"],movie["id"],overview,poster,fanart,profile_id_to_language(movie['qualityProfileId']),sceneName,movie["tmdbId"]))

                # Commit changes to database table
                db.commit()

        # Delete movies not in radarr anymore
        added_movies = list(set(current_movies_radarr) - set(current_movies_db_list))
        removed_movies = list(set(current_movies_db_list) - set(current_movies_radarr))

        for removed_movie in removed_movies:
            c.execute('DELETE FROM table_movies WHERE radarrId = ?', (removed_movie,))
            db.commit()

        for added_movie in added_movies:
            added_path = c.execute('SELECT path FROM table_movies WHERE tmdbId = ?', (added_movie,)).fetchone()
            store_subtitles_movie(path_replace(added_path[0]))

    # Close database connection
    db.close()

def get_profile_list():
    from get_radarr_settings import get_radarr_settings
    url_radarr = get_radarr_settings()[0]
    url_radarr_short = get_radarr_settings()[1]
    apikey_radarr = get_radarr_settings()[2]

    # Get profiles data from radarr
    url_radarr_api_movies = url_radarr + "/api/profile?apikey=" + apikey_radarr
    profiles_json = requests.get(url_radarr_api_movies)
    global profiles_list
    profiles_list = []

    # Parsing data returned from radarr
    for profile in profiles_json.json():
        profiles_list.append([profile['id'], profile['language'].capitalize()])

def profile_id_to_language(id):
    for profile in profiles_list:
        if id == profile[0]:
            return profile[1]

if __name__ == '__main__':
    update_movies()