kodi.py

#!/usr/bin/env python

# For a complete discussion, see http://forum.kodi.tv/showthread.php?tid=254502

import datetime
import json
import requests
import time
import codecs
import urllib
import os
import random
import re
import string
import sys
import codecs
import unicodedata
import roman
from fuzzywuzzy import fuzz, process
from yaep import populate_env, env

sys.path += [os.path.dirname(__file__)]

ENV_FILE = os.path.join(os.path.dirname(__file__), ".env")
os.environ['ENV_FILE'] = ENV_FILE

populate_env()

# Do not use below for your own settings, use the .env file

# We don't use the fallback param in os.getenv() because AWS Lambda actually
# sets any environment variables included in LAMBDA_ENV_VARS, regardless if
# it was unset before.
#
# Furthermore, os.getenv() under AWS Lambda returns 'None' as a string
# instead of None as NoneType as we'd normally expect, so we have to
# explicitly test for that.

SCHEME = os.getenv('KODI_SCHEME')
if not SCHEME or SCHEME == 'None':
  SCHEME = 'http'
SUBPATH = os.getenv('KODI_SUBPATH')
if not SUBPATH or SUBPATH == 'None':
  SUBPATH = ''
KODI = os.getenv('KODI_ADDRESS')
if not KODI or KODI == 'None':
  KODI = '127.0.0.1'
PORT = os.getenv('KODI_PORT')
if not PORT or PORT == 'None':
  PORT = '8080'
USER = os.getenv('KODI_USERNAME')
if not USER or USER == 'None':
  USER = 'kodi'
PASS = os.getenv('KODI_PASSWORD')
if not PASS or PASS == 'None':
  PASS = 'kodi'

# These are words that we ignore when doing a non-exact match on show names
STOPWORDS = [
  "a",
  "about",
  "an",
  "and",
  "are",
  "as",
  "at",
  "be",
  "by",
  "for",
  "from",
  "how",
  "in",
  "is",
  "it",
  "of",
  "on",
  "or",
  "that",
  "the",
  "this",
  "to",
  "was",
  "what",
  "when",
  "where",
  "will",
  "with",
]


def sanitize_name(media_name, remove_between=False):
  # Normalize string
  name = unicodedata.normalize('NFKD', media_name).encode('ASCII', 'ignore')

  if remove_between:
    # Strip things between and including brackets and parentheses
    name = re.sub(r'\([^)]*\)', '', name)
    name = re.sub(r'\[[^)]*\]', '', name)
  else:
    # Just remove the actual brackets and parentheses
    name = name.translate(None, '[]()')

  name = name.translate(None, '"')

  if len(name) > 140:
    name = name[:140].rsplit(' ', 1)[0]

  name = name.strip()
  return name


# Very naive method to remove a leading "the" from the given string
def remove_the(name):
  if name[:4].lower() == "the ":
    return name[4:]
  else:
    return name


# Remove extra slashes
def http_normalize_slashes(url):
  url = str(url)
  segments = url.split('/')
  correct_segments = []
  for segment in segments:
    if segment != '':
      correct_segments.append(segment)
  first_segment = str(correct_segments[0])
  if first_segment.find('http') == -1:
    correct_segments = ['http:'] + correct_segments
  correct_segments[0] = correct_segments[0] + '/'
  normalized_url = '/'.join(correct_segments)
  return normalized_url


def PopulateEnv():
  populate_env()


# These two methods construct the JSON-RPC message and send it to the Kodi player
def SendCommand(command):
  # Join the environment variables into a url
  url = "%s://%s:%s/%s/%s" % (SCHEME, KODI, PORT, SUBPATH, 'jsonrpc')

  # Remove any double slashes in the url
  url = http_normalize_slashes(url)

  print "Sending request to %s" % (url)

  try:
    r = requests.post(url, data=command, auth=(USER, PASS))
  except:
    return {}
  return json.loads(r.text)


def RPCString(method, params=None):
  j = {"jsonrpc":"2.0", "method":method, "id":1}
  if params:
    j["params"] = params
  return json.dumps(j)


# Convert numbers to words.
# XXXLANG: This is currently English-only and shouldn't be!
def num2word(number):
  # based on stackoverflow answer from https://github.com/ralphembree/Loquitor
  leading_zero = 0
  ones = ("", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine")
  tens = ("", "", "twenty", "thirty", "forty", "fifty", "sixty", "seventy", "eighty", "ninety")
  teens = ("ten", "eleven", "twelve", "thirteen", "fourteen", "fifteen", "sixteen", "seventeen", "eighteen", "nineteen")
  levels = ("", "thousand", "million", "billion", "trillion", "quadrillion", "quintillion", "sextillion", "septillion", "octillion", "nonillion")

  word = ""
  if number[0] == "0":
    leading_zero = 1
  # number will now be the reverse of the string form of itself.
  num = reversed(str(number))
  number = ""
  for x in num:
    number += x
  del num
  if len(number) % 3 == 1:
    number += "0"
  x = 0
  for digit in number:
    if x % 3 == 0:
      word = levels[x / 3] + " " + word
      n = int(digit)
    elif x % 3 == 1:
      if digit == "1":
        num = teens[n]
      else:
        num = tens[int(digit)]
        if n:
          if num:
            num += " " + ones[n]
          else:
            num = ones[n]
      word = num + " " + word
    elif x % 3 == 2:
      if digit != "0":
        word = ones[int(digit)] + " hundred and " + word
    x += 1
  reply = word.strip(", ")
  if leading_zero:
    reply = "zero " +reply
  return reply


# Replace word-form numbers with digits.
# XXXLANG: This is currently English-only and shouldn't be!
def word2num(textnum):
  numwords = {}
  units = [
    "zero", "one", "two", "three", "four", "five", "six", "seven", "eight",
    "nine", "ten", "eleven", "twelve", "thirteen", "fourteen", "fifteen",
    "sixteen", "seventeen", "eighteen", "nineteen",
  ]
  tens = ["", "", "twenty", "thirty", "forty", "fifty", "sixty", "seventy", "eighty", "ninety"]
  scales = ["hundred", "thousand", "million", "billion", "trillion"]

  numwords["and"] = (1, 0)
  for idx, word in enumerate(units):
    numwords[word] = (1, idx)
  for idx, word in enumerate(tens):
    numwords[word] = (1, idx * 10)
  for idx, word in enumerate(scales):
    numwords[word] = (10 ** (idx * 3 or 2), 0)

  current = result = 0
  for word in textnum.split():
    if word not in numwords:
      raise Exception("Illegal word: " + word)

    scale, increment = numwords[word]
    current = current * scale + increment
    if scale > 100:
      result += current
      current = 0

  return result + current

# Replace digits with word-form numbers.
# XXXLANG: This is currently English-only and shouldn't be!
def digits2words(phrase):
  # all word variant of the heard phrase if it contains digits
  wordified = ''
  for word in phrase.split():
    word = word.decode('utf-8')
    if word.isnumeric():
      word = num2word(word)
    wordified = wordified + word + " "
  return wordified[:-1]


# Replace digits with roman numerals.
# XXXLANG: This is currently English-only and shouldn't be!
def digits2roman(phrase):
  wordified = ''
  for word in phrase.split():
    word = word.decode('utf-8')
    if word.isnumeric():
      word = roman.toRoman(int(word))
    wordified = wordified + word + " "
  return wordified[:-1]


# Replace word-form numbers with roman numerals.
# XXXLANG: This is currently English-only and shouldn't be!
def words2roman(phrase):
  wordified = ''
  for word in phrase.split():
    word = word.decode('utf-8')
    try:
      word = roman.toRoman(word2num(word))
    except:
      pass
    wordified = wordified + word + " "
  return wordified[:-1]


# Match heard string to something in the results
def matchHeard(heard, results, lookingFor='label'):
  located = None

  heard_minus_the = remove_the(heard)
  print 'Trying to match: ' + heard

  heard_list = set([x for x in heard.split() if x not in STOPWORDS])

  for result in results:
    # Strip out non-ascii symbols and lowercase it
    result_name = sanitize_name(result[lookingFor]).lower()

    # Direct comparison
    if heard == result_name:
      print 'Simple match on direct comparison'
      located = result
      break

    # Remove 'the'
    if remove_the(result_name) == heard_minus_the:
      print 'Simple match minus "the"'
      located = result
      break

    # Remove parentheses
    removed_paren = sanitize_name(result[lookingFor], True).lower()
    if heard == removed_paren:
      print 'Simple match minus parentheses'
      located = result
      break

  if not located:
    print 'Simple match failed, trying fuzzy match...'

    fuzzy_result = False
    for f in (digits2roman, words2roman, str, digits2words):
      try:
        ms = f(heard)
        print "Trying to match %s from %s" % (ms, f)
        rv = process.extract(ms, [d[lookingFor] for d in results], limit=1, scorer=fuzz.QRatio)
        if rv[0][1] >= 75:
          fuzzy_result = rv
          break
      except:
        continue

    # Got a match?
    if fuzzy_result:
      print 'Fuzzy match %s%%' % (fuzzy_result[0][1])
      located = (item for item in results if item[lookingFor] == fuzzy_result[0][0]).next()

  return located


# Helpers to find media

def FindVideoPlaylist(heard_search):
  print 'Searching for video playlist "%s"' % (heard_search)

  playlists = GetVideoPlaylists()
  if 'result' in playlists and 'files' in playlists['result']:
    playlists_list = playlists['result']['files']
    located = matchHeard(heard_search, playlists_list, 'label')

    if located:
      print 'Located video playlist "%s"' % (located['file'])
      return located['file']


def FindAudioPlaylist(heard_search):
  print 'Searching for audio playlist "%s"' % (heard_search)

  playlists = GetMusicPlaylists()
  if 'result' in playlists and 'files' in playlists['result']:
    playlists_list = playlists['result']['files']
    located = matchHeard(heard_search, playlists_list, 'label')

    if located:
      print 'Located audio playlist "%s"' % (located['file'])
      return located['file']


def FindMovie(heard_search):
  print 'Searching for movie "%s"' % (heard_search)

  movies = GetMovies()
  if 'result' in movies and 'movies' in movies['result']:
    movies_array = movies['result']['movies']
    located = matchHeard(heard_search, movies_array)

    if located:
      print 'Located movie "%s"' % (located['label'])
      return located['movieid']


def FindTvShow(heard_search):
  print 'Searching for show "%s"' % (heard_search)

  shows = GetTvShows()
  if 'result' in shows and 'tvshows' in shows['result']:
    shows_array = shows['result']['tvshows']
    located = matchHeard(heard_search, shows_array)

    if located:
      print 'Located tvshow "%s"' % (located['label'])
      return located['tvshowid']


def FindArtist(heard_search):
  print 'Searching for artist "%s"' % (heard_search)

  artists = GetMusicArtists()
  if 'result' in artists and 'artists' in artists['result']:
    artists_list = artists['result']['artists']
    located = matchHeard(heard_search, artists_list, 'artist')

    if located:
      print 'Located artist "%s"' % (located['label'])
      return located['artistid']


def FindAlbum(heard_search):
  print 'Searching for album "%s"' % (heard_search)

  albums = GetAlbums()
  if 'result' in albums and 'albums' in albums['result']:
    albums_list = albums['result']['albums']
    located = matchHeard(heard_search, albums_list, 'label')

    if located:
      print 'Located album "%s"' % (located['label'])
      return located['albumid']


def FindSong(heard_search):
  print 'Searching for song "%s"' % (heard_search)

  songs = GetSongs()
  if 'result' in songs and 'songs' in songs['result']:
    songs_list = songs['result']['songs']
    located = matchHeard(heard_search, songs_list, 'label')

    if located:
      print 'Located song "%s"' % (located['label'])
      return located['songid']


# Playlists

def ClearAudioPlaylist():
  return SendCommand(RPCString("Playlist.Clear", {"playlistid": 0}))


def AddSongToPlaylist(song_id):
  return SendCommand(RPCString("Playlist.Add", {"playlistid": 0, "item": {"songid": int(song_id)}}))


def AddSongsToPlaylist(song_ids, shuffle=False):
  songs_array = []

  for song_id in song_ids:
    temp_song = {}
    temp_song['songid'] = song_id
    songs_array.append(temp_song)

  if shuffle:
    random.shuffle(songs_array)

  # Segment the requests into chunks that Kodi will accept in a single call
  song_groups = [songs_array[x:x+2000] for x in range(0, len(songs_array), 2000)]
  for a in song_groups:
    print "Adding %d items to the queue..." % (len(a))
    res = SendCommand(RPCString("Playlist.Add", {"playlistid": 0, "item": a}))

  return res


def AddAlbumToPlaylist(album_id):
  return SendCommand(RPCString("Playlist.Add", {"playlistid": 0, "item": {"albumid": int(album_id)}}))


def GetAudioPlaylistItems():
  return SendCommand(RPCString("Playlist.GetItems", {"playlistid": 0}))


def StartAudioPlaylist(playlist_file=None):
  if playlist_file is not None and playlist_file != '':
    return SendCommand(RPCString("Player.Open", {"item": {"file": playlist_file}}))
  else:
    return SendCommand(RPCString("Player.Open", {"item": {"playlistid": 0}}))


def ClearVideoPlaylist():
  return SendCommand(RPCString("Playlist.Clear", {"playlistid": 1}))


def AddEpisodeToPlayList(ep_id):
  return SendCommand(RPCString("Playlist.Add", {"playlistid": 1, "item": {"episodeid": int(ep_id)}}))


def AddMovieToPlaylist(movie_id):
  return SendCommand(RPCString("Playlist.Add", {"playlistid": 1, "item": {"movieid": int(movie_id)}}))


def AddVideosToPlaylist(video_files, shuffle=False):
  videos_array = []

  for video_file in video_files:
    temp_video = {}
    temp_video['file'] = video_file
    videos_array.append(temp_video)

  if shuffle:
    random.shuffle(videos_array)

  # Segment the requests into chunks that Kodi will accept in a single call
  video_groups = [videos_array[x:x+2000] for x in range(0, len(videos_array), 2000)]
  for a in video_groups:
    print "Adding %d items to the queue..." % (len(a))
    res = SendCommand(RPCString("Playlist.Add", {"playlistid": 1, "item": a}))

  return res


def GetVideoPlaylistItems():
  return SendCommand(RPCString("Playlist.GetItems", {"playlistid": 1}))


def StartVideoPlaylist(playlist_file=None):
  if playlist_file is not None and playlist_file != '':
    return SendCommand(RPCString("Player.Open", {"item": {"file": playlist_file}}))
  else:
    return SendCommand(RPCString("Player.Open", {"item": {"playlistid": 1}}))


# Direct plays

def PlayEpisode(ep_id, resume=True):
  return SendCommand(RPCString("Player.Open", {"item": {"episodeid": ep_id}, "options": {"resume": resume}}))


def PlayMovie(movie_id, resume=True):
  return SendCommand(RPCString("Player.Open", {"item": {"movieid": movie_id}, "options": {"resume": resume}}))


def PartyPlayMusic():
  return SendCommand(RPCString("Player.Open", {"item": {"partymode": "music"}}))


# Tell Kodi to update its video or music libraries

def UpdateVideo():
  return SendCommand(RPCString("VideoLibrary.Scan"))


def CleanVideo():
  return SendCommand(RPCString("VideoLibrary.Clean"))


def UpdateMusic():
  return SendCommand(RPCString("AudioLibrary.Scan"))


def CleanMusic():
  return SendCommand(RPCString("AudioLibrary.Clean"))


# Perform UI actions that match the normal remote control buttons

def PageUp():
  return SendCommand(RPCString("Input.ExecuteAction", {"action":"pageup"}))


def PageDown():
  return SendCommand(RPCString("Input.ExecuteAction", {"action":"pagedown"}))


def ToggleWatched():
  return SendCommand(RPCString("Input.ExecuteAction", {"action":"togglewatched"}))


def Info():
  return SendCommand(RPCString("Input.Info"))


def Menu():
  return SendCommand(RPCString("Input.ContextMenu"))


def Home():
  return SendCommand(RPCString("Input.Home"))


def Select():
  return SendCommand(RPCString("Input.Select"))


def Up():
  return SendCommand(RPCString("Input.Up"))


def Down():
  return SendCommand(RPCString("Input.Down"))


def Left():
  return SendCommand(RPCString("Input.Left"))


def Right():
  return SendCommand(RPCString("Input.Right"))


def Back():
  return SendCommand(RPCString("Input.Back"))


def ToggleFullscreen():
  return SendCommand(RPCString("GUI.SetFullscreen", {"fullscreen":"toggle"}))


def ToggleMute():
  return SendCommand(RPCString("Application.SetMute", {"mute":"toggle"}))


def GetCurrentVolume():
  return SendCommand(RPCString("Application.GetProperties", {"properties":["volume", "muted"]}))

def VolumeUp():
  resp = GetCurrentVolume()
  vol = resp['result']['volume']
  if vol % 10 == 0:
    # already modulo 10, so just add 10
    vol += 10
  else:
    # round up to nearest 10
    vol -= vol % -10
  if vol > 100:
    vol = 100
  return SendCommand(RPCString("Application.SetVolume", {"volume":vol}))


def VolumeDown():
  resp = GetCurrentVolume()
  vol = resp['result']['volume']
  if vol % 10 != 0:
    # round up to nearest 10 first
    vol -= vol % -10
  vol -= 10
  if vol < 0:
    vol = 0
  return SendCommand(RPCString("Application.SetVolume", {"volume":vol}))


def VolumeSet(vol, percent=True):
  if vol < 0:
    vol = 0
  if not percent:
    # specified with scale of 0 to 10
    vol *= 10
  if vol > 100:
    vol = 100
  return SendCommand(RPCString("Application.SetVolume", {"volume":vol}))


# Player controls

def PlayerPlayPause():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.PlayPause", {"playerid":playerid}))


def PlayerSkip():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.GoTo", {"playerid":playerid, "to": "next"}))


def PlayerPrev():
  playerid = GetPlayerID()
  if playerid is not None:
    SendCommand(RPCString("Player.GoTo", {"playerid":playerid, "to": "previous"}))
    return SendCommand(RPCString("Player.GoTo", {"playerid":playerid, "to": "previous"}))


def PlayerStartOver():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.Seek", {"playerid":playerid, "value": 0}))


def PlayerStop():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.Stop", {"playerid":playerid}))


def PlayerSeek(seconds):
  playerid = GetPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Seek", {"playerid":playerid, "value":{"seconds":seconds}}))


def PlayerSeekSmallForward():
  playerid = GetPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Seek", {"playerid":playerid, "value":"smallforward"}))


def PlayerSeekSmallBackward():
  playerid = GetPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Seek", {"playerid":playerid, "value":"smallbackward"}))


def PlayerSeekBigForward():
  playerid = GetPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Seek", {"playerid":playerid, "value":"bigforward"}))


def PlayerSeekBigBackward():
  playerid = GetPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Seek", {"playerid":playerid, "value":"bigbackward"}))


def PlayerShuffleOn():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.SetShuffle", {"playerid":playerid, "shuffle":True}))


def PlayerShuffleOff():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.SetShuffle", {"playerid":playerid, "shuffle":False}))


def PlayerLoopOn():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.SetRepeat", {"playerid":playerid, "repeat":"cycle"}))


def PlayerLoopOff():
  playerid = GetPlayerID()
  if playerid is not None:
    return SendCommand(RPCString("Player.SetRepeat", {"playerid":playerid, "repeat":"off"}))


def PlayerSubtitlesOn():
  playerid = GetVideoPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.SetSubtitle", {"playerid":playerid, "subtitle":"on"}))


def PlayerSubtitlesOff():
  playerid = GetVideoPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.SetSubtitle", {"playerid":playerid, "subtitle":"off"}))


def PlayerSubtitlesNext():
  playerid = GetVideoPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.SetSubtitle", {"playerid":playerid, "subtitle":"next", "enable":True}))


def PlayerSubtitlesPrevious():
  playerid = GetVideoPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.SetSubtitle", {"playerid":playerid, "subtitle":"previous", "enable":True}))


def PlayerAudioStreamNext():
  playerid = GetVideoPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.SetAudioStream", {"playerid":playerid, "stream":"next"}))


def PlayerAudioStreamPrevious():
  playerid = GetVideoPlayerID()
  if playerid:
    return SendCommand(RPCString("Player.SetAudioStream", {"playerid":playerid, "stream":"previous"}))


def PlayerMoveUp():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Move", {"playerid":playerid, "direction":"up"}))


def PlayerMoveDown():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Move", {"playerid":playerid, "direction":"down"}))


def PlayerMoveLeft():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Move", {"playerid":playerid, "direction":"left"}))


def PlayerMoveRight():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Move", {"playerid":playerid, "direction":"right"}))


def PlayerZoom(lvl=0):
  playerid = GetPicturePlayerID()
  if playerid and lvl > 0 and lvl < 11:
    return SendCommand(RPCString("Player.Zoom", {"playerid":playerid, "zoom":lvl}))


def PlayerZoomIn():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Zoom", {"playerid":playerid, "zoom":"in"}))


def PlayerZoomOut():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Zoom", {"playerid":playerid, "zoom":"out"}))


def PlayerRotateClockwise():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Rotate", {"playerid":playerid, "value":"clockwise"}))


def PlayerRotateCounterClockwise():
  playerid = GetPicturePlayerID()
  if playerid:
    return SendCommand(RPCString("Player.Rotate", {"playerid":playerid, "value":"counterclockwise"}))


# Addons

def AddonExecute(addon_id, params={}):
  return SendCommand(RPCString("Addons.ExecuteAddon", {"addonid":addon_id, "params":params}))

def AddonGlobalSearch(needle=''):
  return AddonExecute("script.globalsearch", {"searchstring":needle})

def AddonCinemaVision():
  return AddonExecute("script.cinemavision", ["experience"])


# Library queries

# content can be: video, audio, image, executable, or unknown
def GetAddons(content):
  if content:
    return SendCommand(RPCString("Addons.GetAddons", {"content":content, "properties":["name"]}))
  else:
    return SendCommand(RPCString("Addons.GetAddons", {"properties":["name"]}))


def GetAddonDetails(addon_id):
  return SendCommand(RPCString("Addons.GetAddonDetails", {"addonid":addon_id, "properties":["name", "version", "description", "summary"]}))


def GetPlaylistItems(playlist_file):
  return SendCommand(RPCString("Files.GetDirectory", {"directory": playlist_file}))


def GetMusicPlaylists():
  return SendCommand(RPCString("Files.GetDirectory", {"directory": "special://musicplaylists"}))


def GetMusicArtists():
  return SendCommand(RPCString("AudioLibrary.GetArtists"))


def GetMusicGenres():
  return SendCommand(RPCString("AudioLibrary.GetGenres"))


def GetArtistAlbums(artist_id):
  return SendCommand(RPCString("AudioLibrary.GetAlbums", {"filter": {"artistid": int(artist_id)}}))


def GetAlbums():
  return SendCommand(RPCString("AudioLibrary.GetAlbums"))


def GetArtistSongs(artist_id):
  return SendCommand(RPCString("AudioLibrary.GetSongs", {"filter": {"artistid": int(artist_id)}}))


def GetArtistSongsPath(artist_id):
  return SendCommand(RPCString("AudioLibrary.GetSongs", {"filter": {"artistid": int(artist_id)}, "properties":["file"]}))


def GetAlbumSongsPath(album_id):
  return SendCommand(RPCString("AudioLibrary.GetSongs", {"filter": {"albumid": int(album_id)}, "properties":["file"]}))


def GetSongs():
  return SendCommand(RPCString("AudioLibrary.GetSongs"))


def GetSongsPath():
  return SendCommand(RPCString("AudioLibrary.GetSongs", {"properties":["file"]}))


def GetSongIdPath(song_id):
  return SendCommand(RPCString("AudioLibrary.GetSongDetails", {"songid": int(song_id), "properties":["file"]}))


def GetRecentlyAddedAlbums():
  return SendCommand(RPCString("AudioLibrary.GetRecentlyAddedAlbums", {'properties':['artist']}))


def GetRecentlyAddedSongs():
  return SendCommand(RPCString("AudioLibrary.GetRecentlyAddedSongs", {'properties':['artist']}))


def GetRecentlyAddedSongsPath():
  return SendCommand(RPCString("AudioLibrary.GetRecentlyAddedSongs", {'properties':['artist', 'file']}))


def GetVideoPlaylists():
  return SendCommand(RPCString("Files.GetDirectory", {"directory": "special://videoplaylists"}))


def GetTvShowDetails(show_id):
  data = SendCommand(RPCString("VideoLibrary.GetTVShowDetails", {'tvshowid':show_id, 'properties':['art']}))
  return data['result']['tvshowdetails']


def GetTvShows():
  return SendCommand(RPCString("VideoLibrary.GetTVShows"))


def GetMovieDetails(movie_id):
  data = SendCommand(RPCString("VideoLibrary.GetMovieDetails", {'movieid':movie_id, 'properties':['resume']}))
  return data['result']['moviedetails']


def GetMovies():
  return SendCommand(RPCString("VideoLibrary.GetMovies"))


def GetMoviesByGenre(genre):
  return SendCommand(RPCString("VideoLibrary.GetMovies", {"filter":{"genre":genre}}))


def GetMovieGenres():
  return SendCommand(RPCString("VideoLibrary.GetGenres", {"type": "movie"}))


def GetEpisodeDetails(ep_id):
  data = SendCommand(RPCString("VideoLibrary.GetEpisodeDetails", {"episodeid": int(ep_id), "properties":["season", "episode", "resume"]}))
  return data['result']['episodedetails']


def GetEpisodesFromShow(show_id):
  return SendCommand(RPCString("VideoLibrary.GetEpisodes", {"tvshowid": int(show_id)}))


def GetUnwatchedEpisodesFromShow(show_id):
  return SendCommand(RPCString("VideoLibrary.GetEpisodes", {"tvshowid": int(show_id), "filter":{"field":"playcount", "operator":"lessthan", "value":"1"}}))


def GetNewestEpisodeFromShow(show_id):
  data = SendCommand(RPCString("VideoLibrary.GetEpisodes", {"limits":{"end":1},"tvshowid": int(show_id), "sort":{"method":"dateadded", "order":"descending"}}))
  if 'episodes' in data['result']:
    episode = data['result']['episodes'][0]
    return episode['episodeid']
  else:
    return None


def GetNextUnwatchedEpisode(show_id):
  data = SendCommand(RPCString("VideoLibrary.GetEpisodes", {"limits":{"end":1},"tvshowid": int(show_id), "filter":{"field":"lastplayed", "operator":"greaterthan", "value":"0"}, "properties":["season", "episode", "lastplayed", "firstaired", "resume"], "sort":{"method":"lastplayed", "order":"descending"}}))
  if 'episodes' in data['result']:
    episode = data['result']['episodes'][0]
    episode_season = episode['season']
    episode_number = episode['episode']

    if episode['resume']['position'] > 0.0:
      next_episode_id = episode['episodeid']
    else:
      next_episode_id = GetSpecificEpisode(show_id, episode_season, int(episode_number) + 1)

    if next_episode_id:
      return next_episode_id
    else:
      next_episode_id = GetSpecificEpisode(show_id, int(episode_season) + 1, 1)

      if next_episode_id:
        return next_episode_id
      else:
        return None
  else:
    return None


def GetLastWatchedShow():
  return SendCommand(RPCString("VideoLibrary.GetEpisodes", {"limits":{"end":1}, "filter":{"field":"playcount", "operator":"greaterthan", "value":"0"}, "filter":{"field":"lastplayed", "operator":"greaterthan", "value":"0"}, "sort":{"method":"lastplayed", "order":"descending"}, "properties":["tvshowid", "showtitle"]}))


def GetSpecificEpisode(show_id, season, episode):
  data = SendCommand(RPCString("VideoLibrary.GetEpisodes", {"tvshowid": int(show_id), "season": int(season), "properties": ["season", "episode"]}))
  if 'episodes' in data['result']:
    correct_id = None
    for episode_data in data['result']['episodes']:
      if int(episode_data['episode']) == int(episode):
        correct_id = episode_data['episodeid']
        break

    return correct_id
  else:
    return None


def GetEpisodesFromShowDetails(show_id):
  return SendCommand(RPCString("VideoLibrary.GetEpisodes", {"tvshowid": int(show_id), "properties": ["season", "episode"]}))


# Returns a list of dictionaries with information about episodes that have been watched.
# May take a long time if you have lots of shows and you set max to a big number
def GetWatchedEpisodes(max=90):
  return SendCommand(RPCString("VideoLibrary.GetEpisodes", {"limits":{"end":max}, "filter":{"field":"playcount", "operator":"greaterthan", "value":"0"}, "properties":["playcount", "showtitle", "season", "episode", "lastplayed" ]}))


# Returns a list of dictionaries with information about unwatched movies. Useful for
# telling/showing users what's ready to be watched. Setting max to very high values
# can take a long time.
def GetUnwatchedMovies(max=90):
  data = SendCommand(RPCString("VideoLibrary.GetMovies", {"limits":{"end":max}, "filter":{"field":"playcount", "operator":"lessthan", "value":"1"}, "sort":{"method":"dateadded", "order":"descending"}, "properties":["title", "playcount", "dateadded"]}))
  answer = []
  for d in data['result']['movies']:
    answer.append({'title':d['title'], 'movieid':d['movieid'], 'label':d['label'], 'dateadded':datetime.datetime.strptime(d['dateadded'], "%Y-%m-%d %H:%M:%S")})
  return answer

# Returns a list of dictionaries with information about unwatched movies in a particular genre. Useful for
# telling/showing users what's ready to be watched. Setting max to very high values
# can take a long time.
def GetUnwatchedMoviesByGenre(genre, max=90):
  data = SendCommand(RPCString("VideoLibrary.GetMovies", {"limits":{"end":max}, "filter":{"and":[{"field":"playcount", "operator":"lessthan", "value":"1"}, {"field":"genre", "operator":"contains", "value":genre}]}, "sort":{"method":"dateadded", "order":"descending"}, "properties":["title", "playcount", "dateadded"]}))
  answer = []
  for d in data['result']['movies']:
    answer.append({'title':d['title'], 'movieid':d['movieid'], 'label':d['label'], 'dateadded':datetime.datetime.strptime(d['dateadded'], "%Y-%m-%d %H:%M:%S")})
  return answer


# Returns a list of dictionaries with information about unwatched episodes. Useful for
# telling/showing users what's ready to be watched. Setting max to very high values
# can take a long time.
def GetUnwatchedEpisodes(max=90):
  data = SendCommand(RPCString("VideoLibrary.GetEpisodes", {"limits":{"end":max}, "filter":{"field":"playcount", "operator":"lessthan", "value":"1"}, "sort":{"method":"dateadded", "order":"descending"}, "properties":["title", "playcount", "showtitle", "tvshowid", "dateadded" ]}))
  answer = []
  shows = set([d['tvshowid'] for d in data['result']['episodes']])
  show_info = {}
  for show in shows:
    show_info[show] = GetTvShowDetails(show_id=show)
  for d in data['result']['episodes']:
    showinfo = show_info[d['tvshowid']]
    answer.append({'title':d['title'], 'episodeid':d['episodeid'], 'show':d['showtitle'], 'label':d['label'], 'dateadded':datetime.datetime.strptime(d['dateadded'], "%Y-%m-%d %H:%M:%S")})
  return answer


# System commands
def ApplicationQuit():
  return SendCommand(RPCString("Application.Quit"))

def SystemHibernate():
  return SendCommand(RPCString("System.Hibernate"))

def SystemReboot():
  return SendCommand(RPCString("System.Reboot"))

def SystemShutdown():
  return SendCommand(RPCString("System.Shutdown"))

def SystemSuspend():
  return SendCommand(RPCString("System.Suspend"))

def SystemEjectMedia():
  return SendCommand(RPCString("System.EjectOpticalDrive"))


# Misc helpers

# Get the first active player.
def GetPlayerID(playertype=['picture', 'audio', 'video']):
  data = SendCommand(RPCString("Player.GetActivePlayers"))
  result = data.get("result", [])
  if len(result) > 0:
    for curitem in result:
      if curitem.get("type") in playertype:
        return curitem.get("playerid")
  return None

# Get the first active Video player.
def GetVideoPlayerID(playertype=['video']):
  data = SendCommand(RPCString("Player.GetActivePlayers"))
  result = data.get("result", [])
  if len(result) > 0:
    for curitem in result:
      if curitem.get("type") in playertype:
        return curitem.get("playerid")
  return None


# Get the first active Audio player.
def GetAudioPlayerID(playertype=['audio']):
  data = SendCommand(RPCString("Player.GetActivePlayers"))
  result = data.get("result", [])
  if len(result) > 0:
    for curitem in result:
      if curitem.get("type") in playertype:
        return curitem.get("playerid")
  return None


# Get the first active Picture player.
def GetPicturePlayerID(playertype=['picture']):
  data = SendCommand(RPCString("Player.GetActivePlayers"))
  result = data.get("result", [])
  if len(result) > 0:
    for curitem in result:
      if curitem.get("type") in playertype:
        return curitem.get("playerid")
  return None


# Information about the video or audio that's currently playing

def GetActivePlayItem():
  playerid = GetPlayerID()
  if playerid is not None:
    data = SendCommand(RPCString("Player.GetItem", {"playerid":playerid, "properties":["title", "album", "artist", "season", "episode", "showtitle", "tvshowid", "description"]}))
    #print data['result']['item']
    return data['result']['item']


def GetActivePlayProperties():
  playerid = GetPlayerID()
  if playerid is not None:
    data = SendCommand(RPCString("Player.GetProperties", {"playerid":playerid, "properties":["currentaudiostream", "currentsubtitle", "canshuffle", "shuffled", "canrepeat", "repeat", "canzoom", "canrotate", "canmove"]}))
    #print data['result']
    return data['result']


# Provide a map from ISO code (both bibliographic and terminologic)
# in ISO 639-2 to a dict with the two letter ISO 639-2 codes (alpha2)
# English and french names
#
# "bibliographic" iso codes are derived from English word for the language
# "terminologic" iso codes are derived from the pronunciation in the target
# language (if different to the bibliographic code)
#
# Source
# http://stackoverflow.com/questions/2879856/get-system-language-in-iso-639-3-letter-codes-in-python/2879958#2879958
#
# Usage
# country_dic = getisocodes_dict()
# print country_dic['eng']
def getisocodes_dict():
  D = {}
  country_dic_file = os.path.join(os.path.dirname(__file__), "ISO-639-2_utf-8.txt")
  f = codecs.open(country_dic_file, 'rb', 'utf-8')
  for line in f:
    iD = {}
    iD['bibliographic'], iD['terminologic'], iD['alpha2'], iD['english'], iD['french'] = line.encode("utf-8").strip().split('|')
    D[iD['bibliographic']] = iD

    if iD['terminologic']:
      D[iD['terminologic']] = iD

    if iD['alpha2']:
      D[iD['alpha2']] = iD

    for k in iD:
      # Assign `None` when columns not available from the data
      iD[k] = iD[k] or None
  f.close()
  return D


# Returns current subtitles as a speakable string
# XXXLANG: Only supports English currently!
def GetCurrentSubtitles():
  subs = ""
  country_dic = getisocodes_dict()
  curprops = GetActivePlayProperties()
  if curprops is not None:
    try:
      # gets 3 character country code e.g. fre
      lang = curprops['currentsubtitle']['language']
      # looks up 3 character code in the dictionary e.g. fre|fra|fr|French|francais
      subslang = country_dic[lang]
      # matches 3 character code with the english (4th) column e.g. French
      subs = subslang['english']
      # joins full language name with the name of the subtitle file e.g. French External
      name = curprops['currentsubtitle']['name']
      if name:
        subs += " " + name
    except:
      pass
  return subs


# Returns current audio stream as a speakable string
# XXXLANG: Only supports English currently!
def GetCurrentAudioStream():
  stream = ""
  country_dic = getisocodes_dict()
  curprops = GetActivePlayProperties()
  if curprops is not None:
    try:
      # gets 3 character country code e.g. fre
      lang = curprops['currentaudiostream']['language']
      # looks up 3 character code in the dictionary e.g. fre|fra|fr|French|francais
      streamlang = country_dic[lang]
      # matches 3 character code with the english (4th) column e.g. French
      stream = streamlang['english']
      # joins full language name with the name of the subtitle file e.g. French External
      name = curprops['currentaudiostream']['name']
      if name:
        stream += " " + name
    except:
      pass
  return stream


# Returns information useful for building a progress bar to show an item's play time
def GetPlayerStatus():
  playerid = GetVideoPlayerID()
  if playerid is None:
    playerid = GetAudioPlayerID()
  if playerid is not None:
    data = SendCommand(RPCString("Player.GetProperties", {"playerid":playerid, "properties":["percentage", "speed", "time", "totaltime"]}))
    if 'result' in data:
      hours_total = data['result']['totaltime']['hours']
      hours_cur = data['result']['time']['hours']
      mins_total = hours_total * 60 + data['result']['totaltime']['minutes']
      mins_cur = hours_cur * 60 + data['result']['time']['minutes']
      speed = data['result']['speed']
      if hours_total > 0:
        total = '%d:%02d:%02d' % (hours_total, data['result']['totaltime']['minutes'], data['result']['totaltime']['seconds'])
        cur = '%d:%02d:%02d' % (data['result']['time']['hours'], data['result']['time']['minutes'], data['result']['time']['seconds'])
      else:
        total = '%02d:%02d' % (data['result']['totaltime']['minutes'], data['result']['totaltime']['seconds'])
        cur = '%02d:%02d' % (data['result']['time']['minutes'], data['result']['time']['seconds'])
      return {'state':'play' if speed > 0 else 'pause', 'time':cur, 'time_hours':hours_cur, 'time_mins':mins_cur, 'totaltime':total, 'total_hours':hours_total, 'total_mins':mins_total, 'pct':data['result']['percentage']}
  return {'state':'stop'}