Esempio n. 1
0
    def get_trakt_movies(self):
        pchtrakt.logger.info(' [YAMJ] Getting movies from trakt.tv')
    
        # Collection
        url = '/sync/collection/movies'
        #url = '/users/%s/collection/movies' % (TraktUsername)
        movies = trakt_apiv2(url)
        
        for movie in movies:
            trakt_movie = {
                'title': movie['movie']['title'],
                'year': movie['movie']['year'],
            }

            if 'imdb' in movie['movie']['ids']:
                trakt_movie['imdb'] = movie['movie']['ids']['imdb']
            if 'tmdb' in movie['movie']['ids']:
                trakt_movie['tmdb'] = movie['movie']['ids']['tmdb']
            #trakt_movie['id'] = ""
    
            self.trakt_movies.append(trakt_movie)
    
        movies = ''
    
        #Clean from collection, keep commented
        #url = '/movie/unlibrary/' + TraktAPI
        #params = {'movies': trakt_movies}
        #response = trakt_api('POST', url, params)
        #url = '/movie/unseen/' + TraktAPI
        #response = trakt_api('POST', url, params)
    
        # Seen
        url = '/sync/watched/movies'
        #url = '/users/%s/watched/movies' % (TraktUsername)
        seen_movies = trakt_apiv2(url)
        
        # Add playcounts to trakt collection
        for seen in seen_movies:
            if 'imdb' in seen['movie']['ids']:
                for movie in self.trakt_movies:
                    if 'imdb' in movie:
                        if seen['movie']['ids']['imdb'] == movie['imdb']:
                            movie['plays'] = seen['plays']
            elif 'tmdb' in seen['movie']['ids']:
                for movie in self.trakt_movies:
                    if 'tmdb' in movie:
                        if seen['movie']['ids']['tmdb'] == movie['tmdb']:
                            movie['plays'] = seen['plays']
    
            elif 'title' in seen:
                for movie in self.trakt_movies:
                    if 'title' in movie:
                        if seen['title'] == movie['title']:
                            movie['plays'] = seen['plays']
    
        for movie in self.trakt_movies:
            if not 'plays' in movie:
                movie['plays'] = 0
    
        seen_movies = ''
Esempio n. 2
0
 def YAMJ_movies_watched_to_trakt(self):
     pchtrakt.logger.info(' [YAMJ] Comparing YAMJ watched movies against trakt.tv')
     self.YAMJ_movies_to_trakt = []
 
     if self.trakt_movies and self.YAMJ_movies:
 
         for i in range(len(self.trakt_movies)):
             for movie in self.YAMJ_movies:
                 if movie['playcount'] != 0:
 
                     if 'imdb' in self.trakt_movies[i]:
                         if movie['imdbnumber'] == self.trakt_movies[i]['imdb']:
                             if self.trakt_movies[i]['plays'] < movie['playcount']:
                                 x_loop_must_break = False
                                 for x in self.YAMJ_movies_to_trakt:
                                     try:
                                         if movie['imdbnumber'] == x['movies'][0]['ids']['imdb']:
                                             x_loop_must_break = True
                                             break
                                     except:
                                         if movie['imdbnumber'] == x['ids']['imdb']:
                                             x_loop_must_break = True
                                             break
 
                                 if x_loop_must_break: break
                                 self.YAMJ_movies_to_trakt.append(self.convert_YAMJ_movie_to_trakt(movie, watched_at = True))
 
                     elif 'tmdb' in self.trakt_movies[i]:
                         if movie['imdbnumber'] == self.trakt_movies[i]['tmdb']:
                             if self.trakt_movies[i]['plays'] < movie['playcount']:
                                 self.YAMJ_movies_to_trakt.append(self.convert_YAMJ_movie_to_trakt(movie, watched_at = True))
 
                     elif movie['title'] == self.trakt_movies[i]['movies'][0]['title']:
                         if self.trakt_movies[i]['plays'] < movie['playcount']:
                             self.YAMJ_movies_to_trakt.append(self.convert_YAMJ_movie_to_trakt(movie, watched_at = True))
 
     if self.YAMJ_movies_to_trakt:
         pchtrakt.logger.info(' [YAMJ] %s movies playcount will be updated on trakt.tv' % len(self.YAMJ_movies_to_trakt))
         # Send request to update playcounts on trakt.tv
         url = '/sync/history'
         params = {'movies': self.YAMJ_movies_to_trakt}
         try:
             pchtrakt.logger.info(' [YAMJ] Updating watched status for movies on trakt.tv...')
             response = trakt_apiv2(url, params, sync=True)
             pchtrakt.logger.info(' [YAMJ]     Marked %s as watched out of %s movies' % (response['added']['movies'], len(self.YAMJ_movies_to_trakt)))
             if len(response['not_found']['movies']) != 0:
                 for skip in response['not_found']['movies']:
                     pchtrakt.logger.info(' [YAMJ]    could not add     -->%s' % skip['not_found']['movies'][0]['title'].encode('utf-8'))
         except Exception, e:
             pchtrakt.logger.info(' [YAMJ] Failed to update playcount for movies on trakt.tv')
             pchtrakt.logger.info(e)
Esempio n. 3
0
 def YAMJ_shows_watched_to_trakt(self):
     pchtrakt.logger.info(' [YAMJ] Comparing YAMJ watched TV shows against trakt.tv')
     self.YAMJ_shows_to_trakt = []
 
     if self.YAMJ_shows and self.trakt_shows:
 
         tvdb_ids = {}
         imdb_ids = {}
 
         for i in range(len(self.trakt_shows)):
             if 'tvdb' in self.trakt_shows[i]:
                 tvdb_ids[self.trakt_shows[i]['tvdb']] = i
 
             if 'imdb' in self.trakt_shows[i]:
                 imdb_ids[self.trakt_shows[i]['imdb']] = i
 
         for show in self.YAMJ_shows.values():
             if 'imdbnumber' in show:
                 if show['imdbnumber'].startswith('tt'):
                     if self.search(imdb_ids, show['imdbnumber']):#if show['imdbnumber'] in imdb_ids.keys():
                         trakt_show = self.trakt_shows[imdb_ids.get(int(show['imdbnumber']))]
 
                         trakt_show_watched = {
                             'title': show['title'],
                             'imdb': show['imdbnumber'],
                             'episodes': []
                         }
 
                         for YAMJ_ep in show['episodes']:
                             for trakt_ep in trakt_show['episodes']:
                                 if trakt_ep['season'] == YAMJ_ep['season']:
                                     if trakt_ep['episode'] == YAMJ_ep['episode']:
                                         if trakt_ep['plays'] == 0 and YAMJ_ep['playcount'] >= 1:
 
                                             trakt_show_watched['episodes'].append(
                                                 {
                                                     'season': YAMJ_ep['season'],
                                                     'episode': YAMJ_ep['episode'],
                                                     'date': YAMJ_ep['date']
                                                 }
                                             )
 
                         if trakt_show_watched['episodes']:
                             self.YAMJ_shows_to_trakt.append(trakt_show_watched)
 
                 else:
                     if self.searchtv(tvdb_ids, show['imdbnumber']):#if show['imdbnumber'] in tvdb_ids.keys():
                         trakt_show = self.trakt_shows[tvdb_ids.get(int(show['imdbnumber']))]
 
                         trakt_show_watched = {
                             'title': show['title'],
                             'tvdb': show['imdbnumber'],
                             'episodes': []
                         }
 
                         for YAMJ_ep in show['episodes']:
                             for trakt_ep in trakt_show['episodes']:
                                 if trakt_ep['season'] == YAMJ_ep['season']:
                                     if trakt_ep['episode'] == YAMJ_ep['episode']:
                                         if trakt_ep['plays'] == 0 and YAMJ_ep['playcount'] >= 1:
 
                                             trakt_show_watched['episodes'].append(
                                                 {
                                                     'season': YAMJ_ep['season'],
                                                     'episode': YAMJ_ep['episode'],
                                                     'watched_at': YAMJ_ep['date']
                                                 }
                                             )
 
                         if trakt_show_watched['episodes']:
                             self.YAMJ_shows_to_trakt.append(trakt_show_watched)
 
         if self.YAMJ_shows_to_trakt:
             pchtrakt.logger.info(' [YAMJ] %s TV shows have episodes that will be marked as watched in trakt.tv collection' % len(self.YAMJ_shows_to_trakt))
 
             for i in range(len(self.YAMJ_shows_to_trakt)):
                 #convert YAMJ show into something trakt will understand
                 self.YAMJ_shows_to_trakt[i] = self.convert_YAMJ_show_to_trakt(self.YAMJ_shows_to_trakt[i])
 
             # Send request to add TV shows to trakt.tv
             url = '/sync/history'
 
             for show in self.YAMJ_shows_to_trakt:
                 try:
                     pchtrakt.logger.info(' [YAMJ]     -->%s' % show['shows'][0]['title'])
                     trakt = trakt_apiv2(url, show, sync=True)
                     if trakt['added']['episodes'] != 0 and len(trakt['not_found']['episodes']) != 0:
                         pchtrakt.logger.info(' [YAMJ] Successfully marked  %s episodes watched out of %s' % (trakt['added']['episodes'], trakt['added']['episodes'] + trakt['not_found']['episodes']))
                     else:
                         pchtrakt.logger.info(' [YAMJ] Successfully marked  %s episodes watched out of %s' % (trakt['added']['episodes'], trakt['added']['episodes']))
                 except Exception, e:
                     pchtrakt.logger.info(' [YAMJ] Failed to mark %s\'s episodes as watched in trakt.tv collection' % show['shows'][0]['title'])
                     pchtrakt.logger.info(e)
 
         else:
             pchtrakt.logger.info(' [YAMJ] trakt.tv TV show watched status is up to date')
Esempio n. 4
0
    def YAMJ_shows_to_trakt(self):
        pchtrakt.logger.info(' [YAMJ] Checking for YAMJ episodes that are not in trakt.tv collection')
        self.YAMJ_shows_to_trakt = []
    
        def clean_episodes(shows):
            if shows:
                for show in shows:
                    episodes = []
                    for episode in show['episodes']:
                        episodes.append({'season': episode['season'], 'episode': episode['episode']})
                    show['episodes'] = episodes
    
            return shows

        if self.YAMJ_shows and self.trakt_shows:
    
            t_shows = copy.deepcopy(self.trakt_shows)
            t_shows = clean_episodes(t_shows)
            x_shows = copy.deepcopy(self.YAMJ_shows.values())
            x_shows = clean_episodes(x_shows)

            tvdb_ids = {}
            imdb_ids = {}
    
            for i in range(len(t_shows)):
                if 'tvdb' in t_shows[i]:
                    tvdb_ids[t_shows[i]['tvdb']] = i
                if 'imdb' in t_shows[i]:
                    imdb_ids[t_shows[i]['imdb']] = i
    
            for show in x_shows:
                if 'imdbnumber' in show:
                    if show['imdbnumber'].startswith('tt'):
                        if self.search(imdb_ids, show['imdbnumber']) == False:#if not show['imdbnumber'] in imdb_ids:
                            self.YAMJ_shows_to_trakt.append(show)
    
                            #trakt_show = convert_YAMJ_show_to_trakt(show)
                            for episode in trakt_show['episodes']:
                                episode['plays'] = 0
    
                            #trakt_shows.append(trakt_show)
    
                        else:
                            t_index = imdb_ids.get(int(show['imdbnumber']))
    
                            YAMJ_show = {
                                'title': show['title'],
                                'imdbnumber': show['imdbnumber'],
                                'episodes': []
                            }
    
                            for episode in show['episodes']:
                                if episode['episode'] == 0:
                                    continue
                                if episode not in t_shows[t_index]['episodes']:
                                    YAMJ_show['episodes'].append(episode)
    
                                    self.trakt_shows[t_index]['episodes'].append(episode)
                                    self.trakt_shows[t_index]['episodes'][-1]['plays'] = 0
    
                            if YAMJ_show['episodes']:
                                self.YAMJ_shows_to_trakt.append(YAMJ_show)
    
                    else:
                        if self.searchtv(tvdb_ids, show['imdbnumber']) == False:# if not show['imdbnumber'] in tvdb_ids:
                            self.YAMJ_shows_to_trakt.append(show)
    
                            #trakt_show = convert_YAMJ_show_to_trakt(show)
                            #for season in trakt_show['shows'][0]['seasons']:
                            #    for episode in season['episodes']:
                            #        episode['plays'] = 0
    
                            #trakt_shows.append(trakt_show)
    
                        else:
                            t_index = tvdb_ids.get(int(show['imdbnumber']))
                            #tvdb_ids.get(int(show['imdbnumber']))
    
                            YAMJ_show = {
                                'title': show['title'],
                                'imdbnumber': show['imdbnumber'],
                                'episodes': []
                            }
    
                            for episode in show['episodes']:
                                if episode['episode'] == 0:
                                    continue
                                if episode not in t_shows[t_index]['episodes']:
                                    YAMJ_show['episodes'].append(episode)
    
                                    self.trakt_shows[t_index]['episodes'].append(episode)
                                    self.trakt_shows[t_index]['episodes'][-1]['plays'] = 0
    
                            if YAMJ_show['episodes']:
                                self.YAMJ_shows_to_trakt.append(YAMJ_show)
    
            if self.YAMJ_shows_to_trakt:
                pchtrakt.logger.info(' [YAMJ] %s TV shows have episodes missing from trakt.tv collection' % len(self.YAMJ_shows_to_trakt))
    
                for i in range(len(self.YAMJ_shows_to_trakt)):
                    #convert YAMJ show into something trakt will understand
                    self.YAMJ_shows_to_trakt[i] = self.convert_YAMJ_show_to_trakt(self.YAMJ_shows_to_trakt[i])
    
                # Send request to add TV shows to trakt.tv
                url = '/sync/collection'
                #data = {'shows': [{'title': 'Mad Men', 'year': 2007, 'ids': {'trakt': 4, 'slug': 'mad-men', 'tvdb': 80337, 'imdb': 'tt0804503', 'tmdb': 1104, 'tvrage': 16356}, 'seasons': [{'number': 1, 'episodes': [{'number': 1},{'number': 2}]}]}]}
    
                for show in self.YAMJ_shows_to_trakt:
                    try:
                        #params = {'shows': [show]}
                        pchtrakt.logger.info(' [YAMJ]     -->%s' % show['shows'][0]['title'])
                        trakt = trakt_apiv2(url, show, sync=True)
                        if trakt['added']['episodes'] > 0:
                            pchtrakt.logger.info(' [YAMJ]       Added %s' % trakt['added']['episodes'])
                        if trakt['updated']['episodes'] > 0:
                            pchtrakt.logger.info(' [YAMJ]       Updated %s' % trakt['updated']['episodes'])
                        if trakt['existing']['episodes'] > 0:
                            pchtrakt.logger.info(' [YAMJ]       Modified %s' % trakt['existing']['episodes'])
                        #if trakt['not_found']['episodes']:
                    except Exception, e:
                        pchtrakt.logger.info(' [YAMJ] Failed to add %s\'s new episodes to trakt.tv collection' % show['title'])
                        pchtrakt.logger.info(e)
    
            else:
                pchtrakt.logger.info(' [YAMJ] trakt.tv TV show collection is up to date')
Esempio n. 5
0
    def get_trakt_shows(self):
        pchtrakt.logger.info(' [YAMJ] Getting TV shows from trakt')

        # Collection
        #url = '/users/%s/collection/shows' % (TraktUsername)
        url = '/sync/collection/shows'
    
        collection_shows = trakt_apiv2(url)
        
        for show in collection_shows:
            trakt_show = {
                'title': show['show']['title'],
                'episodes': []
            }
    
            if 'imdb' in show['show']['ids']:
                trakt_show['imdb'] = show['show']['ids']['imdb']
            if 'tvdb' in show['show']['ids']:
                trakt_show['tvdb'] = show['show']['ids']['tvdb']
    
            for season in show['seasons']:
                for episode in season['episodes']:
                    ep = {'season': season['number'], 'episode': episode['number'], 'plays': 0}
                    trakt_show['episodes'].append(ep)
    
            #Clean from collection, keep commented
            #url = '/show/episode/unseen/' + TraktAPI
            #response = trakt_api('post', url, trakt_show)
            #url = '/show/episode/unlibrary/' + TraktAPI
            #response = trakt_api('post', url, trakt_show)
            self.trakt_shows.append(trakt_show)
    
        collection_shows = ''
    
        # Seen
        url = '/sync/watched/shows'
        #url = '/users/%s/watched/shows' % (TraktUsername)
        seen_shows = trakt_apiv2(url)
        show = ''
        for show in seen_shows:
            for season in show['seasons']:
                for episode in season['episodes']:
                    for trakt_show in self.trakt_shows:
                        if ('imdb' in show['show']['ids'] and 'imdb' in trakt_show) and show['show']['ids']['imdb'] != None and show['show']['ids']['imdb'] == trakt_show['imdb']:
                            try:
                                #if len(show['show']['ids']['imdb']) > 0:
                                    for trakt_episode in trakt_show['episodes']:
                                        if trakt_episode['season'] == season['number'] and trakt_episode['episode'] == episode['number']:
                                            trakt_episode['plays'] = 1
                            except Exception as e:
                                pass
    
                        elif ('tvdb' in show['show']['ids'] and 'tvdb' in trakt_show) and show['show']['ids']['tvdb'] != None and show['show']['ids']['tvdb'] == trakt_show['tvdb']:
                            try:
                                #if len(show['show']['ids']['tvdb']) > 0:
                                    for trakt_episode in trakt_show['episodes']:
                                        if trakt_episode['season'] == season['number'] and trakt_episode['episode'] == episode['number']:
                                            trakt_episode['plays'] = 1
                            except Exception as e:
                                pass
    
                        else:
                            try:
                                if show['show']['title'] == trakt_show['title']:
                                    for trakt_episode in trakt_show['episodes']:
                                        if trakt_episode['season'] == season['number'] and trakt_episode['episode'] == episode['number']:
                                            trakt_episode['plays'] = 1
                            except:
                                if show['show']['title'] == trakt_show['shows'][0]['title']:
                                    for trakt_episode in trakt_show['episodes']:
                                        if trakt_episode['season'] == season['number'] and trakt_episode['episode'] == episode['number']:
                                            trakt_episode['plays'] = 1
    
        seen_shows = ''
        show = ''
Esempio n. 6
0
    def YAMJ_movies_to_trakt(self):
        pchtrakt.logger.info(' [YAMJ] Checking for YAMJ movies that are not in trakt.tv collection')
        self.YAMJ_movies_to_trakt = []
    
        if self.trakt_movies and self.YAMJ_movies:
            imdb_ids = [x['imdb'] for x in self.trakt_movies if 'imdb' in x]
            tmdb_ids = [x['tmdb'] for x in self.trakt_movies if 'tmdb' in x]
            titles = [x['title'] for x in self.trakt_movies if 'title' in x]
    
        if self.YAMJ_movies:
            for movie in self.YAMJ_movies:
                if 'imdbnumber' in movie:
                    if movie['imdbnumber'].startswith('tt'):
                        if self.trakt_movies:
                            if self.search(imdb_ids, movie['imdbnumber']) == False:
                                self.YAMJ_movies_to_trakt.append(movie)
                                #trakt_movie = convert_YAMJ_movie_to_trakt(movie)# do we need these below?
                                #if not 'plays' in trakt_movie[0]:trakt_movie['movies']
                                #    trakt_movie[0]['plays'] = 0
                                #trakt_movies.append(trakt_movie)
                        else:
                            self.YAMJ_movies_to_trakt.append(movie)
                            #trakt_movie = convert_YAMJ_movie_to_trakt(movie)
                            #if not 'plays' in trakt_movie[0]:
                            #    trakt_movie[0]['plays'] = 0
                            ##trakt_movies.append(trakt_movie)
                    else:
                        if self.trakt_movies:
                            if self.searchtv(tmdb_ids, movie['imdbnumber']) == False:#if not movie['imdbnumber'] in tmdb_ids:
                                self.YAMJ_movies_to_trakt.append(movie)
                                #trakt_movie = convert_YAMJ_movie_to_trakt(movie)
                                #if not 'plays' in trakt_movie[0]:
                                #    trakt_movie[0]['plays'] = 0
                                #trakt_movies.append(trakt_movie)
                        else:
                            self.YAMJ_movies_to_trakt.append(movie)
                            #trakt_movie = convert_YAMJ_movie_to_trakt(movie)
                            #if not 'plays' in trakt_movie[0]:
                            #    trakt_movie[0]['plays'] = 0
                            ##trakt_movies.append(trakt_movie)
                elif not movie['title'] in titles and not movie in self.YAMJ_movies_to_trakt:
                    self.YAMJ_movies_to_trakt.append(movie)
                    #trakt_movie = convert_YAMJ_movie_to_trakt(movie)
                    #if not 'plays' in trakt_movie[0]:
                    #    trakt_movie[0]['plays'] = 0
                    #trakt_movies.append(trakt_movie)

        if self.YAMJ_movies_to_trakt:
            pchtrakt.logger.info(' [YAMJ] Checking for %s movies will be added to trakt.tv collection' % len(self.YAMJ_movies_to_trakt))
            for i in range(len(self.YAMJ_movies_to_trakt)):
                #convert YAMJ movie into something trakt will understand
                self.YAMJ_movies_to_trakt[i] = self.convert_YAMJ_movie_to_trakt(self.YAMJ_movies_to_trakt[i])
    
            # Send request to add movies to trakt.tv
            url = '/sync/collection'
            params = {'movies': self.YAMJ_movies_to_trakt}

            try:
                pchtrakt.logger.info(' [YAMJ] Adding movies to trakt.tv collection...')
                response = trakt_apiv2(url, params, sync=True)
                if response['added']['movies'] != 0:
                    if len(response['not_found']['movies']) !=0:
                        pchtrakt.logger.info(' [YAMJ] Successfully added %s out of %s to your collection' % (response['added']['movies'], response['added']['movies'] + response['existing']['movies'] + len(response['not_found']['movies'])))
                        pchtrakt.logger.info(' [YAMJ] Failed to add the following %s titles to your collection' % len(response['not_found']['movies']))
                        for failed in response['not_found']['movies']:
                            pchtrakt.logger.info(' [YAMJ] Failed to add %s' % failed['movies'][0]['title'].encode('utf-8', 'replace'))
                    else:
                        pchtrakt.logger.info(' [YAMJ] Successfully added %s out of %s to your collection' % (response['added']['movies'], response['added']['movies'] + response['existing']['movies']))
                if response['existing']['movies'] != 0:
                    pchtrakt.logger.info(' [YAMJ] %s titles were found in your collection already' % response['existing']['movies'])
            except Exception, e:
                pchtrakt.logger.info(' [YAMJ] Failed to add movies to trakt.tv collection')
                pchtrakt.logger.info(e)