public void ImportRatings()
{
mImportCancelled = false;
var lRateItems = new List <Dictionary <string, string> >();
var lWatchlistItems = new List <Dictionary <string, string> >();
var lCustomLists = new Dictionary <string, List <Dictionary <string, string> > >();
#region Parse Ratings CSV
UIUtils.UpdateStatus("Reading IMDb ratings export...");
if (mImportCsvRatings && !ParseCSVFile(mRatingsFileCsv, out lRateItems))
{
UIUtils.UpdateStatus("Failed to parse IMDb ratings file!", true);
Thread.Sleep(2000);
return;
}
if (mImportCancelled)
{
return;
}
#endregion
#region Parse Watchlist CSV
UIUtils.UpdateStatus("Reading IMDb watchlist export...");
if (mImportCsvWatchlist && !ParseCSVFile(mWatchlistFileCsv, out lWatchlistItems))
{
UIUtils.UpdateStatus("Failed to parse IMDb watchlist file!", true);
Thread.Sleep(2000);
return;
}
if (mImportCancelled)
{
return;
}
#endregion
#region Parse Custom List CSVs
UIUtils.UpdateStatus("Reading IMDb custom lists export...");
if (mImportCsvCustomLists)
{
foreach (var list in mCustomListsCsvs)
{
UIUtils.UpdateStatus("Reading IMDb custom list '{0}' export...", list);
var lCustomList = new List <Dictionary <string, string> >();
if (!ParseCSVFile(list, out lCustomList))
{
UIUtils.UpdateStatus("Failed to parse IMDb custom list file!", true);
Thread.Sleep(2000);
return;
}
lCustomLists.Add(list, lCustomList);
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Rated Movies
var lMovies = lRateItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
FileLog.Info("Found {0} movie ratings in CSV file", lMovies.Count);
if (lMovies.Any())
{
UIUtils.UpdateStatus("Retrieving existing movie ratings from trakt.tv");
var lCurrentUserMovieRatings = TraktAPI.GetRatedMovies();
if (lCurrentUserMovieRatings != null)
{
UIUtils.UpdateStatus("Found {0} user movie ratings on trakt.tv", lCurrentUserMovieRatings.Count());
// Filter out movies to rate from existing ratings online
lMovies.RemoveAll(m => lCurrentUserMovieRatings.Any(c => c.Movie.Ids.ImdbId == m[IMDbFieldMapping.cIMDbID]));
}
UIUtils.UpdateStatus("Importing {0} new movie ratings to trakt.tv", lMovies.Count());
if (lMovies.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lMovies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated movies...", i + 1, pages);
TraktSyncResponse response = TraktAPI.AddMoviesToRatings(Helper.GetRateMoviesData(lMovies.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb movie ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Rated TV Shows
var lShows = lRateItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Show && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
FileLog.Info("Found {0} tv show ratings in CSV file", lShows.Count);
if (lShows.Any())
{
UIUtils.UpdateStatus("Retrieving existing tv show ratings from trakt.tv");
var currentUserShowRatings = TraktAPI.GetRatedShows();
if (currentUserShowRatings != null)
{
UIUtils.UpdateStatus("Found {0} user tv show ratings on trakt.tv", currentUserShowRatings.Count());
// Filter out shows to rate from existing ratings online
lShows.RemoveAll(s => currentUserShowRatings.Any(c => (c.Show.Ids.ImdbId == s[IMDbFieldMapping.cIMDbID]) || (c.Show.Title == s[IMDbFieldMapping.cTitle] && c.Show.Year.ToString() == s[IMDbFieldMapping.cYear])));
}
UIUtils.UpdateStatus("Importing {0} tv show ratings to trakt.tv", lShows.Count());
if (lShows.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lShows.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated shows...", i + 1, pages);
TraktSyncResponse response = TraktAPI.AddShowsToRatings(Helper.GetRateShowsData(lShows.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb tv show ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} shows as they're not found on trakt.tv!", response.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Rated Episodes
var lImdbEpisodes = new List <IMDbEpisode>();
var lImdbCsvEpisodes = lRateItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Episode).ToList();
FileLog.Info("Found {0} tv episode ratings in CSV file", lImdbCsvEpisodes.Count);
if (lImdbCsvEpisodes.Any())
{
// we can't rely on the imdb id as trakt most likely wont have the info for episodes
// search and cache all series info needed for syncing
// use the tvdb API to first search for each unique series name
// then GetSeries by TVDb ID to get a list of all episodes
// each episode will have TVDb ID which we can use for syncing.
lImdbEpisodes.AddRange(lImdbCsvEpisodes.Select(Helper.GetIMDbEpisodeFromTVDb).Where(imdbEpisode => imdbEpisode != null));
UIUtils.UpdateStatus("Retrieving existing tv episode ratings from trakt.tv");
var currentUserEpisodeRatings = TraktAPI.GetRatedEpisodes();
if (currentUserEpisodeRatings != null)
{
UIUtils.UpdateStatus("Found {0} user tv episode ratings on trakt.tv", currentUserEpisodeRatings.Count());
// Filter out episodes to rate from existing ratings online
lImdbEpisodes.RemoveAll(e => currentUserEpisodeRatings.Any(c => c.Episode.Ids.TvdbId == e.TvdbId || c.Episode.Ids.ImdbId == e.ImdbId));
}
UIUtils.UpdateStatus("Importing {0} episode ratings to trakt.tv", lImdbEpisodes.Count());
if (lImdbEpisodes.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lImdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
var episodesRated = Helper.GetTraktEpisodeRateData(lImdbEpisodes.Skip(i * pageSize).Take(pageSize));
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated episodes...", i + 1, pages);
var response = TraktAPI.AddsEpisodesToRatings(episodesRated);
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb episodes as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Mark Rated Items as Watched
IEnumerable <TraktMoviePlays> lWatchedTraktMovies = null;
if (AppSettings.MarkAsWatched)
{
#region Movies
// compare all movies rated against what's not watched on trakt
lMovies = lRateItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie).ToList();
FileLog.Info("Found {0} movies in CSV file", lMovies.Count);
if (lMovies.Count > 0)
{
// get watched movies from trakt.tv
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
lWatchedTraktMovies = TraktAPI.GetWatchedMovies();
if (lWatchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv, skipping watched movie import", true);
Thread.Sleep(2000);
}
else
{
if (mImportCancelled)
{
return;
}
UIUtils.UpdateStatus("Found {0} watched movies on trakt", lWatchedTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watched movies that are already watched on trakt.tv");
lMovies.RemoveAll(w => lWatchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
// mark all rated movies as watched
UIUtils.UpdateStatus("Importing {0} IMDb movies as watched...", lMovies.Count);
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lMovies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb movies as watched...", i + 1, pages);
var response = TraktAPI.AddMoviesToWatchedHistory(Helper.GetSyncWatchedMoviesData(lMovies.Skip(i * pageSize).Take(pageSize).ToList()));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watched status for IMDb movies to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watched states for {0} movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
#endregion
#region Episodes
if (lImdbEpisodes != null && lImdbEpisodes.Any())
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lImdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
var episodesWatched = Helper.GetTraktEpisodeWatchedData(lImdbEpisodes.Skip(i * pageSize).Take(pageSize));
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb watched episodes...", i + 1, pages);
var response = TraktAPI.AddEpisodesToWatchedHistory(episodesWatched);
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes as watched to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync {0} IMDb episodes as watched, as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
#endregion
}
#endregion
#region Import Watchlist Movies
lMovies = lWatchlistItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie).ToList();
FileLog.Info("Found {0} movies watchlisted in CSV file", lMovies.Count);
if (lMovies.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlist movies from trakt...");
var watchlistTraktMovies = TraktAPI.GetWatchlistMovies();
if (watchlistTraktMovies != null)
{
UIUtils.UpdateStatus("Found {0} watchlist movies on trakt", watchlistTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watchlist movies that are already in watchlist on trakt.tv");
lMovies.RemoveAll(w => watchlistTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lMovies.Count > 0)
{
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
// get watched movies from trakt so we don't import movies into watchlist that are already watched
if (lWatchedTraktMovies != null)
{
lWatchedTraktMovies = TraktAPI.GetWatchedMovies();
if (lWatchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv", true);
Thread.Sleep(2000);
}
}
if (lWatchedTraktMovies != null)
{
UIUtils.UpdateStatus("Found {0} watched movies on trakt", lWatchedTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watchlist movies that are watched on trakt.tv");
// remove movies from sync list which are watched already
lMovies.RemoveAll(w => lWatchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
}
// add movies to watchlist
UIUtils.UpdateStatus("Importing {0} IMDb watchlist movies to trakt.tv...", lMovies.Count());
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lMovies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb watchlist movies...", i + 1, pages);
var response = TraktAPI.AddMoviesToWatchlist(Helper.GetSyncMoviesData(lMovies.Skip(i * pageSize).Take(pageSize).ToList()));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb movies", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Watchlist TV Shows
IEnumerable <TraktShowPlays> watchedTraktShows = null;
lShows = lWatchlistItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Show).ToList();
FileLog.Info("Found {0} tv shows watchlisted in CSV file", lShows.Count);
if (lShows.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlist shows from trakt...");
var watchlistTraktShows = TraktAPI.GetWatchlistShows();
if (watchlistTraktShows != null)
{
UIUtils.UpdateStatus("Found {0} watchlist shows on trakt", watchlistTraktShows.Count());
UIUtils.UpdateStatus("Filtering out watchlist shows that are already in watchlist on trakt.tv");
lShows.RemoveAll(w => watchlistTraktShows.FirstOrDefault(t => t.Show.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Show.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle].ToLowerInvariant() && t.Show.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lShows.Count > 0)
{
UIUtils.UpdateStatus("Requesting watched shows from trakt...");
// get watched movies from trakt so we don't import shows into watchlist that are already watched
watchedTraktShows = TraktAPI.GetWatchedShows();
if (watchedTraktShows != null)
{
UIUtils.UpdateStatus("Found {0} watched shows on trakt", watchedTraktShows.Count());
UIUtils.UpdateStatus("Filtering out watchlist shows containing watched episodes on trakt.tv.");
// remove shows from sync list which are watched already
lShows.RemoveAll(w => watchedTraktShows.FirstOrDefault(t => (t.Show.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID]) || (t.Show.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle].ToLowerInvariant() && t.Show.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
}
// add shows to watchlist
UIUtils.UpdateStatus("Importing {0} IMDb watchlist shows to trakt.tv...", lShows.Count());
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lShows.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb watchlist shows...", i + 1, pages);
var response = TraktAPI.AddShowsToWatchlist(Helper.GetSyncShowsData(lShows.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb tv shows", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} shows as they're not found on trakt.tv!", response.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Watchlist Episodes
lImdbEpisodes.Clear();
lImdbCsvEpisodes = lWatchlistItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Episode).ToList();
FileLog.Info("Found {0} tv episodes watchlisted in CSV file", lImdbCsvEpisodes.Count);
if (lImdbCsvEpisodes.Any())
{
UIUtils.UpdateStatus("Found {0} IMDb watchlist episodes", lImdbCsvEpisodes.Count());
lImdbEpisodes.AddRange(lImdbCsvEpisodes.Select(Helper.GetIMDbEpisodeFromTVDb).Where(imdbEpisode => imdbEpisode != null));
// filter out existing watchlist episodes
UIUtils.UpdateStatus("Requesting existing watchlist episodes from trakt...");
var watchlistTraktEpisodes = TraktAPI.GetWatchlistEpisodes();
if (watchlistTraktEpisodes != null)
{
UIUtils.UpdateStatus("Found {0} watchlist episodes on trakt", watchlistTraktEpisodes.Count());
UIUtils.UpdateStatus("Filtering out watchlist episodes that are already in watchlist on trakt.tv");
lImdbEpisodes.RemoveAll(e => watchlistTraktEpisodes.FirstOrDefault(w => w.Episode.Ids.ImdbId == e.ImdbId || w.Episode.Ids.TvdbId == e.TvdbId) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lImdbEpisodes.Count > 0)
{
// we already might have it from the shows sync
if (watchedTraktShows == null)
{
UIUtils.UpdateStatus("Requesting watched episodes from trakt...");
// get watched episodes from trakt so we don't import episodes into watchlist that are already watched
watchedTraktShows = TraktAPI.GetWatchedShows();
}
if (watchedTraktShows != null)
{
UIUtils.UpdateStatus("Filtering out watchlist episodes containing watched episodes on trakt.tv");
lImdbEpisodes.RemoveAll(e => watchedTraktShows.Where(s => s.Show.Ids.ImdbId == e.ImdbId)
.Any(s => s.Seasons.Exists(se => se.Number == e.SeasonNumber && se.Episodes.Exists(ep => ep.Number == e.EpisodeNumber))));
}
}
UIUtils.UpdateStatus("Importing {0} episodes in watchlist to trakt.tv", lImdbEpisodes.Count());
if (lImdbEpisodes.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lImdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb watchlist episodes...", i + 1, pages);
var response = TraktAPI.AddEpisodesToWatchlist(Helper.GetTraktEpisodeData(lImdbEpisodes.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episode watchlist to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} IMDb episodes as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
#endregion
#region Import Custom Lists
if (lCustomLists.Count > 0)
{
UIUtils.UpdateStatus("Requesting custom lists from trakt...");
var lTraktCustomLists = TraktAPI.GetCustomLists();
if (lTraktCustomLists == null)
{
UIUtils.UpdateStatus("Error requesting custom lists from trakt.tv", true);
Thread.Sleep(2000);
return;
}
UIUtils.UpdateStatus("Found {0} custom lists on trakt.tv", lTraktCustomLists.Count());
foreach (var list in lCustomLists)
{
bool lListCreated = false;
string lListName = Path.GetFileNameWithoutExtension(list.Key);
// create the list if we don't have it
TraktListDetail lTraktCustomList = lTraktCustomLists.FirstOrDefault(l => l.Name == lListName);
if (lTraktCustomList == null)
{
UIUtils.UpdateStatus("Creating new custom list '{0}' on trakt.tv", lListName);
var lTraktList = new TraktList
{
Name = lListName,
DisplayNumbers = true,
};
lTraktCustomList = TraktAPI.CreateCustomList(lTraktList);
if (lTraktCustomList == null)
{
UIUtils.UpdateStatus("Error creating custom list on trakt.tv, skipping list creation", true);
Thread.Sleep(2000);
continue;
}
lListCreated = true;
}
// get the CSV list items parsed
var lIMDbListItems = list.Value;
var lImdbListMovies = lIMDbListItems.Where(l => l.ItemType() == IMDbType.Movie).ToList();
var lImdbListShows = lIMDbListItems.Where(l => l.ItemType() == IMDbType.Show).ToList();
// if the list already exists, get current items for list
if (!lListCreated)
{
lTraktCustomList = lTraktCustomLists.FirstOrDefault(l => l.Name == lListName);
UIUtils.UpdateStatus("Requesting existing custom list '{0}' items from trakt...", lListName);
var lTraktListItems = TraktAPI.GetCustomListItems(lTraktCustomList.Ids.Trakt.ToString());
if (lTraktListItems == null)
{
UIUtils.UpdateStatus("Error requesting custom list items on trakt.tv, skipping list creation", true);
Thread.Sleep(2000);
continue;
}
// filter out existing items from CSV so we don't send again
FileLog.Info("Filtering out existing items from IMDb list '{0}' so we don't send again to trakt.tv", lListName);
lImdbListMovies.RemoveAll(d => d.ItemType() == IMDbType.Movie && lTraktListItems.FirstOrDefault(l => l.Movie.Ids.ImdbId == d[IMDbFieldMapping.cIMDbID]) != null);
lImdbListShows.RemoveAll(d => d.ItemType() == IMDbType.Show && lTraktListItems.FirstOrDefault(l => l.Show.Ids.ImdbId == d[IMDbFieldMapping.cIMDbID]) != null);
}
#region Movies
UIUtils.UpdateStatus("Importing {0} movies into {1} custom list...", lImdbListMovies.Count(), lListName);
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)lImdbListMovies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb custom list movies...", i + 1, pages);
// create list sync object to hold list items
var lTraktMovieSync = new TraktSyncAll
{
Movies = Helper.GetSyncMoviesData(lImdbListMovies.Skip(i * pageSize).Take(pageSize).ToList()).Movies
};
var response = TraktAPI.AddItemsToList(lTraktCustomList.Ids.Trakt.ToString(), lTraktMovieSync);
if (response == null)
{
UIUtils.UpdateStatus("Failed to send custom list items for IMDb movies", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync custom list items for {0} movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
#endregion
#region Shows
UIUtils.UpdateStatus("Importing {0} shows into {1} custom list...", lImdbListShows.Count(), lListName);
pageSize = AppSettings.BatchSize;
pages = (int)Math.Ceiling((double)lImdbListShows.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb custom list shows...", i + 1, pages);
// create list sync object to hold list items
var lTraktShowSync = new TraktSyncAll
{
Shows = Helper.GetSyncShowsData(lImdbListShows.Skip(i * pageSize).Take(pageSize).ToList()).Shows
};
var response = TraktAPI.AddItemsToList(lTraktCustomList.Ids.Trakt.ToString(), lTraktShowSync);
if (response == null)
{
UIUtils.UpdateStatus("Failed to send custom list items for IMDb shows", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync custom list items for {0} shows as they're not found on trakt.tv!", response.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
#endregion
}
}
#endregion
}
public void ImportRatings()
{
importCancelled = false;
List <Dictionary <string, string> > watchlistMovies = new List <Dictionary <string, string> >();
List <Dictionary <string, string> > watchlistShows = new List <Dictionary <string, string> >();
List <Dictionary <string, string> > ratedItems = new List <Dictionary <string, string> >();
List <Dictionary <string, string> > watchlistItems = new List <Dictionary <string, string> >();
#region Download Data
UIUtils.UpdateStatus("Requesting ratings from IMDb.com...");
int count;
int movieIndex = 1;
int movieIncrement = 250;
#region Ratings
do
{
count = ratedItems.Count;
UIUtils.UpdateStatus("Requesting ratings {0} - {1}, Total Results: {2}", movieIndex, (movieIncrement + movieIndex - 1), count);
string url = "http://www.imdb.com/user/" + username + "/ratings?start=" + movieIndex + "&view=compact";
string response = TraktWeb.Transmit(url, null, false);
if (response == null)
{
break;
}
int begin = 0;
// only log response when set to trace as it's very verbose in this case
if (AppSettings.LogSeverityLevel >= AppSettings.LoggingSeverity.Trace)
{
FileLog.Trace("Response: {0}", response);
}
while ((begin = response.IndexOf("<tr data-item-id", begin)) > 0)
{
var rateItem = new Dictionary <string, string>();
string sub = response.Substring(begin, response.IndexOf("</tr>", begin) - begin);
Regex reg = new Regex("<td class=\"title[^\"]*\"><a href=\"/title/(?<cIMDbID>tt\\d+)/[^\"]*\">(?<cTitle>[^<]+)</a>(?:\\s*<br>\\s*Episode:\\s*<a href=\"/title/(?<cEpisodeID>tt\\d+)/[^\"]*\">(?<cEpisodeTitle>[^<]+)</a>)?</td>");
// Get IMDb ID
var find = reg.Match(sub);
rateItem.Add(IMDbFieldMapping.cIMDbID, find.Groups["cIMDbID"].ToString());
// Get Title
// If it's a TV Episode then include both show and episode title
if (!string.IsNullOrEmpty(find.Groups["cEpisodeTitle"].ToString()))
{
rateItem.Add(IMDbFieldMapping.cTitle, string.Format("{0}: {1}", find.Groups["cTitle"], find.Groups["cEpisodeTitle"]));
}
else
{
rateItem.Add(IMDbFieldMapping.cTitle, find.Groups["cTitle"].ToString());
}
// Get User Rating
reg = new Regex("<td class=\"your_ratings\">\\n <a>([1-9][0-9]{0,1})</a>\\n</td>");
find = reg.Match(sub);
rateItem.Add(IMDbFieldMapping.cRating, find.Groups[1].ToString());
// Get Year
reg = new Regex("<td class=\"year\">([1-2][0-9]{3})</td>");
find = reg.Match(sub);
rateItem.Add(IMDbFieldMapping.cYear, find.Groups[1].ToString());
// Get Type
reg = new Regex("<td class=\"title_type\"> (.*)</td>");
find = reg.Match(sub);
if (find.Groups[1].ToString() == string.Empty)
{
rateItem.Add(IMDbFieldMapping.cType, "Feature Film");
}
else
{
rateItem.Add(IMDbFieldMapping.cType, find.Groups[1].ToString());
}
// Set provider to web or csv
rateItem.Add(IMDbFieldMapping.cProvider, "web");
ratedItems.Add(rateItem);
begin += 10;
}
// fetch next page
movieIndex += movieIncrement;
}while (count < ratedItems.Count);
#endregion
#region Watchlist
if (AppSettings.IMDbSyncWatchlist)
{
UIUtils.UpdateStatus("Reading IMDb watchlist from web...");
movieIndex = 1;
movieIncrement = 100;
do
{
count = watchlistItems.Count;
UIUtils.UpdateStatus("Requesting watchlist items {0} - {1}, Total Results: {2}", movieIndex, (movieIncrement + movieIndex - 1), count);
string url = "http://www.imdb.com/user/" + username + "/watchlist?start=" + movieIndex + "&view=compact";
string response = TraktWeb.Transmit(url, null, false);
if (response == null)
{
break;
}
int begin = 0;
// only log response when set to trace as it's very verbose in this case
if (AppSettings.LogSeverityLevel >= AppSettings.LoggingSeverity.Trace)
{
FileLog.Trace("Response: {0}", response);
}
if (response == null)
{
continue;
}
while ((begin = response.IndexOf("<tr data-item-id", begin)) > 0)
{
var watchListItem = new Dictionary <string, string>();
var sub = response.Substring(begin, response.IndexOf("</tr>", begin) - begin);
Regex reg = new Regex("<td class=\"title[^\"]*\"><a href=\"/title/(?<cIMDbID>tt\\d+)/[^\"]*\">(?<cTitle>[^<]+)</a>(?:\\s*<br>\\s*Episode:\\s*<a href=\"/title/(?<cEpisodeID>tt\\d+)/[^\"]*\">(?<cEpisodeTitle>[^<]+)</a>)?</td>");
// Get IMDb ID
var find = reg.Match(sub);
watchListItem.Add(IMDbFieldMapping.cIMDbID, find.Groups["cIMDbID"].ToString());
// Get Title
// If it's a TV Episode then include both show and episode title
if (!string.IsNullOrEmpty(find.Groups["cEpisodeTitle"].ToString()))
{
watchListItem.Add(IMDbFieldMapping.cTitle, string.Format("{0}: {1}", find.Groups["cTitle"], find.Groups["cEpisodeTitle"]));
}
else
{
watchListItem.Add(IMDbFieldMapping.cTitle, find.Groups["cTitle"].ToString());
}
// Get Year
reg = new Regex("<td class=\"year\">([1-2][0-9]{3})</td>");
find = reg.Match(sub);
watchListItem.Add(IMDbFieldMapping.cYear, find.Groups[1].ToString());
// Get Type
reg = new Regex("<td class=\"title_type\"> (.*)</td>");
find = reg.Match(sub);
if (find.Groups[1].ToString() == string.Empty)
{
watchListItem.Add(IMDbFieldMapping.cType, "Feature Film");
}
else
{
watchListItem.Add(IMDbFieldMapping.cType, find.Groups[1].ToString());
}
// Set provider to web or csv
watchListItem.Add(IMDbFieldMapping.cProvider, "web");
watchlistItems.Add(watchListItem);
begin += 10;
}
// fetch next page
movieIndex += movieIncrement;
}while (count < watchlistItems.Count);
}
#endregion
#endregion
if (importCancelled)
{
return;
}
#region Sync Ratings
#region Movies
var movies = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
if (movies.Any())
{
UIUtils.UpdateStatus("Retrieving existing movie ratings from trakt.tv");
var currentUserMovieRatings = TraktAPI.GetRatedMovies();
if (currentUserMovieRatings != null)
{
UIUtils.UpdateStatus(string.Format("Found {0} user movie ratings on trakt.tv", currentUserMovieRatings.Count()));
// Filter out movies to rate from existing ratings online
movies.RemoveAll(m => currentUserMovieRatings.Any(c => c.Movie.Ids.ImdbId == m[IMDbFieldMapping.cIMDbID]));
}
UIUtils.UpdateStatus("Importing {0} new IMDb movie ratings to trakt.tv", movies.Count());
if (movies.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)movies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated movies...", i + 1, pages);
TraktSyncResponse response = TraktAPI.AddMoviesToRatings(Helper.GetRateMoviesData(movies.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb movie ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
if (importCancelled)
{
return;
}
#endregion
#region TV Shows
var shows = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Show && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
if (shows.Any())
{
UIUtils.UpdateStatus("Retrieving existing tv show ratings from trakt.tv");
var currentUserShowRatings = TraktAPI.GetRatedShows();
if (currentUserShowRatings != null)
{
UIUtils.UpdateStatus("Found {0} user tv show ratings on trakt.tv", currentUserShowRatings.Count());
// Filter out shows to rate from existing ratings online
shows.RemoveAll(s => currentUserShowRatings.Any(c => (c.Show.Ids.ImdbId == s[IMDbFieldMapping.cIMDbID]) || (c.Show.Title == s[IMDbFieldMapping.cTitle] && c.Show.Year.ToString() == s[IMDbFieldMapping.cYear])));
}
if (shows.Count > 0)
{
UIUtils.UpdateStatus("Importing {0} new IMDb tv show ratings to trakt.tv", shows.Count());
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)shows.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated shows...", i + 1, pages);
TraktSyncResponse response = TraktAPI.AddShowsToRatings(Helper.GetRateShowsData(shows.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb show ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb shows as they're not found on trakt.tv!", response.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
if (importCancelled)
{
return;
}
#endregion
#region Episodes
var imdbEpisodes = new List <IMDbEpisode>();
var episodes = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Episode).ToList();
if (episodes.Any())
{
// we can't rely on the imdb id as trakt most likely wont have the info for episodes
// search and cache all series info needed for syncing
// use the tvdb API to first search for each unique series name
// then GetSeries by TVDb ID to get a list of all episodes
// each episode will have TVDb ID which we can use for syncing.
imdbEpisodes.AddRange(episodes.Select(Helper.GetIMDbEpisodeFromTVDb).Where(imdbEpisode => imdbEpisode != null));
UIUtils.UpdateStatus("Retrieving existing tv episode ratings from trakt.tv");
var currentUserEpisodeRatings = TraktAPI.GetRatedEpisodes();
if (currentUserEpisodeRatings != null)
{
UIUtils.UpdateStatus("Found {0} user tv episode ratings on trakt.tv", currentUserEpisodeRatings.Count());
// Filter out episodes to rate from existing ratings online
imdbEpisodes.RemoveAll(e => currentUserEpisodeRatings.Any(c => c.Episode.Ids.TvdbId == e.TvdbId));
}
UIUtils.UpdateStatus("Importing {0} episode ratings to trakt.tv", imdbEpisodes.Count());
if (imdbEpisodes.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)imdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
var episodesRated = Helper.GetTraktEpisodeRateData(imdbEpisodes.Skip(i * pageSize).Take(pageSize));
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated episodes...", i + 1, pages);
var response = TraktAPI.AddsEpisodesToRatings(episodesRated);
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb episodes as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
if (importCancelled)
{
return;
}
#endregion
#endregion
#region Mark as Watched
IEnumerable <TraktMoviePlays> watchedTraktMovies = null;
if (AppSettings.MarkAsWatched)
{
#region Movies
// compare all movies rated against what's not watched on trakt
movies = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
if (movies.Count > 0)
{
// get watched movies from trakt.tv
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
watchedTraktMovies = TraktAPI.GetWatchedMovies();
if (watchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv, skipping watched movie import", true);
Thread.Sleep(2000);
}
else
{
if (importCancelled)
{
return;
}
UIUtils.UpdateStatus("Found {0} watched movies on trakt", watchedTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watched movies that are already watched on trakt.tv");
movies.RemoveAll(w => watchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
// mark all rated movies as watched
UIUtils.UpdateStatus("Importing {0} IMDb movies as watched...", movies.Count);
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)movies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb movies as watched...", i + 1, pages);
var response = TraktAPI.AddMoviesToWatchedHistory(Helper.GetSyncWatchedMoviesData(movies.Skip(i * pageSize).Take(pageSize).ToList()));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watched status for IMDb movies", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watched state for {0} IMDb movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
#endregion
#region Episodes
if (imdbEpisodes != null && imdbEpisodes.Any())
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)imdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
var episodesWatched = Helper.GetTraktEpisodeWatchedData(imdbEpisodes.Skip(i * pageSize).Take(pageSize));
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb watched episodes...", i + 1, pages);
var response = TraktAPI.AddEpisodesToWatchedHistory(episodesWatched);
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes as watched to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync {0} IMDb episodes as watched, as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
#endregion
}
#endregion
#region Sync Watchlist
if (AppSettings.IMDbSyncWatchlist)
{
#region Movies
watchlistMovies.AddRange(watchlistItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie));
if (watchlistMovies.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlist movies from trakt...");
var watchlistTraktMovies = TraktAPI.GetWatchlistMovies();
if (watchlistTraktMovies != null)
{
UIUtils.UpdateStatus("Found {0} watchlist movies on trakt", watchlistTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watchlist movies that are already in watchlist on trakt.tv");
watchlistMovies.RemoveAll(w => watchlistTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && movies.Count > 0)
{
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
// get watched movies from trakt so we don't import movies into watchlist that are already watched
if (watchedTraktMovies != null)
{
watchedTraktMovies = TraktAPI.GetWatchedMovies();
if (watchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv", true);
Thread.Sleep(2000);
}
else
{
UIUtils.UpdateStatus("Found {0} watched movies on trakt", watchedTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watchlist movies that are watched on trakt.tv");
// remove movies from sync list which are watched already
watchlistMovies.RemoveAll(w => watchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
}
}
// add all movies to watchlist
UIUtils.UpdateStatus("Importing {0} IMDb watchlist movies to trakt.tv ...", watchlistMovies.Count());
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)watchlistMovies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb movies into watchlist...", i + 1, pages);
var response = TraktAPI.AddMoviesToWatchlist(Helper.GetSyncMoviesData(watchlistMovies.Skip(i * pageSize).Take(pageSize).ToList()));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb movies.", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} IMDb movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
#endregion
#region TV Shows
IEnumerable <TraktShowPlays> watchedTraktShows = null;
watchlistShows.AddRange(watchlistItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Show));
if (watchlistShows.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlist shows from trakt...");
var watchlistTraktShows = TraktAPI.GetWatchlistShows();
if (watchlistTraktShows != null)
{
UIUtils.UpdateStatus("Found {0} watchlist shows on trakt", watchlistTraktShows.Count());
UIUtils.UpdateStatus("Filtering out watchlist shows that are already in watchlist on trakt.tv");
watchlistShows.RemoveAll(w => watchlistTraktShows.FirstOrDefault(t => t.Show.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Show.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle].ToLowerInvariant() && t.Show.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && shows.Count > 0)
{
UIUtils.UpdateStatus("Requesting watched shows from trakt...");
// get watched movies from trakt so we don't import shows into watchlist that are already watched
watchedTraktShows = TraktAPI.GetWatchedShows();
if (watchedTraktShows != null)
{
UIUtils.UpdateStatus("Found {0} watched shows on trakt", watchedTraktShows.Count());
UIUtils.UpdateStatus("Filtering out watchlist shows containing watched episodes on trakt.tv");
// remove shows from sync list which are watched already
watchlistShows.RemoveAll(w => watchedTraktShows.Count(t => (t.Show.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID]) || (t.Show.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle].ToLowerInvariant() && t.Show.Year.ToString() == w[IMDbFieldMapping.cYear])) != 0);
}
}
//add all shows to watchlist
UIUtils.UpdateStatus("Importing {0} IMDb watchlist shows to trakt.tv...", watchlistShows.Count());
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)watchlistShows.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb shows into watchlist...", i + 1, pages);
var response = TraktAPI.AddShowsToWatchlist(Helper.GetSyncShowsData(watchlistShows.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb tv shows", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} IMDb shows as they're not found on trakt.tv!", response.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
#endregion
#region Episodes
imdbEpisodes.Clear();
episodes = watchlistItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Episode).ToList();
if (episodes.Any())
{
UIUtils.UpdateStatus("Found {0} IMDb watchlist episodes", episodes.Count());
imdbEpisodes.AddRange(episodes.Select(Helper.GetIMDbEpisodeFromTVDb).Where(imdbEpisode => imdbEpisode != null));
// filter out existing watchlist episodes
UIUtils.UpdateStatus("Requesting existing watchlist episodes from trakt...");
var watchlistTraktEpisodes = TraktAPI.GetWatchlistEpisodes();
if (watchlistTraktEpisodes != null)
{
UIUtils.UpdateStatus("Found {0} watchlist episodes on trakt", watchlistTraktEpisodes.Count());
UIUtils.UpdateStatus("Filtering out watchlist episodes that are already in watchlist on trakt.tv");
imdbEpisodes.RemoveAll(e => watchlistTraktEpisodes.FirstOrDefault(w => w.Episode.Ids.ImdbId == e.ImdbId || w.Episode.Ids.TvdbId == e.TvdbId) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && episodes.Count > 0)
{
// we already might have it from the shows sync
if (watchedTraktShows == null)
{
UIUtils.UpdateStatus("Requesting watched episodes from trakt...");
// get watched episodes from trakt so we don't import episodes into watchlist that are already watched
watchedTraktShows = TraktAPI.GetWatchedShows();
}
if (watchedTraktShows != null)
{
UIUtils.UpdateStatus("Filtering out watchlist episodes containing watched episodes on trakt.tv.");
imdbEpisodes.RemoveAll(e => watchedTraktShows.Where(s => s.Show.Ids.ImdbId == e.ImdbId)
.Any(s => s.Seasons.Exists(se => se.Number == e.SeasonNumber && se.Episodes.Exists(ep => ep.Number == e.EpisodeNumber))));
}
}
UIUtils.UpdateStatus("Importing {0} episodes in watchlist to trakt.tv", imdbEpisodes.Count());
if (imdbEpisodes.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)imdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus(string.Format("Importing page {0}/{1} IMDb watchlist episodes...", i + 1, pages));
var response = TraktAPI.AddEpisodesToWatchlist(Helper.GetTraktEpisodeData(imdbEpisodes.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episode watchlist to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} IMDb episodes as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
#endregion
}
#endregion
}
public void ImportRatings()
{
importCancelled = false;
List <Dictionary <string, string> > watchlistMovies = new List <Dictionary <string, string> >();
List <Dictionary <string, string> > watchlistShows = new List <Dictionary <string, string> >();
List <Dictionary <string, string> > ratedItems = new List <Dictionary <string, string> >();
List <Dictionary <string, string> > watchlistItems = new List <Dictionary <string, string> >();
List <IMDbListItem> lWatchlistItems = new List <IMDbListItem>();
#region Download Data
UIUtils.UpdateStatus("Requesting ratings from IMDb.com...");
int movieIndex = 0;
int movieIncrement = 100;
string paginationKey = string.Empty;
Regex reg;
Match find;
#region Ratings
do
{
UIUtils.UpdateStatus("Requesting ratings {0} - {1}, Total Results: {2}", movieIndex + 1, (movieIncrement + movieIndex), ratedItems.Count);
string url = "http://www.imdb.com/user/" + username + "/ratings?sort=date_added%2Cdesc&mode=detail&lastPosition=" + movieIndex
+ (string.IsNullOrEmpty(paginationKey) ? string.Empty : "&paginationKey=" + paginationKey);
string response = TraktWeb.Transmit(url, null, false);
if (response == null)
{
break;
}
int begin = 0;
// only log response when set to trace as it's very verbose in this case
if (AppSettings.LogSeverityLevel >= AppSettings.LoggingSeverity.Trace)
{
FileLog.Trace("Response: {0}", response);
}
while ((begin = response.IndexOf("lister-item-content", begin)) > 0)
{
var rateItem = new Dictionary <string, string>();
string sub = response.Substring(begin, response.IndexOf("class=\"clear\"", begin) - begin);
reg = new Regex("<a href=\"/title/(?<cIMDbID>tt\\d+)/[^\"]*\"\n>(?<cTitle>[^<]*)</a>(?:[.\\s\\S]*Episode:</small>\\s*<a href=\"/title/(?<cEpisodeID>tt\\d+)/[^\"]*\"\n*>(?<cEpisodeTitle>[^<]+)</a>){0,1}");
// Get IMDb ID
find = reg.Match(sub);
rateItem.Add(IMDbFieldMapping.cIMDbID, find.Groups["cIMDbID"].ToString());
// Get Title
// If it's a TV Episode then include both show and episode title
if (!string.IsNullOrEmpty(find.Groups["cEpisodeTitle"].ToString()))
{
rateItem.Add(IMDbFieldMapping.cTitle, string.Format("{0}: {1}", find.Groups["cTitle"], find.Groups["cEpisodeTitle"]));
}
else
{
rateItem.Add(IMDbFieldMapping.cTitle, find.Groups["cTitle"].ToString());
}
// Get User Rating
reg = new Regex("<span class=\"ipl-rating-star__rating\">([1-9][0-9]{0,1})</span>");
find = reg.Match(sub);
rateItem.Add(IMDbFieldMapping.cRating, find.Groups[1].ToString());
// Get Year
reg = new Regex("<span class=\"lister-item-year text-muted unbold\">\\(([1-2][0-9]{3}).*\\)</span>");
find = reg.Match(sub);
rateItem.Add(IMDbFieldMapping.cYear, find.Groups[1].ToString());
// Get Type
reg = new Regex("<span class=\"lister-item-year text-muted unbold\">\\(([1-2][0-9]{3}–).*\\)</span>");
find = reg.Match(sub);
if (find.Groups[1].ToString() == string.Empty)
{
rateItem.Add(IMDbFieldMapping.cType, "Feature Film");
}
else
{
rateItem.Add(IMDbFieldMapping.cType, "tvseries");
}
// Set provider to web or csv
rateItem.Add(IMDbFieldMapping.cProvider, "web");
ratedItems.Add(rateItem);
begin += 10;
}
// fetch next page
movieIndex += movieIncrement;
reg = new Regex("<a class=\"flat-button lister-page-next next-page\" href=.*&paginationKey=(?<cPaginationKey>.*)&");
find = reg.Match(response);
paginationKey = find.Groups["cPaginationKey"].ToString();
}while (movieIndex == ratedItems.Count);
#endregion
#region Watchlist
if (AppSettings.IMDbSyncWatchlist)
{
UIUtils.UpdateStatus("Reading IMDb watchlist from web...");
string url = "http://www.imdb.com/user/" + username + "/watchlist";
string response = TraktWeb.Transmit(url, null, false);
if (response != null)
{
var sub = response.Substring(0, response.IndexOf("</head>", 0));
reg = new Regex("ls\\d+");
// Get IMDb Watchlist ID
find = reg.Match(sub);
url = "http://www.imdb.com/list/" + find.Value + "/export";
using (StreamReader reader = TraktWeb.GetCsvStream(url))
{
if (reader == null)
{
UIUtils.UpdateStatus("Failed to download watchlist from IMDb.", true);
Thread.Sleep(2000);
}
else
{
lWatchlistItems = ParseCsvFile <IMDbListItem>(reader);
if (lWatchlistItems == null)
{
UIUtils.UpdateStatus("Failed to parse IMDb watchlist file!", true);
Thread.Sleep(2000);
}
}
}
}
// only log response when set to trace as it's very verbose in this case
if (AppSettings.LogSeverityLevel >= AppSettings.LoggingSeverity.Trace)
{
FileLog.Trace("Response: {0}", response);
}
}
#endregion
#endregion
if (importCancelled)
{
return;
}
#region Sync Ratings
#region Movies
var movies = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
if (movies.Any())
{
UIUtils.UpdateStatus("Retrieving existing movie ratings from trakt.tv");
var currentUserMovieRatings = TraktAPI.GetRatedMovies();
if (currentUserMovieRatings != null)
{
UIUtils.UpdateStatus(string.Format("Found {0} user movie ratings on trakt.tv", currentUserMovieRatings.Count()));
// Filter out movies to rate from existing ratings online
movies.RemoveAll(m => currentUserMovieRatings.Any(c => c.Movie.Ids.ImdbId == m[IMDbFieldMapping.cIMDbID]));
}
UIUtils.UpdateStatus("Importing {0} new IMDb movie ratings to trakt.tv", movies.Count());
if (movies.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)movies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated movies...", i + 1, pages);
TraktSyncResponse response = TraktAPI.AddMoviesToRatings(Helper.GetRateMoviesData(movies.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb movie ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
if (importCancelled)
{
return;
}
#endregion
#region TV Shows
var shows = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Show && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
if (shows.Any())
{
UIUtils.UpdateStatus("Retrieving existing tv show ratings from trakt.tv");
var currentUserShowRatings = TraktAPI.GetRatedShows();
if (currentUserShowRatings != null)
{
UIUtils.UpdateStatus("Found {0} user tv show ratings on trakt.tv", currentUserShowRatings.Count());
// Filter out shows to rate from existing ratings online
shows.RemoveAll(s => currentUserShowRatings.Any(c => (c.Show.Ids.ImdbId == s[IMDbFieldMapping.cIMDbID]) || (c.Show.Title == s[IMDbFieldMapping.cTitle] && c.Show.Year.ToString() == s[IMDbFieldMapping.cYear])));
}
if (shows.Count > 0)
{
UIUtils.UpdateStatus("Importing {0} new IMDb tv show ratings to trakt.tv", shows.Count());
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)shows.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated shows...", i + 1, pages);
TraktSyncResponse response = TraktAPI.AddShowsToRatings(Helper.GetRateShowsData(shows.Skip(i * pageSize).Take(pageSize)));
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb show ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb shows as they're not found on trakt.tv!", response.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
if (importCancelled)
{
return;
}
#endregion
#region Episodes
var imdbEpisodes = new List <IMDbEpisode>();
var episodes = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Episode).ToList();
if (episodes.Any())
{
// we can't rely on the imdb id as trakt most likely wont have the info for episodes
// search and cache all series info needed for syncing
// use the tvdb API to first search for each unique series name
// then GetSeries by TVDb ID to get a list of all episodes
// each episode will have TVDb ID which we can use for syncing.
imdbEpisodes.AddRange(episodes.Select(Helper.GetIMDbEpisodeFromTrakt).Where(imdbEpisode => imdbEpisode != null));
UIUtils.UpdateStatus("Retrieving existing tv episode ratings from trakt.tv");
var currentUserEpisodeRatings = TraktAPI.GetRatedEpisodes();
if (currentUserEpisodeRatings != null)
{
UIUtils.UpdateStatus("Found {0} user tv episode ratings on trakt.tv", currentUserEpisodeRatings.Count());
// Filter out episodes to rate from existing ratings online
imdbEpisodes.RemoveAll(e => currentUserEpisodeRatings.Any(c => c.Episode.Ids.Trakt == e.TraktId));
}
UIUtils.UpdateStatus("Importing {0} episode ratings to trakt.tv", imdbEpisodes.Count());
if (imdbEpisodes.Count > 0)
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)imdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
var episodesRated = Helper.GetTraktEpisodeRateData(imdbEpisodes.Skip(i * pageSize).Take(pageSize));
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb rated episodes...", i + 1, pages);
var response = TraktAPI.AddsEpisodesToRatings(episodesRated);
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync ratings for {0} IMDb episodes as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
if (importCancelled)
{
return;
}
#endregion
#endregion
#region Mark as Watched
IEnumerable <TraktMoviePlays> watchedTraktMovies = null;
if (AppSettings.MarkAsWatched)
{
#region Movies
// compare all movies rated against what's not watched on trakt
movies = ratedItems.Where(r => r[IMDbFieldMapping.cType].ItemType() == IMDbType.Movie && !string.IsNullOrEmpty(r[IMDbFieldMapping.cRating])).ToList();
if (movies.Count > 0)
{
// get watched movies from trakt.tv
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
watchedTraktMovies = TraktAPI.GetWatchedMovies();
if (watchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv, skipping watched movie import", true);
Thread.Sleep(2000);
}
else
{
if (importCancelled)
{
return;
}
UIUtils.UpdateStatus("Found {0} watched movies on trakt", watchedTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watched movies that are already watched on trakt.tv");
movies.RemoveAll(w => watchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w[IMDbFieldMapping.cIMDbID] || (t.Movie.Title.ToLowerInvariant() == w[IMDbFieldMapping.cTitle] && t.Movie.Year.ToString() == w[IMDbFieldMapping.cYear])) != null);
// mark all rated movies as watched
UIUtils.UpdateStatus("Importing {0} IMDb movies as watched...", movies.Count);
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)movies.Count / pageSize);
for (int i = 0; i < pages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb movies as watched...", i + 1, pages);
var response = TraktAPI.AddMoviesToWatchedHistory(Helper.GetSyncWatchedMoviesData(movies.Skip(i * pageSize).Take(pageSize).ToList()));
if (response == null)
{
UIUtils.UpdateStatus("Failed to send watched status for IMDb movies", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watched state for {0} IMDb movies as they're not found on trakt.tv!", response.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
#endregion
#region Episodes
if (imdbEpisodes != null && imdbEpisodes.Any())
{
int pageSize = AppSettings.BatchSize;
int pages = (int)Math.Ceiling((double)imdbEpisodes.Count / pageSize);
for (int i = 0; i < pages; i++)
{
var episodesWatched = Helper.GetTraktEpisodeWatchedData(imdbEpisodes.Skip(i * pageSize).Take(pageSize));
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb watched episodes...", i + 1, pages);
var response = TraktAPI.AddEpisodesToWatchedHistory(episodesWatched);
if (response == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes as watched to trakt.tv", true);
Thread.Sleep(2000);
}
else if (response.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync {0} IMDb episodes as watched, as they're not found on trakt.tv!", response.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
#endregion
}
#endregion
#region Sync Watchlist
if (AppSettings.IMDbSyncWatchlist)
{
#region Movies
var lWatchlistedMovies = lWatchlistItems.Where(r => r.Type.ItemType() == IMDbType.Movie)
.Select(s => s.ToTraktMovie()).ToList();
if (lWatchlistedMovies.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlist movies from trakt...");
var lWatchlistTraktMovies = TraktAPI.GetWatchlistMovies();
if (lWatchlistTraktMovies != null)
{
UIUtils.UpdateStatus($"Found {0} watchlist movies on trakt", lWatchlistTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watchlist movies that are already in watchlist on trakt.tv");
lWatchlistedMovies.RemoveAll(w => lWatchlistTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w.Ids.ImdbId ||
(t.Movie.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Movie.Year == w.Year)) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lWatchlistedMovies.Count > 0)
{
// get watched movies from trakt so we don't import movies into watchlist that are already watched
// we may already have this if we imported rated items as watched
if (watchedTraktMovies == null)
{
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
watchedTraktMovies = TraktAPI.GetWatchedMovies();
if (watchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv", true);
Thread.Sleep(2000);
}
}
if (watchedTraktMovies != null)
{
UIUtils.UpdateStatus($"Found {0} watched movies on trakt", watchedTraktMovies.Count());
UIUtils.UpdateStatus("Filtering out watchlist movies that are watched on trakt.tv");
// remove movies from sync list which are watched already
lWatchlistedMovies.RemoveAll(w => watchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w.Ids.ImdbId ||
(t.Movie.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Movie.Year == w.Year)) != null);
}
}
// add movies to watchlist
UIUtils.UpdateStatus($"Importing {0} IMDb watchlist movies to trakt.tv...", lWatchlistedMovies.Count());
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lWatchlistedMovies.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {0}/{1} IMDb watchlist movies...", i + 1, lPages);
var lWatchlistedToSync = new TraktMovieSync()
{
Movies = lWatchlistedMovies.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddMoviesToWatchlist(lWatchlistedToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb movies", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync watchlist for {0} movies as they're not found on trakt.tv!", lResponse.NotFound.Movies.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
if (importCancelled)
{
return;
}
#endregion
#region TV Shows
IEnumerable <TraktShowPlays> lWatchedTraktShows = null;
var lWatchlistedShows = lWatchlistItems.Where(r => r.Type.ItemType() == IMDbType.Show)
.Select(s => s.ToTraktShow()).ToList();
if (lWatchlistedShows.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlisted shows from trakt...");
var lWatchlistTraktShows = TraktAPI.GetWatchlistShows();
if (lWatchlistTraktShows != null)
{
UIUtils.UpdateStatus($"Found {0} watchlist shows on trakt", lWatchlistTraktShows.Count());
UIUtils.UpdateStatus("Filtering out watchlist shows that are already in watchlist on trakt.tv");
lWatchlistedShows.RemoveAll(w => lWatchlistTraktShows.FirstOrDefault(t => t.Show.Ids.ImdbId == w.Ids.ImdbId ||
(t.Show.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Show.Year == w.Year)) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lWatchlistedShows.Count > 0)
{
UIUtils.UpdateStatus("Requesting watched shows from trakt...");
// get watched movies from trakt so we don't import shows into watchlist that are already watched
lWatchedTraktShows = TraktAPI.GetWatchedShows();
if (lWatchedTraktShows != null)
{
UIUtils.UpdateStatus($"Found {0} watched shows on trakt", lWatchedTraktShows.Count());
UIUtils.UpdateStatus("Filtering out watchlist shows containing watched episodes on trakt.tv.");
// remove shows from sync list which are watched already
lWatchlistedShows.RemoveAll(w => lWatchedTraktShows.FirstOrDefault(t => (t.Show.Ids.ImdbId == w.Ids.ImdbId) ||
(t.Show.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Show.Year == w.Year)) != null);
}
}
// add shows to watchlist
UIUtils.UpdateStatus($"Importing {0} IMDb watchlist shows to trakt.tv...", lWatchlistedShows.Count());
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lWatchlistedShows.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {0}/{1} IMDb watchlist shows...", i + 1, lPages);
var lWatchlistShowsToSync = new TraktShowSync
{
Shows = lWatchlistedShows.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddShowsToWatchlist(lWatchlistShowsToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb tv shows", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync watchlist for {0} shows as they're not found on trakt.tv!", lResponse.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
if (importCancelled)
{
return;
}
#endregion
#region Episodes
var lImdbEpisodes = new List <IMDbEpisode>();
var lImdbWatchlistedEpisodes = lWatchlistItems.Where(r => r.Type.ItemType() == IMDbType.Episode).ToList();
if (lImdbWatchlistedEpisodes.Any())
{
UIUtils.UpdateStatus($"Found {0} IMDb watchlisted episodes", lImdbWatchlistedEpisodes.Count());
lImdbEpisodes.AddRange(lImdbWatchlistedEpisodes.Select(Helper.GetIMDbEpisodeFromTrakt).Where(imdbEpisode => imdbEpisode != null));
if (importCancelled)
{
return;
}
// filter out existing watchlist episodes
UIUtils.UpdateStatus("Requesting existing watchlist episodes from trakt...");
var lWatchlistedTraktEpisodes = TraktAPI.GetWatchlistEpisodes();
if (lWatchlistedTraktEpisodes != null)
{
UIUtils.UpdateStatus($"Found {0} watchlist episodes on trakt", lWatchlistedTraktEpisodes.Count());
UIUtils.UpdateStatus("Filtering out watchlist episodes that are already in watchlist on trakt.tv");
lImdbEpisodes.RemoveAll(e => lWatchlistedTraktEpisodes.FirstOrDefault(w => w.Episode.Ids.Trakt == e.TraktId) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lImdbEpisodes.Count > 0)
{
// we already might have it from the shows sync
if (lWatchedTraktShows == null)
{
UIUtils.UpdateStatus("Requesting watched episodes from trakt...");
// get watched episodes from trakt so we don't import episodes into watchlist that are already watched
lWatchedTraktShows = TraktAPI.GetWatchedShows();
}
if (lWatchedTraktShows != null)
{
UIUtils.UpdateStatus("Filtering out watchlist episodes containing watched episodes on trakt.tv");
// this wont work atm due to show IMDb ID not being set in the IMDbEpisode object
lImdbEpisodes.RemoveAll(e => lWatchedTraktShows.Where(s => s.Show.Ids.ImdbId == e.ShowImdbId)
.Any(s => s.Seasons.Exists(se => se.Number == e.SeasonNumber &&
se.Episodes.Exists(ep => ep.Number == e.EpisodeNumber))));
}
}
UIUtils.UpdateStatus($"Importing {0} episodes in watchlist to trakt.tv", lImdbEpisodes.Count());
if (lImdbEpisodes.Count > 0)
{
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lImdbEpisodes.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {0}/{1} IMDb watchlisted episodes...", i + 1, lPages);
var lResponse = TraktAPI.AddEpisodesToWatchlist(Helper.GetTraktEpisodeData(lImdbEpisodes.Skip(i * lPageSize).Take(lPageSize)));
if (lResponse == null)
{
UIUtils.UpdateStatus("Error importing IMDb episode watchlist to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} IMDb episodes as they're not found on trakt.tv!", lResponse.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (importCancelled)
{
return;
}
}
}
}
#endregion
}
#endregion
}
public void ImportRatings()
{
mImportCancelled = false;
var lRatedCsvItems = new List <IMDbRateItem>();
var lWatchlistCsvItems = new List <IMDbListItem>();
var lCustomLists = new Dictionary <string, List <IMDbListItem> >();
#region Parse Ratings CSV
UIUtils.UpdateStatus("Reading IMDb ratings export...");
if (mImportCsvRatings)
{
mCsvConfiguration.RegisterClassMap <IMDbRatingCsvMap>();
lRatedCsvItems = ParseCsvFile <IMDbRateItem>(mRatingsFileCsv);
if (lRatedCsvItems == null)
{
UIUtils.UpdateStatus("Failed to parse IMDb ratings file!", true);
Thread.Sleep(2000);
return;
}
mCsvConfiguration.UnregisterClassMap <IMDbRatingCsvMap>();
}
if (mImportCancelled)
{
return;
}
#endregion
#region Parse Watchlist CSV
UIUtils.UpdateStatus("Reading IMDb watchlist export...");
if (mImportCsvWatchlist)
{
mCsvConfiguration.RegisterClassMap <IMDbListCsvMap>();
lWatchlistCsvItems = ParseCsvFile <IMDbListItem>(mWatchlistFileCsv);
if (lWatchlistCsvItems == null)
{
UIUtils.UpdateStatus("Failed to parse IMDb watchlist file!", true);
Thread.Sleep(2000);
return;
}
mCsvConfiguration.UnregisterClassMap <IMDbListCsvMap>();
}
if (mImportCancelled)
{
return;
}
#endregion
#region Parse Custom List CSVs
UIUtils.UpdateStatus("Reading IMDb custom lists export...");
if (mImportCsvCustomLists)
{
mCsvConfiguration.RegisterClassMap <IMDbListCsvMap>();
foreach (var list in mCustomListsCsvs)
{
UIUtils.UpdateStatus($"Reading IMDb custom list '{list}'");
var lListCsvItems = ParseCsvFile <IMDbListItem>(list);
if (lListCsvItems == null)
{
UIUtils.UpdateStatus("Failed to parse IMDb custom list file!", true);
Thread.Sleep(2000);
continue;
}
lCustomLists.Add(list, lListCsvItems);
}
mCsvConfiguration.UnregisterClassMap <IMDbListCsvMap>();
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Rated Movies
var lRatedCsvMovies = lRatedCsvItems.Where(r => r.Type.ItemType() == IMDbType.Movie && r.MyRating != null)
.Select(s => s.ToTraktRatedMovie()).ToList();
FileLog.Info($"Found {lRatedCsvMovies.Count} movie ratings in CSV file");
if (lRatedCsvMovies.Any())
{
UIUtils.UpdateStatus("Retrieving existing movie ratings from trakt.tv");
var lCurrentUserMovieRatings = TraktAPI.GetRatedMovies();
if (lCurrentUserMovieRatings != null)
{
UIUtils.UpdateStatus($"Found {lCurrentUserMovieRatings.Count()} user movie ratings on trakt.tv");
// Filter out movies to rate from existing ratings online
lRatedCsvMovies.RemoveAll(m => lCurrentUserMovieRatings.Any(c => c.Movie.Ids.ImdbId == m.Ids.ImdbId));
}
UIUtils.UpdateStatus($"Importing {lRatedCsvMovies.Count()} new movie ratings to trakt.tv");
if (lRatedCsvMovies.Count > 0)
{
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lRatedCsvMovies.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb rated movies...");
var lRatingsToSync = new TraktMovieRatingSync()
{
movies = lRatedCsvMovies.Skip(i * lPageSize).Take(lPageSize).ToList()
};
TraktSyncResponse lResponse = TraktAPI.AddMoviesToRatings(lRatingsToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Error importing IMDb movie ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync ratings for {lResponse.NotFound.Movies.Count} movies as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Rated TV Shows
var lRatedCsvShows = lRatedCsvItems.Where(r => r.Type.ItemType() == IMDbType.Show && r.MyRating != null)
.Select(s => s.ToTraktRatedShow()).ToList();
FileLog.Info($"Found {lRatedCsvShows.Count} tv show ratings in CSV file");
if (lRatedCsvShows.Any())
{
UIUtils.UpdateStatus("Retrieving existing tv show ratings from trakt.tv");
var lCurrentUserShowRatings = TraktAPI.GetRatedShows();
if (lCurrentUserShowRatings != null)
{
UIUtils.UpdateStatus($"Found {lCurrentUserShowRatings.Count()} user tv show ratings on trakt.tv");
// Filter out shows to rate from existing ratings online
lRatedCsvShows.RemoveAll(s => lCurrentUserShowRatings.Any(c => (c.Show.Ids.ImdbId == s.Ids.ImdbId) ||
(c.Show.Title.ToLowerInvariant() == s.Title.ToLowerInvariant() && c.Show.Year == s.Year)));
}
UIUtils.UpdateStatus($"Importing {lRatedCsvShows.Count()} tv show ratings to trakt.tv");
if (lRatedCsvShows.Count > 0)
{
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lRatedCsvShows.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb rated shows...");
var lRatingsToSync = new TraktShowRatingSync()
{
shows = lRatedCsvShows.Skip(i * lPageSize).Take(lPageSize).ToList()
};
TraktSyncResponse lResponse = TraktAPI.AddShowsToRatings(lRatingsToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Error importing IMDb tv show ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync ratings for {lResponse.NotFound.Shows.Count} shows as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Rated Episodes
var lImdbRatedEpisodes = new List <IMDbEpisode>();
var lImdbRatedCsvEpisodes = lRatedCsvItems.Where(r => r.Type.ItemType() == IMDbType.Episode).ToList();
FileLog.Info($"Found {lImdbRatedCsvEpisodes.Count} tv episode ratings in CSV file");
if (lImdbRatedCsvEpisodes.Any())
{
// we can't rely on the episode imdb id as trakt most likely wont have that info at the episode level
lImdbRatedEpisodes.AddRange(lImdbRatedCsvEpisodes.Select(Helper.GetIMDbEpisodeFromTrakt).Where(imdbEpisode => imdbEpisode != null));
if (mImportCancelled)
{
return;
}
UIUtils.UpdateStatus("Retrieving existing tv episode ratings from trakt.tv");
var lCurrentUserEpisodeRatings = TraktAPI.GetRatedEpisodes();
if (lCurrentUserEpisodeRatings != null)
{
UIUtils.UpdateStatus($"Found {lCurrentUserEpisodeRatings.Count()} user tv episode ratings on trakt.tv");
// Filter out episodes to rate from existing ratings online
lImdbRatedEpisodes.RemoveAll(e => lCurrentUserEpisodeRatings.Any(c => c.Episode.Ids.Trakt == e.TraktId));
}
UIUtils.UpdateStatus($"Importing {lImdbRatedEpisodes.Count()} episode ratings to trakt.tv");
if (lImdbRatedEpisodes.Count > 0)
{
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lImdbRatedEpisodes.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
var lEpisodesRated = Helper.GetTraktEpisodeRateData(lImdbRatedEpisodes.Skip(i * lPageSize).Take(lPageSize));
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb rated episodes...");
var lResponse = TraktAPI.AddsEpisodesToRatings(lEpisodesRated);
if (lResponse == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes ratings to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync ratings for {lResponse.NotFound.Episodes.Count} IMDb episodes as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Mark Rated Items as Watched
IEnumerable <TraktMoviePlays> lWatchedTraktMovies = null;
if (AppSettings.MarkAsWatched)
{
#region Movies
var lWatchedCsvMovies = lRatedCsvItems.Where(r => r.Type.ItemType() == IMDbType.Movie)
.Select(s => s.ToTraktWatchedMovie()).ToList();
FileLog.Info("Found {0} movies in CSV file", lWatchedCsvMovies.Count);
// compare all movies rated against what's not watched on trakt
if (lWatchedCsvMovies.Count > 0)
{
// get watched movies from trakt.tv
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
lWatchedTraktMovies = TraktAPI.GetWatchedMovies();
if (lWatchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv, skipping watched movie import", true);
Thread.Sleep(2000);
}
else
{
if (mImportCancelled)
{
return;
}
UIUtils.UpdateStatus($"Found {lWatchedTraktMovies.Count()} watched movies on trakt");
UIUtils.UpdateStatus("Filtering out watched movies that are already watched on trakt.tv");
lWatchedCsvMovies.RemoveAll(w => lWatchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w.Ids.ImdbId ||
(t.Movie.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Movie.Year == w.Year)) != null);
// mark all rated movies as watched
UIUtils.UpdateStatus($"Importing {lWatchedCsvMovies.Count} IMDb movies as watched...");
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lRatedCsvMovies.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb movies as watched...");
var lMoviesToSync = new TraktMovieWatchedSync()
{
Movies = lWatchedCsvMovies.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddMoviesToWatchedHistory(lMoviesToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send watched status for IMDb movies to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync watched states for {lResponse.NotFound.Movies.Count} movies as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
#endregion
#region Episodes
if (lImdbRatedEpisodes != null && lImdbRatedEpisodes.Any())
{
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lImdbRatedEpisodes.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
var lEpisodesWatched = Helper.GetTraktEpisodeWatchedData(lImdbRatedEpisodes.Skip(i * lPageSize).Take(lPageSize));
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb watched episodes...");
var lResponse = TraktAPI.AddEpisodesToWatchedHistory(lEpisodesWatched);
if (lResponse == null)
{
UIUtils.UpdateStatus("Error importing IMDb episodes as watched to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync {lResponse.NotFound.Episodes.Count} IMDb episodes as watched, as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
#endregion
}
#endregion
#region Import Watchlist Movies
var lWatchlistedCsvMovies = lWatchlistCsvItems.Where(r => r.Type.ItemType() == IMDbType.Movie)
.Select(s => s.ToTraktMovie()).ToList();
FileLog.Info($"Found {lWatchlistedCsvMovies.Count} movies watchlisted in CSV file");
if (lWatchlistedCsvMovies.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlisted movies from trakt...");
var lWatchlistTraktMovies = TraktAPI.GetWatchlistMovies();
if (lWatchlistTraktMovies != null)
{
UIUtils.UpdateStatus($"Found {lWatchlistTraktMovies.Count()} watchlist movies on trakt");
UIUtils.UpdateStatus("Filtering out watchlist movies that are already in watchlist on trakt.tv");
lWatchlistedCsvMovies.RemoveAll(w => lWatchlistTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w.Ids.ImdbId ||
(t.Movie.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Movie.Year == w.Year)) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lWatchlistedCsvMovies.Count > 0)
{
// get watched movies from trakt so we don't import movies into watchlist that are already watched
// we may already have this if we imported rated items as watched
if (lWatchedTraktMovies != null)
{
UIUtils.UpdateStatus("Requesting watched movies from trakt...");
lWatchedTraktMovies = TraktAPI.GetWatchedMovies();
if (lWatchedTraktMovies == null)
{
UIUtils.UpdateStatus("Failed to get watched movies from trakt.tv", true);
Thread.Sleep(2000);
}
}
if (lWatchedTraktMovies != null)
{
UIUtils.UpdateStatus($"Found {lWatchedTraktMovies.Count()} watched movies on trakt");
UIUtils.UpdateStatus("Filtering out watchlist movies that are watched on trakt.tv");
// remove movies from sync list which are watched already
lWatchlistedCsvMovies.RemoveAll(w => lWatchedTraktMovies.FirstOrDefault(t => t.Movie.Ids.ImdbId == w.Ids.ImdbId ||
(t.Movie.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Movie.Year == w.Year)) != null);
}
}
// add movies to watchlist
UIUtils.UpdateStatus($"Importing {lWatchlistedCsvMovies.Count()} IMDb watchlist movies to trakt.tv...");
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lWatchlistedCsvMovies.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb watchlist movies...");
var lWatchlistedToSync = new TraktMovieSync()
{
Movies = lWatchlistedCsvMovies.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddMoviesToWatchlist(lWatchlistedToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb movies", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync watchlist for {lResponse.NotFound.Movies.Count} movies as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Watchlist TV Shows
IEnumerable <TraktShowPlays> lWatchedTraktShows = null;
var lWatchlistedCsvShows = lWatchlistCsvItems.Where(r => r.Type.ItemType() == IMDbType.Show)
.Select(s => s.ToTraktShow()).ToList();
FileLog.Info($"Found {lWatchlistedCsvShows.Count} tv shows watchlisted in CSV file");
if (lWatchlistedCsvShows.Any())
{
UIUtils.UpdateStatus("Requesting existing watchlisted shows from trakt...");
var lWatchlistTraktShows = TraktAPI.GetWatchlistShows();
if (lWatchlistTraktShows != null)
{
UIUtils.UpdateStatus($"Found {lWatchlistTraktShows.Count()} watchlist shows on trakt");
UIUtils.UpdateStatus("Filtering out watchlist shows that are already in watchlist on trakt.tv");
lWatchlistedCsvShows.RemoveAll(w => lWatchlistTraktShows.FirstOrDefault(t => t.Show.Ids.ImdbId == w.Ids.ImdbId ||
(t.Show.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Show.Year == w.Year)) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lWatchlistedCsvShows.Count > 0)
{
UIUtils.UpdateStatus("Requesting watched shows from trakt...");
// get watched movies from trakt so we don't import shows into watchlist that are already watched
lWatchedTraktShows = TraktAPI.GetWatchedShows();
if (lWatchedTraktShows != null)
{
UIUtils.UpdateStatus($"Found {lWatchedTraktShows.Count()} watched shows on trakt");
UIUtils.UpdateStatus("Filtering out watchlist shows containing watched episodes on trakt.tv.");
// remove shows from sync list which are watched already
lWatchlistedCsvShows.RemoveAll(w => lWatchedTraktShows.FirstOrDefault(t => (t.Show.Ids.ImdbId == w.Ids.ImdbId) ||
(t.Show.Title.ToLowerInvariant() == w.Title.ToLowerInvariant() && t.Show.Year == w.Year)) != null);
}
}
// add shows to watchlist
UIUtils.UpdateStatus($"Importing {lWatchlistedCsvShows.Count()} IMDb watchlist shows to trakt.tv...");
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lWatchlistedCsvShows.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb watchlist shows...");
var lWatchlistShowsToSync = new TraktShowSync
{
Shows = lWatchlistedCsvShows.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddShowsToWatchlist(lWatchlistShowsToSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send watchlist for IMDb tv shows", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync watchlist for {lResponse.NotFound.Shows.Count} shows as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
if (mImportCancelled)
{
return;
}
#endregion
#region Import Watchlist Episodes
var lImdbWatchlistedEpisodes = new List <IMDbEpisode>();
var lImdbWatchlistedCsvEpisodes = lWatchlistCsvItems.Where(r => r.Type.ItemType() == IMDbType.Episode).ToList();
FileLog.Info($"Found {lImdbWatchlistedCsvEpisodes.Count} tv episodes watchlisted in CSV file");
if (lImdbWatchlistedCsvEpisodes.Any())
{
UIUtils.UpdateStatus($"Found {lImdbWatchlistedCsvEpisodes.Count()} IMDb watchlisted episodes");
lImdbWatchlistedEpisodes.AddRange(lImdbWatchlistedCsvEpisodes.Select(Helper.GetIMDbEpisodeFromTrakt).Where(imdbEpisode => imdbEpisode != null));
if (mImportCancelled)
{
return;
}
// filter out existing watchlist episodes
UIUtils.UpdateStatus("Requesting existing watchlist episodes from trakt...");
var lWatchlistedTraktEpisodes = TraktAPI.GetWatchlistEpisodes();
if (lWatchlistedTraktEpisodes != null)
{
UIUtils.UpdateStatus($"Found {lWatchlistedTraktEpisodes.Count()} watchlist episodes on trakt");
UIUtils.UpdateStatus("Filtering out watchlist episodes that are already in watchlist on trakt.tv");
lImdbWatchlistedEpisodes.RemoveAll(e => lWatchlistedTraktEpisodes.FirstOrDefault(w => w.Episode.Ids.Trakt == e.TraktId) != null);
}
if (AppSettings.IgnoreWatchedForWatchlist && lImdbWatchlistedEpisodes.Count > 0)
{
// we already might have it from the shows sync
if (lWatchedTraktShows == null)
{
UIUtils.UpdateStatus("Requesting watched episodes from trakt...");
// get watched episodes from trakt so we don't import episodes into watchlist that are already watched
lWatchedTraktShows = TraktAPI.GetWatchedShows();
}
if (lWatchedTraktShows != null)
{
UIUtils.UpdateStatus("Filtering out watchlist episodes containing watched episodes on trakt.tv");
// this wont work atm due to show IMDb ID not being set in the IMDbEpisode object
lImdbWatchlistedEpisodes.RemoveAll(e => lWatchedTraktShows.Where(s => s.Show.Ids.ImdbId == e.ShowImdbId)
.Any(s => s.Seasons.Exists(se => se.Number == e.SeasonNumber &&
se.Episodes.Exists(ep => ep.Number == e.EpisodeNumber))));
}
}
UIUtils.UpdateStatus($"Importing {lImdbWatchlistedEpisodes.Count()} episodes in watchlist to trakt.tv");
if (lImdbWatchlistedEpisodes.Count > 0)
{
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lImdbWatchlistedEpisodes.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb watchlisted episodes...");
var lResponse = TraktAPI.AddEpisodesToWatchlist(Helper.GetTraktEpisodeData(lImdbWatchlistedEpisodes.Skip(i * lPageSize).Take(lPageSize)));
if (lResponse == null)
{
UIUtils.UpdateStatus("Error importing IMDb episode watchlist to trakt.tv", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Episodes.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync watchlist for {0} IMDb episodes as they're not found on trakt.tv!", lResponse.NotFound.Episodes.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
}
}
#endregion
#region Import Custom Lists
if (lCustomLists.Count > 0)
{
UIUtils.UpdateStatus("Requesting custom lists from trakt...");
var lTraktCustomLists = TraktAPI.GetCustomLists();
if (lTraktCustomLists == null)
{
UIUtils.UpdateStatus("Error requesting custom lists from trakt.tv", true);
Thread.Sleep(2000);
return;
}
UIUtils.UpdateStatus($"Found {lTraktCustomLists.Count()} custom lists on trakt.tv");
foreach (var list in lCustomLists)
{
bool lListCreated = false;
string lListName = Path.GetFileNameWithoutExtension(list.Key);
// create the list if we don't have it
TraktListDetail lTraktCustomList = lTraktCustomLists.FirstOrDefault(l => l.Name == lListName);
if (lTraktCustomList == null)
{
UIUtils.UpdateStatus($"Creating new custom list '{lListName}' on trakt.tv");
var lTraktList = new TraktList
{
Name = lListName,
DisplayNumbers = true,
};
lTraktCustomList = TraktAPI.CreateCustomList(lTraktList);
if (lTraktCustomList == null)
{
UIUtils.UpdateStatus("Error creating custom list on trakt.tv, skipping list creation", true);
Thread.Sleep(2000);
continue;
}
lListCreated = true;
}
// get the CSV list items parsed
var lIMDbCsvListItems = list.Value;
var lImdbCsvListMovies = lIMDbCsvListItems.Where(l => l.Type.ItemType() == IMDbType.Movie).Select(m => m.ToTraktMovie()).ToList();
var lImdbCsvListShows = lIMDbCsvListItems.Where(l => l.Type.ItemType() == IMDbType.Show).Select(m => m.ToTraktShow()).ToList();
FileLog.Info($"Found {lIMDbCsvListItems.Count} movies and {lImdbCsvListShows.Count} shows in IMDb {lListName} list", lListName);
// if the list already exists, get current items for list
if (!lListCreated)
{
lTraktCustomList = lTraktCustomLists.FirstOrDefault(l => l.Name == lListName);
UIUtils.UpdateStatus($"Requesting existing custom list '{lListName}' items from trakt...");
var lTraktListItems = TraktAPI.GetCustomListItems(lTraktCustomList.Ids.Trakt.ToString());
if (lTraktListItems == null)
{
UIUtils.UpdateStatus("Error requesting custom list items from trakt.tv, skipping list creation", true);
Thread.Sleep(2000);
continue;
}
// filter out existing items from CSV so we don't send again
FileLog.Info($"Filtering out existing items from IMDb list '{lListName}' so we don't send again to trakt.tv");
lImdbCsvListMovies.RemoveAll(i => lTraktListItems.FirstOrDefault(l => l.Movie.Ids.ImdbId == i.Ids.ImdbId) != null);
lImdbCsvListShows.RemoveAll(i => lTraktListItems.FirstOrDefault(l => l.Show.Ids.ImdbId == i.Ids.ImdbId) != null);
}
#region Movies
UIUtils.UpdateStatus($"Importing {lImdbCsvListMovies.Count} new movies into {lListName} custom list...");
int lPageSize = AppSettings.BatchSize;
int lPages = (int)Math.Ceiling((double)lImdbCsvListMovies.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus($"Importing page {i + 1}/{lPages} IMDb custom list movies...");
// create list sync object to hold list items
var lTraktMovieSync = new TraktSyncAll
{
Movies = lImdbCsvListMovies.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddItemsToList(lTraktCustomList.Ids.Trakt.ToString(), lTraktMovieSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send custom list items for IMDb movies", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Movies.Count > 0)
{
UIUtils.UpdateStatus($"Unable to sync custom list items for {lResponse.NotFound.Movies.Count} movies as they're not found on trakt.tv!");
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
#endregion
#region Shows
UIUtils.UpdateStatus("Importing {0} new shows into {1} custom list...", lImdbCsvListShows.Count(), lListName);
lPageSize = AppSettings.BatchSize;
lPages = (int)Math.Ceiling((double)lImdbCsvListShows.Count / lPageSize);
for (int i = 0; i < lPages; i++)
{
UIUtils.UpdateStatus("Importing page {0}/{1} IMDb custom list shows...", i + 1, lPages);
// create list sync object to hold list items
var lTraktShowSync = new TraktSyncAll
{
Shows = lImdbCsvListShows.Skip(i * lPageSize).Take(lPageSize).ToList()
};
var lResponse = TraktAPI.AddItemsToList(lTraktCustomList.Ids.Trakt.ToString(), lTraktShowSync);
if (lResponse == null)
{
UIUtils.UpdateStatus("Failed to send custom list items for IMDb shows", true);
Thread.Sleep(2000);
}
else if (lResponse.NotFound.Shows.Count > 0)
{
UIUtils.UpdateStatus("Unable to sync custom list items for {0} shows as they're not found on trakt.tv!", lResponse.NotFound.Shows.Count);
Thread.Sleep(1000);
}
if (mImportCancelled)
{
return;
}
}
#endregion
}
}
#endregion
}
