# -*- coding: utf-8 -*- ############################################################################### import logging from urllib import urlencode from ntpath import dirname from datetime import datetime from xbmc import sleep import artwork from utils import tryEncode, tryDecode, window, kodiSQL, CatchExceptions import plexdb_functions as plexdb import kodidb_functions as kodidb import PlexAPI from PlexFunctions import GetPlexMetadata import variables as v ############################################################################### log = logging.getLogger("PLEX."+__name__) MARK_PLAYED_AT = 0.90 ############################################################################### class Items(object): """ Items to be called with "with Items() as xxx:" to ensure that __enter__ method is called (opens db connections) Input: kodiType: optional argument; e.g. 'video' or 'music' """ def __init__(self): self.directpath = window('useDirectPaths') == 'true' self.artwork = artwork.Artwork() self.userid = window('currUserId') self.server = window('pms_server') def __enter__(self): """ Open DB connections and cursors """ self.plexconn = kodiSQL('plex') self.plexcursor = self.plexconn.cursor() self.kodiconn = kodiSQL('video') self.kodicursor = self.kodiconn.cursor() self.plex_db = plexdb.Plex_DB_Functions(self.plexcursor) self.kodi_db = kodidb.Kodidb_Functions(self.kodicursor) return self def __exit__(self, exc_type, exc_val, exc_tb): """ Make sure DB changes are committed and connection to DB is closed. """ self.plexconn.commit() self.kodiconn.commit() self.plexconn.close() self.kodiconn.close() return self @CatchExceptions(warnuser=True) def getfanart(self, plex_id, refresh=False): """ Tries to get additional fanart for movies (+sets) and TV shows. Returns True if successful, False otherwise """ with plexdb.Get_Plex_DB() as plex_db: db_item = plex_db.getItem_byId(plex_id) try: kodi_id = db_item[0] kodi_type = db_item[4] except TypeError: log.error('Could not get Kodi id for plex id %s, abort getfanart' % plex_id) return False if refresh is True: # Leave the Plex art untouched allartworks = None else: with kodidb.GetKodiDB('video') as kodi_db: allartworks = kodi_db.existingArt(kodi_id, kodi_type) # Check if we even need to get additional art needsupdate = False for key, value in allartworks.iteritems(): if not value and not key == 'BoxRear': needsupdate = True break if needsupdate is False: log.debug('Already got all fanart for Plex id %s' % plex_id) return True xml = GetPlexMetadata(plex_id) if xml is None: # Did not receive a valid XML - skip that item for now log.error("Could not get metadata for %s. Skipping that item " "for now" % plex_id) return False elif xml == 401: log.error('HTTP 401 returned by PMS. Too much strain? ' 'Cancelling sync for now') # Kill remaining items in queue (for main thread to cont.) return False API = PlexAPI.API(xml[0]) if allartworks is None: allartworks = API.getAllArtwork() self.artwork.addArtwork(API.getFanartArtwork(allartworks), kodi_id, kodi_type, self.kodicursor) # Also get artwork for collections/movie sets if kodi_type == v.KODI_TYPE_MOVIE: for setname in API.getCollections(): log.debug('Getting artwork for movie set %s' % setname) setid = self.kodi_db.createBoxset(setname) self.artwork.addArtwork(API.getSetArtwork(), setid, v.KODI_TYPE_SET, self.kodicursor) self.kodi_db.assignBoxset(setid, kodi_id) return True def updateUserdata(self, xml, viewtag=None, viewid=None): """ Updates the Kodi watched state of the item from PMS. Also retrieves Plex resume points for movies in progress. viewtag and viewid only serve as dummies """ for mediaitem in xml: API = PlexAPI.API(mediaitem) # Get key and db entry on the Kodi db side db_item = self.plex_db.getItem_byId(API.getRatingKey()) try: fileid = db_item[1] except TypeError: continue # Grab the user's viewcount, resume points etc. from PMS' answer userdata = API.getUserData() # Write to Kodi DB self.kodi_db.addPlaystate(fileid, userdata['Resume'], userdata['Runtime'], userdata['PlayCount'], userdata['LastPlayedDate']) if v.KODIVERSION >= 17: self.kodi_db.update_userrating(db_item[0], db_item[4], userdata['UserRating']) def updatePlaystate(self, item): """ Use with websockets, not xml """ # If the playback was stopped, check whether we need to increment the # playcount. PMS won't tell us the playcount via websockets if item['state'] in ('stopped', 'ended'): complete = float(item['viewOffset']) / float(item['duration']) log.info('Item %s stopped with completion rate %s percent.' 'Mark item played at %s percent.' % (item['ratingKey'], str(complete), MARK_PLAYED_AT), 1) if complete >= MARK_PLAYED_AT: log.info('Marking as completely watched in Kodi', 1) sleep(500) try: item['viewCount'] += 1 except TypeError: item['viewCount'] = 1 item['viewOffset'] = 0 # Do the actual update self.kodi_db.addPlaystate(item['file_id'], item['viewOffset'], item['duration'], item['viewCount'], item['lastViewedAt']) class Movies(Items): @CatchExceptions(warnuser=True) def add_update(self, item, viewtag=None, viewid=None): # Process single movie kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork API = PlexAPI.API(item) # If the item already exist in the local Kodi DB we'll perform a full # item update # If the item doesn't exist, we'll add it to the database update_item = True itemid = API.getRatingKey() # Cannot parse XML, abort if not itemid: log.error("Cannot parse XML data for movie") return plex_dbitem = plex_db.getItem_byId(itemid) try: movieid = plex_dbitem[0] fileid = plex_dbitem[1] pathid = plex_dbitem[2] except TypeError: # movieid update_item = False kodicursor.execute("select coalesce(max(idMovie),0) from movie") movieid = kodicursor.fetchone()[0] + 1 else: # Verification the item is still in Kodi query = "SELECT * FROM movie WHERE idMovie = ?" kodicursor.execute(query, (movieid,)) try: kodicursor.fetchone()[0] except TypeError: # item is not found, let's recreate it. update_item = False log.info("movieid: %s missing from Kodi, repairing the entry." % movieid) # fileId information checksum = API.getChecksum() dateadded = API.getDateCreated() userdata = API.getUserData() playcount = userdata['PlayCount'] dateplayed = userdata['LastPlayedDate'] resume = userdata['Resume'] runtime = userdata['Runtime'] # item details people = API.getPeople() writer = API.joinList(people['Writer']) director = API.joinList(people['Director']) genres = API.getGenres() genre = API.joinList(genres) title, sorttitle = API.getTitle() plot = API.getPlot() shortplot = None tagline = API.getTagline() votecount = None collections = API.getCollections() rating = userdata['Rating'] year = API.getYear() imdb = API.getProvider('imdb') mpaa = API.getMpaa() countries = API.getCountry() country = API.joinList(countries) studios = API.getStudios() try: studio = studios[0] except IndexError: studio = None # Find one trailer trailer = None extras = API.getExtras() for extra in extras: # Only get 1st trailer element if extra['extraType'] == 1: trailer = ("plugin://plugin.video.plexkodiconnect/trailer/?" "id=%s&mode=play") % extra['key'] break # GET THE FILE AND PATH ##### doIndirect = not self.directpath if self.directpath: # Direct paths is set the Kodi way playurl = API.getFilePath(forceFirstMediaStream=True) if playurl is None: # Something went wrong, trying to use non-direct paths doIndirect = True else: playurl = API.validatePlayurl(playurl, API.getType()) if playurl is None: return False if "\\" in playurl: # Local path filename = playurl.rsplit("\\", 1)[1] else: # Network share filename = playurl.rsplit("/", 1)[1] path = playurl.replace(filename, "") if doIndirect: # Set plugin path and media flags using real filename path = "plugin://plugin.video.plexkodiconnect/movies/" params = { 'filename': API.getKey(), 'id': itemid, 'dbid': movieid, 'mode': "play" } filename = "%s?%s" % (path, urlencode(params)) playurl = filename # movie table: # c22 - playurl # c23 - pathid # This information is used later by file browser. # add/retrieve pathid and fileid # if the path or file already exists, the calls return current value pathid = self.kodi_db.addPath(path) fileid = self.kodi_db.addFile(filename, pathid) # UPDATE THE MOVIE ##### if update_item: log.info("UPDATE movie itemid: %s - Title: %s" % (itemid, title)) # Update the movie entry if v.KODIVERSION >= 17: # update new ratings Kodi 17 rating_id = self.kodi_db.get_ratingid(movieid, v.KODI_TYPE_MOVIE) self.kodi_db.update_ratings(movieid, v.KODI_TYPE_MOVIE, "default", rating, votecount, rating_id) # update new uniqueid Kodi 17 uniqueid = self.kodi_db.get_uniqueid(movieid, v.KODI_TYPE_MOVIE) self.kodi_db.update_uniqueid(movieid, v.KODI_TYPE_MOVIE, imdb, "imdb", uniqueid) query = ''' UPDATE movie SET c00 = ?, c01 = ?, c02 = ?, c03 = ?, c04 = ?, c05 = ?, c06 = ?, c07 = ?, c09 = ?, c10 = ?, c11 = ?, c12 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, c21 = ?, c22 = ?, c23 = ?, idFile=?, premiered = ?, userrating = ? WHERE idMovie = ? ''' kodicursor.execute(query, (title, plot, shortplot, tagline, votecount, rating_id, writer, year, imdb, sorttitle, runtime, mpaa, genre, director, title, studio, trailer, country, playurl, pathid, fileid, year, userdata['UserRating'], movieid)) else: query = ''' UPDATE movie SET c00 = ?, c01 = ?, c02 = ?, c03 = ?, c04 = ?, c05 = ?, c06 = ?, c07 = ?, c09 = ?, c10 = ?, c11 = ?, c12 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, c21 = ?, c22 = ?, c23 = ?, idFile=? WHERE idMovie = ? ''' kodicursor.execute(query, (title, plot, shortplot, tagline, votecount, rating, writer, year, imdb, sorttitle, runtime, mpaa, genre, director, title, studio, trailer, country, playurl, pathid, fileid, movieid)) # OR ADD THE MOVIE ##### else: log.info("ADD movie itemid: %s - Title: %s" % (itemid, title)) if v.KODIVERSION >= 17: # add new ratings Kodi 17 rating_id = self.kodi_db.create_entry_rating() self.kodi_db.add_ratings(rating_id, movieid, v.KODI_TYPE_MOVIE, "default", rating, votecount) # add new uniqueid Kodi 17 self.kodi_db.add_uniqueid(self.kodi_db.create_entry_uniqueid(), movieid, v.KODI_TYPE_MOVIE, imdb, "imdb") query = ''' INSERT INTO movie(idMovie, idFile, c00, c01, c02, c03, c04, c05, c06, c07, c09, c10, c11, c12, c14, c15, c16, c18, c19, c21, c22, c23, premiered, userrating) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' kodicursor.execute(query, (movieid, fileid, title, plot, shortplot, tagline, votecount, rating_id, writer, year, imdb, sorttitle, runtime, mpaa, genre, director, title, studio, trailer, country, playurl, pathid, year, userdata['UserRating'])) else: query = ''' INSERT INTO movie(idMovie, idFile, c00, c01, c02, c03, c04, c05, c06, c07, c09, c10, c11, c12, c14, c15, c16, c18, c19, c21, c22, c23) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' kodicursor.execute(query, (movieid, fileid, title, plot, shortplot, tagline, votecount, rating, writer, year, imdb, sorttitle, runtime, mpaa, genre, director, title, studio, trailer, country, playurl, pathid)) # Create or update the reference in plex table Add reference is # idempotent; the call here updates also fileid and pathid when item is # moved or renamed plex_db.addReference(itemid, v.PLEX_TYPE_MOVIE, movieid, v.KODI_TYPE_MOVIE, kodi_fileid=fileid, kodi_pathid=pathid, parent_id=None, checksum=checksum, view_id=viewid) # Update the path query = ' '.join(( "UPDATE path", "SET strPath = ?, strContent = ?, strScraper = ?, noUpdate = ?", "WHERE idPath = ?" )) kodicursor.execute(query, (path, "movies", "metadata.local", 1, pathid)) # Update the file query = ' '.join(( "UPDATE files", "SET idPath = ?, strFilename = ?, dateAdded = ?", "WHERE idFile = ?" )) kodicursor.execute(query, (pathid, filename, dateadded, fileid)) # Process countries self.kodi_db.addCountries(movieid, countries, "movie") # Process cast self.kodi_db.addPeople(movieid, API.getPeopleList(), "movie") # Process genres self.kodi_db.addGenres(movieid, genres, "movie") # Process artwork artwork.addArtwork(API.getAllArtwork(), movieid, "movie", kodicursor) # Process stream details self.kodi_db.addStreams(fileid, API.getMediaStreams(), runtime) # Process studios self.kodi_db.addStudios(movieid, studios, "movie") # Process tags: view, Plex collection tags tags = [viewtag] tags.extend(collections) if userdata['Favorite']: tags.append("Favorite movies") self.kodi_db.addTags(movieid, tags, "movie") # Add any sets from Plex collection tags self.kodi_db.addSets(movieid, collections, kodicursor) # Process playstates self.kodi_db.addPlaystate(fileid, resume, runtime, playcount, dateplayed) def remove(self, itemid): # Remove movieid, fileid, plex reference plex_db = self.plex_db kodicursor = self.kodicursor artwork = self.artwork plex_dbitem = plex_db.getItem_byId(itemid) try: kodi_id = plex_dbitem[0] file_id = plex_dbitem[1] kodi_type = plex_dbitem[4] log.info("Removing %sid: %s file_id: %s" % (kodi_type, kodi_id, file_id)) except TypeError: return # Remove the plex reference plex_db.removeItem(itemid) # Remove artwork artwork.deleteArtwork(kodi_id, kodi_type, kodicursor) if kodi_type == v.KODI_TYPE_MOVIE: # Delete kodi movie and file kodicursor.execute("DELETE FROM movie WHERE idMovie = ?", (kodi_id,)) kodicursor.execute("DELETE FROM files WHERE idFile = ?", (file_id,)) if v.KODIVERSION >= 17: self.kodi_db.remove_uniqueid(kodi_id, kodi_type) self.kodi_db.remove_ratings(kodi_id, kodi_type) elif kodi_type == v.KODI_TYPE_SET: # Delete kodi boxset boxset_movies = plex_db.getItem_byParentId(kodi_id, v.KODI_TYPE_MOVIE) for movie in boxset_movies: plexid = movie[0] movieid = movie[1] self.kodi_db.removefromBoxset(movieid) # Update plex reference plex_db.updateParentId(plexid, None) kodicursor.execute("DELETE FROM sets WHERE idSet = ?", (kodi_id,)) log.info("Deleted %s %s from kodi database" % (kodi_type, itemid)) class TVShows(Items): @CatchExceptions(warnuser=True) def add_update(self, item, viewtag=None, viewid=None): # Process single tvshow kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork API = PlexAPI.API(item) update_item = True itemid = API.getRatingKey() if not itemid: log.error("Cannot parse XML data for TV show") return update_item = True force_episodes = False plex_dbitem = plex_db.getItem_byId(itemid) try: showid = plex_dbitem[0] pathid = plex_dbitem[2] except TypeError: update_item = False kodicursor.execute("select coalesce(max(idShow),0) from tvshow") showid = kodicursor.fetchone()[0] + 1 else: # Verification the item is still in Kodi query = "SELECT * FROM tvshow WHERE idShow = ?" kodicursor.execute(query, (showid,)) try: kodicursor.fetchone()[0] except TypeError: # item is not found, let's recreate it. update_item = False log.info("showid: %s missing from Kodi, repairing the entry." % showid) # Force re-add episodes after the show is re-created. force_episodes = True # fileId information checksum = API.getChecksum() # item details genres = API.getGenres() title, sorttitle = API.getTitle() plot = API.getPlot() rating = API.getAudienceRating() votecount = None premieredate = API.getPremiereDate() tvdb = API.getProvider('tvdb') mpaa = API.getMpaa() genre = API.joinList(genres) studios = API.getStudios() collections = API.getCollections() try: studio = studios[0] except IndexError: studio = None # GET THE FILE AND PATH ##### doIndirect = not self.directpath if self.directpath: # Direct paths is set the Kodi way playurl = API.getTVShowPath() if playurl is None: # Something went wrong, trying to use non-direct paths doIndirect = True else: playurl = API.validatePlayurl(playurl, API.getType(), folder=True) if playurl is None: return False if "\\" in playurl: # Local path path = "%s\\" % playurl toplevelpath = "%s\\" % dirname(dirname(path)) else: # Network path path = "%s/" % playurl toplevelpath = "%s/" % dirname(dirname(path)) if doIndirect: # Set plugin path toplevelpath = "plugin://plugin.video.plexkodiconnect/tvshows/" path = "%s%s/" % (toplevelpath, itemid) # Add top path toppathid = self.kodi_db.addPath(toplevelpath) # add/retrieve pathid and fileid # if the path or file already exists, the calls return current value pathid = self.kodi_db.addPath(path) # UPDATE THE TVSHOW ##### if update_item: log.info("UPDATE tvshow itemid: %s - Title: %s" % (itemid, title)) # Add reference is idempotent; the call here updates also fileid # and pathid when item is moved or renamed plex_db.addReference(itemid, v.PLEX_TYPE_SHOW, showid, v.KODI_TYPE_SHOW, kodi_pathid=pathid, checksum=checksum, view_id=viewid) if v.KODIVERSION >= 17: # update new ratings Kodi 17 rating_id = self.kodi_db.get_ratingid(showid, v.KODI_TYPE_SHOW) self.kodi_db.update_ratings(showid, v.KODI_TYPE_SHOW, "default", rating, votecount, rating_id) # update new uniqueid Kodi 17 uniqueid = self.kodi_db.get_uniqueid(showid, v.KODI_TYPE_SHOW) self.kodi_db.update_uniqueid(showid, v.KODI_TYPE_SHOW, tvdb, "tvdb", uniqueid) # Update the tvshow entry query = ''' UPDATE tvshow SET c00 = ?, c01 = ?, c04 = ?, c05 = ?, c08 = ?, c09 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ? WHERE idShow = ? ''' kodicursor.execute(query, (title, plot, rating_id, premieredate, genre, title, tvdb, mpaa, studio, sorttitle, showid)) else: # Update the tvshow entry query = ''' UPDATE tvshow SET c00 = ?, c01 = ?, c04 = ?, c05 = ?, c08 = ?, c09 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ? WHERE idShow = ? ''' kodicursor.execute(query, (title, plot, rating, premieredate, genre, title, tvdb, mpaa, studio, sorttitle, showid)) # OR ADD THE TVSHOW ##### else: log.info("ADD tvshow itemid: %s - Title: %s" % (itemid, title)) query = ''' UPDATE path SET strPath = ?, strContent = ?, strScraper = ?, noUpdate = ? WHERE idPath = ? ''' kodicursor.execute(query, (toplevelpath, "tvshows", "metadata.local", 1, toppathid)) # Link the path query = "INSERT INTO tvshowlinkpath(idShow, idPath) values (?, ?)" kodicursor.execute(query, (showid, pathid)) # Create the reference in plex table plex_db.addReference(itemid, v.PLEX_TYPE_SHOW, showid, v.KODI_TYPE_SHOW, kodi_pathid=pathid, checksum=checksum, view_id=viewid) if v.KODIVERSION >= 17: # add new ratings Kodi 17 rating_id = self.kodi_db.create_entry_rating() self.kodi_db.add_ratings(rating_id, showid, v.KODI_TYPE_SHOW, "default", rating, votecount) # add new uniqueid Kodi 17 self.kodi_db.add_uniqueid(self.kodi_db.create_entry_uniqueid(), showid, v.KODI_TYPE_SHOW, tvdb, "tvdb") # Create the tvshow entry query = ''' INSERT INTO tvshow( idShow, c00, c01, c04, c05, c08, c09, c12, c13, c14, c15) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' kodicursor.execute(query, (showid, title, plot, rating_id, premieredate, genre, title, tvdb, mpaa, studio, sorttitle)) else: # Create the tvshow entry query = ''' INSERT INTO tvshow( idShow, c00, c01, c04, c05, c08, c09, c12, c13, c14, c15) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' kodicursor.execute(query, (showid, title, plot, rating, premieredate, genre, title, tvdb, mpaa, studio, sorttitle)) # Update the path query = ''' UPDATE path SET strPath = ?, strContent = ?, strScraper = ?, noUpdate = ?, idParentPath = ? WHERE idPath = ? ''' kodicursor.execute(query, (path, None, None, 1, toppathid, pathid)) # Process cast people = API.getPeopleList() self.kodi_db.addPeople(showid, people, "tvshow") # Process genres self.kodi_db.addGenres(showid, genres, "tvshow") # Process artwork allartworks = API.getAllArtwork() artwork.addArtwork(allartworks, showid, "tvshow", kodicursor) # Process studios self.kodi_db.addStudios(showid, studios, "tvshow") # Process tags: view, PMS collection tags tags = [viewtag] tags.extend(collections) self.kodi_db.addTags(showid, tags, "tvshow") @CatchExceptions(warnuser=True) def add_updateSeason(self, item, viewtag=None, viewid=None): API = PlexAPI.API(item) plex_id = API.getRatingKey() if not plex_id: log.error('Error getting plex_id for season, skipping') return kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork seasonnum = API.getIndex() # Get parent tv show Plex id plexshowid = item.attrib.get('parentRatingKey') # Get Kodi showid plex_dbitem = plex_db.getItem_byId(plexshowid) try: showid = plex_dbitem[0] except: log.error('Could not find parent tv show for season %s. ' 'Skipping season for now.' % (plex_id)) return seasonid = self.kodi_db.addSeason(showid, seasonnum) checksum = API.getChecksum() # Check whether Season already exists update_item = True plex_dbitem = plex_db.getItem_byId(plex_id) try: plexdbItemId = plex_dbitem[0] except TypeError: update_item = False # Process artwork allartworks = API.getAllArtwork() artwork.addArtwork(allartworks, seasonid, "season", kodicursor) if update_item: # Update a reference: checksum in plex table plex_db.updateReference(plex_id, checksum) else: # Create the reference in plex table plex_db.addReference(plex_id, v.PLEX_TYPE_SEASON, seasonid, v.KODI_TYPE_SEASON, parent_id=showid, view_id=viewid, checksum=checksum) @CatchExceptions(warnuser=True) def add_updateEpisode(self, item, viewtag=None, viewid=None): """ """ # Process single episode kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork API = PlexAPI.API(item) # If the item already exist in the local Kodi DB we'll perform a full # item update # If the item doesn't exist, we'll add it to the database update_item = True itemid = API.getRatingKey() if not itemid: log.error('Error getting itemid for episode, skipping') return plex_dbitem = plex_db.getItem_byId(itemid) try: episodeid = plex_dbitem[0] fileid = plex_dbitem[1] pathid = plex_dbitem[2] except TypeError: update_item = False # episodeid kodicursor.execute("select coalesce(max(idEpisode),0) from episode") episodeid = kodicursor.fetchone()[0] + 1 else: # Verification the item is still in Kodi query = "SELECT * FROM episode WHERE idEpisode = ?" kodicursor.execute(query, (episodeid,)) try: kodicursor.fetchone()[0] except TypeError: # item is not found, let's recreate it. update_item = False log.info("episodeid: %s missing from Kodi, repairing entry." % episodeid) # fileId information checksum = API.getChecksum() dateadded = API.getDateCreated() userdata = API.getUserData() playcount = userdata['PlayCount'] dateplayed = userdata['LastPlayedDate'] tvdb = API.getProvider('tvdb') votecount = None # item details peoples = API.getPeople() director = API.joinList(peoples['Director']) writer = API.joinList(peoples['Writer']) title, sorttitle = API.getTitle() plot = API.getPlot() rating = userdata['Rating'] resume, runtime = API.getRuntime() premieredate = API.getPremiereDate() # episode details seriesId, seriesName, season, episode = API.getEpisodeDetails() if season is None: season = -1 if episode is None: episode = -1 # if item.get('AbsoluteEpisodeNumber'): # # Anime scenario # season = 1 # episode = item['AbsoluteEpisodeNumber'] # else: # season = -1 # Specials ordering within season if item.get('AirsAfterSeasonNumber'): airsBeforeSeason = item['AirsAfterSeasonNumber'] # Kodi default number for afterseason ordering airsBeforeEpisode = 4096 else: airsBeforeSeason = item.get('AirsBeforeSeasonNumber') airsBeforeEpisode = item.get('AirsBeforeEpisodeNumber') airsBeforeSeason = "-1" airsBeforeEpisode = "-1" # Append multi episodes to title # if item.get('IndexNumberEnd'): # title = "| %02d | %s" % (item['IndexNumberEnd'], title) # Get season id show = plex_db.getItem_byId(seriesId) try: showid = show[0] except TypeError: log.error("Parent tvshow now found, skip item") return False seasonid = self.kodi_db.addSeason(showid, season) # GET THE FILE AND PATH ##### doIndirect = not self.directpath playurl = API.getFilePath(forceFirstMediaStream=True) if self.directpath: # Direct paths is set the Kodi way if playurl is None: # Something went wrong, trying to use non-direct paths doIndirect = True else: playurl = API.validatePlayurl(playurl, API.getType()) if playurl is None: return False if "\\" in playurl: # Local path filename = playurl.rsplit("\\", 1)[1] else: # Network share filename = playurl.rsplit("/", 1)[1] path = playurl.replace(filename, "") parentPathId = self.kodi_db.getParentPathId(path) if doIndirect: # Set plugin path and media flags using real filename if playurl is not None: if '\\' in playurl: filename = playurl.rsplit('\\', 1)[1] else: filename = playurl.rsplit('/', 1)[1] else: filename = 'file_not_found.mkv' path = "plugin://plugin.video.plexkodiconnect/tvshows/%s/" % seriesId params = { 'filename': tryEncode(filename), 'id': itemid, 'dbid': episodeid, 'mode': "play" } filename = "%s?%s" % (path, tryDecode(urlencode(params))) playurl = filename parentPathId = self.kodi_db.addPath( 'plugin://plugin.video.plexkodiconnect/tvshows/') # episodes table: # c18 - playurl # c19 - pathid # This information is used later by file browser. # add/retrieve pathid and fileid # if the path or file already exists, the calls return current value pathid = self.kodi_db.addPath(path) fileid = self.kodi_db.addFile(filename, pathid) # UPDATE THE EPISODE ##### if update_item: log.info("UPDATE episode itemid: %s" % (itemid)) # Update the movie entry if v.KODIVERSION >= 17: # update new ratings Kodi 17 ratingid = self.kodi_db.get_ratingid(episodeid, v.KODI_TYPE_EPISODE) self.kodi_db.update_ratings(episodeid, v.KODI_TYPE_EPISODE, "default", rating, votecount, ratingid) # update new uniqueid Kodi 17 uniqueid = self.kodi_db.get_uniqueid(episodeid, v.KODI_TYPE_EPISODE) self.kodi_db.update_uniqueid(episodeid, v.KODI_TYPE_EPISODE, tvdb, "tvdb", uniqueid) query = ''' UPDATE episode SET c00 = ?, c01 = ?, c03 = ?, c04 = ?, c05 = ?, c09 = ?, c10 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, idFile=?, idSeason = ?, userrating = ? WHERE idEpisode = ? ''' kodicursor.execute(query, (title, plot, rating, writer, premieredate, runtime, director, season, episode, title, airsBeforeSeason, airsBeforeEpisode, playurl, pathid, fileid, seasonid, userdata['UserRating'], episodeid)) elif v.KODIVERSION == 16: # Kodi Jarvis query = ''' UPDATE episode SET c00 = ?, c01 = ?, c03 = ?, c04 = ?, c05 = ?, c09 = ?, c10 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, idFile=?, idSeason = ? WHERE idEpisode = ? ''' kodicursor.execute(query, (title, plot, rating, writer, premieredate, runtime, director, season, episode, title, airsBeforeSeason, airsBeforeEpisode, playurl, pathid, fileid, seasonid, episodeid)) else: query = ''' UPDATE episode SET c00 = ?, c01 = ?, c03 = ?, c04 = ?, c05 = ?, c09 = ?, c10 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, idFile = ? WHERE idEpisode = ? ''' kodicursor.execute(query, (title, plot, rating, writer, premieredate, runtime, director, season, episode, title, airsBeforeSeason, airsBeforeEpisode, playurl, pathid, fileid, episodeid)) # Update parentid reference plex_db.updateParentId(itemid, seasonid) # OR ADD THE EPISODE ##### else: log.info("ADD episode itemid: %s - Title: %s" % (itemid, title)) # Create the episode entry if v.KODIVERSION >= 17: # add new ratings Kodi 17 rating_id = self.kodi_db.create_entry_rating() self.kodi_db.add_ratings(rating_id, episodeid, v.KODI_TYPE_EPISODE, "default", rating, votecount) # add new uniqueid Kodi 17 self.kodi_db.add_uniqueid(self.kodi_db.create_entry_uniqueid(), episodeid, v.KODI_TYPE_EPISODE, tvdb, "tvdb") query = ''' INSERT INTO episode( idEpisode, idFile, c00, c01, c03, c04, c05, c09, c10, c12, c13, c14, idShow, c15, c16, c18, c19, idSeason, userrating) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' kodicursor.execute(query, (episodeid, fileid, title, plot, rating_id, writer, premieredate, runtime, director, season, episode, title, showid, airsBeforeSeason, airsBeforeEpisode, playurl, pathid, seasonid, userdata['UserRating'])) elif v.KODIVERSION == 16: # Kodi Jarvis query = ''' INSERT INTO episode( idEpisode, idFile, c00, c01, c03, c04, c05, c09, c10, c12, c13, c14, idShow, c15, c16, c18, c19, idSeason) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' kodicursor.execute(query, (episodeid, fileid, title, plot, rating, writer, premieredate, runtime, director, season, episode, title, showid, airsBeforeSeason, airsBeforeEpisode, playurl, pathid, seasonid)) else: query = ( ''' INSERT INTO episode( idEpisode, idFile, c00, c01, c03, c04, c05, c09, c10, c12, c13, c14, idShow, c15, c16, c18, c19) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' ) kodicursor.execute(query, (episodeid, fileid, title, plot, rating, writer, premieredate, runtime, director, season, episode, title, showid, airsBeforeSeason, airsBeforeEpisode, playurl, pathid)) # Create or update the reference in plex table Add reference is # idempotent; the call here updates also fileid and pathid when item is # moved or renamed plex_db.addReference(itemid, v.PLEX_TYPE_EPISODE, episodeid, v.KODI_TYPE_EPISODE, kodi_fileid=fileid, kodi_pathid=pathid, parent_id=seasonid, checksum=checksum, view_id=viewid) # Update the path query = ' '.join(( "UPDATE path", "SET strPath = ?, strContent = ?, strScraper = ?, noUpdate = ?, ", "idParentPath = ?" "WHERE idPath = ?" )) kodicursor.execute(query, (path, None, None, 1, parentPathId, pathid)) # Update the file query = ' '.join(( "UPDATE files", "SET idPath = ?, strFilename = ?, dateAdded = ?", "WHERE idFile = ?" )) kodicursor.execute(query, (pathid, filename, dateadded, fileid)) # Process cast people = API.getPeopleList() self.kodi_db.addPeople(episodeid, people, "episode") # Process artwork # Wide "screenshot" of particular episode poster = item.attrib.get('thumb') if poster: poster = API.addPlexCredentialsToUrl( "%s%s" % (self.server, poster)) artwork.addOrUpdateArt( poster, episodeid, "episode", "thumb", kodicursor) # poster of TV show itself # poster = item.attrib.get('grandparentThumb') # if poster: # poster = API.addPlexCredentialsToUrl( # "%s%s" % (self.server, poster)) # artwork.addOrUpdateArt( # poster, episodeid, "episode", "poster", kodicursor) # Process stream details streams = API.getMediaStreams() self.kodi_db.addStreams(fileid, streams, runtime) # Process playstates self.kodi_db.addPlaystate(fileid, resume, runtime, playcount, dateplayed) if not self.directpath and resume: # Create additional entry for widgets. This is only required for plugin/episode. temppathid = self.kodi_db.getPath("plugin://plugin.video.plexkodiconnect/tvshows/") tempfileid = self.kodi_db.addFile(filename, temppathid) query = ' '.join(( "UPDATE files", "SET idPath = ?, strFilename = ?, dateAdded = ?", "WHERE idFile = ?" )) kodicursor.execute(query, (temppathid, filename, dateadded, tempfileid)) self.kodi_db.addPlaystate(tempfileid, resume, runtime, playcount, dateplayed) def remove(self, itemid): # Remove showid, fileid, pathid, plex reference plex_db = self.plex_db kodicursor = self.kodicursor plex_dbitem = plex_db.getItem_byId(itemid) try: kodiid = plex_dbitem[0] fileid = plex_dbitem[1] parentid = plex_dbitem[3] mediatype = plex_dbitem[4] log.info("Removing %s kodiid: %s fileid: %s" % (mediatype, kodiid, fileid)) except TypeError: return ##### PROCESS ITEM ##### # Remove the plex reference plex_db.removeItem(itemid) ##### IF EPISODE ##### if mediatype == v.KODI_TYPE_EPISODE: # Delete kodi episode and file, verify season and tvshow self.removeEpisode(kodiid, fileid) # Season verification season = plex_db.getItem_byKodiId(parentid, v.KODI_TYPE_SEASON) try: showid = season[1] except TypeError: return season_episodes = plex_db.getItem_byParentId(parentid, v.KODI_TYPE_EPISODE) if not season_episodes: self.removeSeason(parentid) plex_db.removeItem(season[0]) # Show verification show = plex_db.getItem_byKodiId(showid, v.KODI_TYPE_SHOW) query = ' '.join(( "SELECT totalCount", "FROM tvshowcounts", "WHERE idShow = ?" )) kodicursor.execute(query, (showid,)) result = kodicursor.fetchone() if result and result[0] is None: # There's no episodes left, delete show and any possible remaining seasons seasons = plex_db.getItem_byParentId(showid, v.KODI_TYPE_SEASON) for season in seasons: self.removeSeason(season[1]) else: # Delete plex season entries plex_db.removeItems_byParentId(showid, v.KODI_TYPE_SEASON) self.removeShow(showid) plex_db.removeItem(show[0]) ##### IF TVSHOW ##### elif mediatype == v.KODI_TYPE_SHOW: # Remove episodes, seasons, tvshow seasons = plex_db.getItem_byParentId(kodiid, v.KODI_TYPE_SEASON) for season in seasons: seasonid = season[1] season_episodes = plex_db.getItem_byParentId( seasonid, v.KODI_TYPE_EPISODE) for episode in season_episodes: self.removeEpisode(episode[1], episode[2]) else: # Remove plex episodes plex_db.removeItems_byParentId(seasonid, v.KODI_TYPE_EPISODE) else: # Remove plex seasons plex_db.removeItems_byParentId(kodiid, v.KODI_TYPE_SEASON) # Remove tvshow self.removeShow(kodiid) ##### IF SEASON ##### elif mediatype == v.KODI_TYPE_SEASON: # Remove episodes, season, verify tvshow season_episodes = plex_db.getItem_byParentId(kodiid, v.KODI_TYPE_EPISODE) for episode in season_episodes: self.removeEpisode(episode[1], episode[2]) else: # Remove plex episodes plex_db.removeItems_byParentId(kodiid, v.KODI_TYPE_EPISODE) # Remove season self.removeSeason(kodiid) # Show verification seasons = plex_db.getItem_byParentId(parentid, v.KODI_TYPE_SEASON) if not seasons: # There's no seasons, delete the show self.removeShow(parentid) plex_db.removeItem_byKodiId(parentid, v.KODI_TYPE_SHOW) log.debug("Deleted %s: %s from kodi database" % (mediatype, itemid)) def removeShow(self, kodi_id): kodicursor = self.kodicursor self.artwork.deleteArtwork(kodi_id, v.KODI_TYPE_SHOW, kodicursor) kodicursor.execute("DELETE FROM tvshow WHERE idShow = ?", (kodi_id,)) if v.KODIVERSION >= 17: self.kodi_db.remove_uniqueid(kodi_id, v.KODI_TYPE_SHOW) self.kodi_db.remove_ratings(kodi_id, v.KODI_TYPE_SHOW) log.info("Removed tvshow: %s." % kodi_id) def removeSeason(self, kodi_id): kodicursor = self.kodicursor self.artwork.deleteArtwork(kodi_id, "season", kodicursor) kodicursor.execute("DELETE FROM seasons WHERE idSeason = ?", (kodi_id,)) log.info("Removed season: %s." % kodi_id) def removeEpisode(self, kodi_id, fileid): kodicursor = self.kodicursor self.artwork.deleteArtwork(kodi_id, "episode", kodicursor) kodicursor.execute("DELETE FROM episode WHERE idEpisode = ?", (kodi_id,)) kodicursor.execute("DELETE FROM files WHERE idFile = ?", (fileid,)) if v.KODIVERSION >= 17: self.kodi_db.remove_uniqueid(kodi_id, v.KODI_TYPE_EPISODE) self.kodi_db.remove_ratings(kodi_id, v.KODI_TYPE_EPISODE) log.info("Removed episode: %s." % kodi_id) class Music(Items): def __enter__(self): """ OVERWRITE this method, because we need to open another DB. Open DB connections and cursors """ self.plexconn = kodiSQL('plex') self.plexcursor = self.plexconn.cursor() # Here it is, not 'video' but 'music' self.kodiconn = kodiSQL('music') self.kodicursor = self.kodiconn.cursor() self.plex_db = plexdb.Plex_DB_Functions(self.plexcursor) self.kodi_db = kodidb.Kodidb_Functions(self.kodicursor) return self @CatchExceptions(warnuser=True) def add_updateArtist(self, item, viewtag=None, viewid=None, artisttype="MusicArtist"): kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork API = PlexAPI.API(item) update_item = True itemid = API.getRatingKey() plex_dbitem = plex_db.getItem_byId(itemid) try: artistid = plex_dbitem[0] except TypeError: update_item = False # The artist details ##### lastScraped = datetime.now().strftime('%Y-%m-%d %H:%M:%S') dateadded = API.getDateCreated() checksum = API.getChecksum() name, sortname = API.getTitle() # musicBrainzId = API.getProvider('MusicBrainzArtist') musicBrainzId = None genres = API.joinList(API.getGenres()) bio = API.getPlot() # Associate artwork artworks = API.getAllArtwork(parentInfo=True) thumb = artworks['Primary'] backdrops = artworks['Backdrop'] # List if thumb: thumb = "%s" % thumb if backdrops: fanart = "%s" % backdrops[0] else: fanart = "" # UPDATE THE ARTIST ##### if update_item: log.info("UPDATE artist itemid: %s - Name: %s" % (itemid, name)) # Update the checksum in plex table plex_db.updateReference(itemid, checksum) # OR ADD THE ARTIST ##### else: log.info("ADD artist itemid: %s - Name: %s" % (itemid, name)) # safety checks: It looks like plex supports the same artist # multiple times. # Kodi doesn't allow that. In case that happens we just merge the # artist entries. artistid = self.kodi_db.addArtist(name, musicBrainzId) # Create the reference in plex table plex_db.addReference(itemid, v.PLEX_TYPE_ARTIST, artistid, v.KODI_TYPE_ARTIST, view_id=viewid, checksum=checksum) # Process the artist if v.KODIVERSION >= 16: query = ' '.join(( "UPDATE artist", "SET strGenres = ?, strBiography = ?, strImage = ?, strFanart = ?,", "lastScraped = ?", "WHERE idArtist = ?" )) kodicursor.execute(query, (genres, bio, thumb, fanart, lastScraped, artistid)) else: query = ' '.join(( "UPDATE artist", "SET strGenres = ?, strBiography = ?, strImage = ?, strFanart = ?,", "lastScraped = ?, dateAdded = ?", "WHERE idArtist = ?" )) kodicursor.execute(query, (genres, bio, thumb, fanart, lastScraped, dateadded, artistid)) # Update artwork artwork.addArtwork(artworks, artistid, "artist", kodicursor) @CatchExceptions(warnuser=True) def add_updateAlbum(self, item, viewtag=None, viewid=None): kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork API = PlexAPI.API(item) update_item = True itemid = API.getRatingKey() if not itemid: log.error('Error processing Album, skipping') return plex_dbitem = plex_db.getItem_byId(itemid) try: albumid = plex_dbitem[0] except TypeError: # Albumid not found update_item = False # The album details ##### lastScraped = datetime.now().strftime('%Y-%m-%d %H:%M:%S') dateadded = API.getDateCreated() userdata = API.getUserData() checksum = API.getChecksum() name, sorttitle = API.getTitle() # musicBrainzId = API.getProvider('MusicBrainzAlbum') musicBrainzId = None year = API.getYear() genres = API.getGenres() genre = API.joinList(genres) bio = API.getPlot() rating = userdata['UserRating'] studio = API.getMusicStudio() # artists = item['AlbumArtists'] # if not artists: # artists = item['ArtistItems'] # artistname = [] # for artist in artists: # artistname.append(artist['Name']) artistname = item.attrib.get('parentTitle') if not artistname: artistname = item.attrib.get('originalTitle') # Associate artwork artworks = API.getAllArtwork(parentInfo=True) thumb = artworks['Primary'] if thumb: thumb = "%s" % thumb # UPDATE THE ALBUM ##### if update_item: log.info("UPDATE album itemid: %s - Name: %s" % (itemid, name)) # Update the checksum in plex table plex_db.updateReference(itemid, checksum) # OR ADD THE ALBUM ##### else: log.info("ADD album itemid: %s - Name: %s" % (itemid, name)) # safety checks: It looks like plex supports the same artist # multiple times. # Kodi doesn't allow that. In case that happens we just merge the # artist entries. albumid = self.kodi_db.addAlbum(name, musicBrainzId) # Create the reference in plex table plex_db.addReference(itemid, v.PLEX_TYPE_ALBUM, albumid, v.KODI_TYPE_ALBUM, view_id=viewid, checksum=checksum) # Process the album info if v.KODIVERSION >= 17: # Kodi Krypton query = ' '.join(( "UPDATE album", "SET strArtists = ?, iYear = ?, strGenres = ?, strReview = ?, strImage = ?,", "iUserrating = ?, lastScraped = ?, strReleaseType = ?, " "strLabel = ? ", "WHERE idAlbum = ?" )) kodicursor.execute(query, (artistname, year, genre, bio, thumb, rating, lastScraped, "album", studio, albumid)) elif v.KODIVERSION == 16: # Kodi Jarvis query = ' '.join(( "UPDATE album", "SET strArtists = ?, iYear = ?, strGenres = ?, strReview = ?, strImage = ?,", "iRating = ?, lastScraped = ?, strReleaseType = ?, " "strLabel = ? ", "WHERE idAlbum = ?" )) kodicursor.execute(query, (artistname, year, genre, bio, thumb, rating, lastScraped, "album", studio, albumid)) elif v.KODIVERSION == 15: # Kodi Isengard query = ' '.join(( "UPDATE album", "SET strArtists = ?, iYear = ?, strGenres = ?, strReview = ?, strImage = ?,", "iRating = ?, lastScraped = ?, dateAdded = ?, " "strReleaseType = ?, strLabel = ? ", "WHERE idAlbum = ?" )) kodicursor.execute(query, (artistname, year, genre, bio, thumb, rating, lastScraped, dateadded, "album", studio, albumid)) else: # Kodi Helix query = ' '.join(( "UPDATE album", "SET strArtists = ?, iYear = ?, strGenres = ?, strReview = ?, strImage = ?,", "iRating = ?, lastScraped = ?, dateAdded = ?, " "strLabel = ? ", "WHERE idAlbum = ?" )) kodicursor.execute(query, (artistname, year, genre, bio, thumb, rating, lastScraped, dateadded, studio, albumid)) # Associate the parentid for plex reference parentId = item.attrib.get('parentRatingKey') if parentId is not None: plex_dbartist = plex_db.getItem_byId(parentId) try: artistid = plex_dbartist[0] except TypeError: log.info('Artist %s does not exist in plex database' % parentId) artist = GetPlexMetadata(parentId) # Item may not be an artist, verification necessary. if artist is not None and artist != 401: if artist[0].attrib.get('type') == "artist": # Update with the parentId, for remove reference plex_db.addReference(parentId, v.PLEX_TYPE_ARTIST, parentId, v.KODI_TYPE_ARTIST, view_id=viewid) plex_db.updateParentId(itemid, parentId) else: # Update plex reference with the artistid plex_db.updateParentId(itemid, artistid) # Assign main artists to album # Plex unfortunately only supports 1 artist :-( artistId = parentId plex_dbartist = plex_db.getItem_byId(artistId) try: artistid = plex_dbartist[0] except TypeError: # Artist does not exist in plex database, create the reference log.info('Artist %s does not exist in Plex database' % artistId) artist = GetPlexMetadata(artistId) if artist is not None and artist != 401: self.add_updateArtist(artist[0], artisttype="AlbumArtist") plex_dbartist = plex_db.getItem_byId(artistId) artistid = plex_dbartist[0] else: # Best take this name over anything else. query = "UPDATE artist SET strArtist = ? WHERE idArtist = ?" kodicursor.execute(query, (artistname, artistid,)) log.info("UPDATE artist: strArtist: %s, idArtist: %s" % (artistname, artistid)) # Add artist to album query = ( ''' INSERT OR REPLACE INTO album_artist(idArtist, idAlbum, strArtist) VALUES (?, ?, ?) ''' ) kodicursor.execute(query, (artistid, albumid, artistname)) # Update discography query = ( ''' INSERT OR REPLACE INTO discography(idArtist, strAlbum, strYear) VALUES (?, ?, ?) ''' ) kodicursor.execute(query, (artistid, name, year)) # Update plex reference with parentid plex_db.updateParentId(artistId, albumid) # Add genres self.kodi_db.addMusicGenres(albumid, genres, "album") # Update artwork artwork.addArtwork(artworks, albumid, "album", kodicursor) @CatchExceptions(warnuser=True) def add_updateSong(self, item, viewtag=None, viewid=None): # Process single song kodicursor = self.kodicursor plex_db = self.plex_db artwork = self.artwork API = PlexAPI.API(item) update_item = True itemid = API.getRatingKey() if not itemid: log.error('Error processing Song; skipping') return plex_dbitem = plex_db.getItem_byId(itemid) try: songid = plex_dbitem[0] pathid = plex_dbitem[2] albumid = plex_dbitem[3] except TypeError: # Songid not found update_item = False kodicursor.execute("select coalesce(max(idSong),0) from song") songid = kodicursor.fetchone()[0] + 1 # The song details ##### checksum = API.getChecksum() dateadded = API.getDateCreated() userdata = API.getUserData() playcount = userdata['PlayCount'] if playcount is None: # This is different to Video DB! playcount = 0 dateplayed = userdata['LastPlayedDate'] # item details title, sorttitle = API.getTitle() # musicBrainzId = API.getProvider('MusicBrainzTrackId') musicBrainzId = None genres = API.getGenres() genre = API.joinList(genres) artists = item.attrib.get('grandparentTitle') tracknumber = int(item.attrib.get('index', 0)) disc = int(item.attrib.get('parentIndex', 1)) if disc == 1: track = tracknumber else: track = disc*2**16 + tracknumber year = API.getYear() resume, duration = API.getRuntime() rating = userdata['UserRating'] hasEmbeddedCover = False comment = None # GET THE FILE AND PATH ##### doIndirect = not self.directpath if self.directpath: # Direct paths is set the Kodi way playurl = API.getFilePath(forceFirstMediaStream=True) if playurl is None: # Something went wrong, trying to use non-direct paths doIndirect = True else: playurl = API.validatePlayurl(playurl, API.getType()) if playurl is None: return False if "\\" in playurl: # Local path filename = playurl.rsplit("\\", 1)[1] else: # Network share filename = playurl.rsplit("/", 1)[1] path = playurl.replace(filename, "") if doIndirect: # Plex works a bit differently path = "%s%s" % (self.server, item[0][0].attrib.get('key')) path = API.addPlexCredentialsToUrl(path) filename = path.rsplit('/', 1)[1] path = path.replace(filename, '') # UPDATE THE SONG ##### if update_item: log.info("UPDATE song itemid: %s - Title: %s with path: %s" % (itemid, title, path)) # Update path # Use dummy strHash '123' for Kodi query = "UPDATE path SET strPath = ?, strHash = ? WHERE idPath = ?" kodicursor.execute(query, (path, '123', pathid)) # Update the song entry query = ' '.join(( "UPDATE song", "SET idAlbum = ?, strArtists = ?, strGenres = ?, strTitle = ?, iTrack = ?,", "iDuration = ?, iYear = ?, strFilename = ?, iTimesPlayed = ?, lastplayed = ?,", "rating = ?, comment = ?", "WHERE idSong = ?" )) kodicursor.execute(query, (albumid, artists, genre, title, track, duration, year, filename, playcount, dateplayed, rating, comment, songid)) # Update the checksum in plex table plex_db.updateReference(itemid, checksum) # OR ADD THE SONG ##### else: log.info("ADD song itemid: %s - Title: %s" % (itemid, title)) # Add path pathid = self.kodi_db.addPath(path, strHash="123") try: # Get the album plex_dbalbum = plex_db.getItem_byId( item.attrib.get('parentRatingKey')) albumid = plex_dbalbum[0] except KeyError: # Verify if there's an album associated. album_name = item.get('parentTitle') if album_name: log.info("Creating virtual music album for song: %s." % itemid) albumid = self.kodi_db.addAlbum(album_name, API.getProvider('MusicBrainzAlbum')) plex_db.addReference("%salbum%s" % (itemid, albumid), v.PLEX_TYPE_ALBUM, albumid, v.KODI_TYPE_ALBUM, view_id=viewid) else: # No album Id associated to the song. log.error("Song itemid: %s has no albumId associated." % itemid) return False except TypeError: # No album found. Let's create it log.info("Album database entry missing.") plex_albumId = item.attrib.get('parentRatingKey') album = GetPlexMetadata(plex_albumId) if album is None or album == 401: log.error('Could not download album, abort') return self.add_updateAlbum(album[0]) plex_dbalbum = plex_db.getItem_byId(plex_albumId) try: albumid = plex_dbalbum[0] log.debug("Found albumid: %s" % albumid) except TypeError: # No album found, create a single's album log.info("Failed to add album. Creating singles.") kodicursor.execute("select coalesce(max(idAlbum),0) from album") albumid = kodicursor.fetchone()[0] + 1 if v.KODIVERSION >= 16: # Kodi Jarvis query = ( ''' INSERT INTO album(idAlbum, strGenres, iYear, strReleaseType) VALUES (?, ?, ?, ?) ''' ) kodicursor.execute(query, (albumid, genre, year, "single")) elif v.KODIVERSION == 15: # Kodi Isengard query = ( ''' INSERT INTO album(idAlbum, strGenres, iYear, dateAdded, strReleaseType) VALUES (?, ?, ?, ?, ?) ''' ) kodicursor.execute(query, (albumid, genre, year, dateadded, "single")) else: # Kodi Helix query = ( ''' INSERT INTO album(idAlbum, strGenres, iYear, dateAdded) VALUES (?, ?, ?, ?) ''' ) kodicursor.execute(query, (albumid, genre, year, dateadded)) # Create the song entry query = ( ''' INSERT INTO song( idSong, idAlbum, idPath, strArtists, strGenres, strTitle, iTrack, iDuration, iYear, strFileName, strMusicBrainzTrackID, iTimesPlayed, lastplayed, rating, iStartOffset, iEndOffset) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''' ) kodicursor.execute( query, (songid, albumid, pathid, artists, genre, title, track, duration, year, filename, musicBrainzId, playcount, dateplayed, rating, 0, 0)) # Create the reference in plex table plex_db.addReference(itemid, v.PLEX_TYPE_SONG, songid, v.KODI_TYPE_SONG, kodi_pathid=pathid, parent_id=albumid, checksum=checksum, view_id=viewid) # Link song to album query = ( ''' INSERT OR REPLACE INTO albuminfosong( idAlbumInfoSong, idAlbumInfo, iTrack, strTitle, iDuration) VALUES (?, ?, ?, ?, ?) ''' ) kodicursor.execute(query, (songid, albumid, track, title, duration)) # Link song to artists artistLoop = [{ 'Name': item.attrib.get('grandparentTitle'), 'Id': item.attrib.get('grandparentRatingKey') }] # for index, artist in enumerate(item['ArtistItems']): for index, artist in enumerate(artistLoop): artist_name = artist['Name'] artist_eid = artist['Id'] artist_edb = plex_db.getItem_byId(artist_eid) try: artistid = artist_edb[0] except TypeError: # Artist is missing from plex database, add it. artistXml = GetPlexMetadata(artist_eid) if artistXml is None or artistXml == 401: log.error('Error getting artist, abort') return self.add_updateArtist(artistXml[0]) artist_edb = plex_db.getItem_byId(artist_eid) artistid = artist_edb[0] finally: if v.KODIVERSION >= 17: # Kodi Krypton query = ( ''' INSERT OR REPLACE INTO song_artist(idArtist, idSong, idRole, iOrder, strArtist) VALUES (?, ?, ?, ?, ?) ''' ) kodicursor.execute(query,(artistid, songid, 1, index, artist_name)) # May want to look into only doing this once? query = ( ''' INSERT OR REPLACE INTO role(idRole, strRole) VALUES (?, ?) ''' ) kodicursor.execute(query, (1, 'Composer')) else: query = ( ''' INSERT OR REPLACE INTO song_artist(idArtist, idSong, iOrder, strArtist) VALUES (?, ?, ?, ?) ''' ) kodicursor.execute(query, (artistid, songid, index, artist_name)) # Verify if album artist exists album_artists = [] # for artist in item['AlbumArtists']: if False: artist_name = artist['Name'] album_artists.append(artist_name) artist_eid = artist['Id'] artist_edb = plex_db.getItem_byId(artist_eid) try: artistid = artist_edb[0] except TypeError: # Artist is missing from plex database, add it. artistXml = GetPlexMetadata(artist_eid) if artistXml is None or artistXml == 401: log.error('Error getting artist, abort') return self.add_updateArtist(artistXml) artist_edb = plex_db.getItem_byId(artist_eid) artistid = artist_edb[0] finally: query = ( ''' INSERT OR REPLACE INTO album_artist(idArtist, idAlbum, strArtist) VALUES (?, ?, ?) ''' ) kodicursor.execute(query, (artistid, albumid, artist_name)) # Update discography if item.get('Album'): query = ( ''' INSERT OR REPLACE INTO discography(idArtist, strAlbum, strYear) VALUES (?, ?, ?) ''' ) kodicursor.execute(query, (artistid, item['Album'], 0)) # else: if False: album_artists = " / ".join(album_artists) query = ' '.join(( "SELECT strArtists", "FROM album", "WHERE idAlbum = ?" )) kodicursor.execute(query, (albumid,)) result = kodicursor.fetchone() if result and result[0] != album_artists: # Field is empty if v.KODIVERSION >= 16: # Kodi Jarvis, Krypton query = "UPDATE album SET strArtists = ? WHERE idAlbum = ?" kodicursor.execute(query, (album_artists, albumid)) elif v.KODIVERSION == 15: # Kodi Isengard query = "UPDATE album SET strArtists = ? WHERE idAlbum = ?" kodicursor.execute(query, (album_artists, albumid)) else: # Kodi Helix query = "UPDATE album SET strArtists = ? WHERE idAlbum = ?" kodicursor.execute(query, (album_artists, albumid)) # Add genres self.kodi_db.addMusicGenres(songid, genres, "song") # Update artwork allart = API.getAllArtwork(parentInfo=True) if hasEmbeddedCover: allart["Primary"] = "image://music@" + artwork.single_urlencode( playurl ) artwork.addArtwork(allart, songid, "song", kodicursor) # if item.get('AlbumId') is None: if item.get('parentKey') is None: # Update album artwork artwork.addArtwork(allart, albumid, "album", kodicursor) def remove(self, itemid): # Remove kodiid, fileid, pathid, plex reference plex_db = self.plex_db plex_dbitem = plex_db.getItem_byId(itemid) try: kodiid = plex_dbitem[0] mediatype = plex_dbitem[4] log.info("Removing %s kodiid: %s" % (mediatype, kodiid)) except TypeError: return ##### PROCESS ITEM ##### # Remove the plex reference plex_db.removeItem(itemid) ##### IF SONG ##### if mediatype == v.KODI_TYPE_SONG: # Delete song self.removeSong(kodiid) # This should only address single song scenario, where server doesn't actually # create an album for the song. plex_db.removeWildItem(itemid) for item in plex_db.getItem_byWildId(itemid): item_kid = item[0] item_mediatype = item[1] if item_mediatype == v.KODI_TYPE_ALBUM: childs = plex_db.getItem_byParentId(item_kid, v.KODI_TYPE_SONG) if not childs: # Delete album self.removeAlbum(item_kid) ##### IF ALBUM ##### elif mediatype == v.KODI_TYPE_ALBUM: # Delete songs, album album_songs = plex_db.getItem_byParentId(kodiid, v.KODI_TYPE_SONG) for song in album_songs: self.removeSong(song[1]) else: # Remove plex songs plex_db.removeItems_byParentId(kodiid, v.KODI_TYPE_SONG) # Remove the album self.removeAlbum(kodiid) ##### IF ARTIST ##### elif mediatype == v.KODI_TYPE_ARTIST: # Delete songs, album, artist albums = plex_db.getItem_byParentId(kodiid, v.KODI_TYPE_ALBUM) for album in albums: albumid = album[1] album_songs = plex_db.getItem_byParentId(albumid, v.KODI_TYPE_SONG) for song in album_songs: self.removeSong(song[1]) else: # Remove plex song plex_db.removeItems_byParentId(albumid, v.KODI_TYPE_SONG) # Remove plex artist plex_db.removeItems_byParentId(albumid, v.KODI_TYPE_ARTIST) # Remove kodi album self.removeAlbum(albumid) else: # Remove plex albums plex_db.removeItems_byParentId(kodiid, v.KODI_TYPE_ALBUM) # Remove artist self.removeArtist(kodiid) log.info("Deleted %s: %s from kodi database" % (mediatype, itemid)) def removeSong(self, kodiid): self.artwork.deleteArtwork(kodiid, v.KODI_TYPE_SONG, self.kodicursor) self.kodicursor.execute("DELETE FROM song WHERE idSong = ?", (kodiid,)) def removeAlbum(self, kodiid): self.artwork.deleteArtwork(kodiid, v.KODI_TYPE_ALBUM, self.kodicursor) self.kodicursor.execute("DELETE FROM album WHERE idAlbum = ?", (kodiid,)) def removeArtist(self, kodiid): self.artwork.deleteArtwork(kodiid, v.KODI_TYPE_ARTIST, self.kodicursor) self.kodicursor.execute("DELETE FROM artist WHERE idArtist = ?", (kodiid,))