com.moviejukebox.plugin.TheMovieDbPlugin Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of yamj Show documentation
Show all versions of yamj Show documentation
Static analysis of MovieJukebox project
The newest version!
/*
* Copyright (c) 2004-2013 YAMJ Members
* http://code.google.com/p/moviejukebox/people/list
*
* This file is part of the Yet Another Movie Jukebox (YAMJ).
*
* The YAMJ is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* any later version.
*
* YAMJ is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with the YAMJ. If not, see movieList;
List movieReleaseInfo = new ArrayList();
List moviePeople = new ArrayList();
MovieDb moviedb = null;
boolean retval = false;
ThreadExecutor.enterIO(webhost);
try {
// First look to see if we have a TMDb ID as this will make looking the film up easier
if (StringTools.isValidString(tmdbID)) {
// Search based on TMdb ID
logger.debug(LOG_MESSAGE + "Using TMDb ID (" + tmdbID + ") for " + movie.getBaseFilename());
try {
moviedb = TMDb.getMovieInfo(Integer.parseInt(tmdbID), languageCode);
} catch (MovieDbException ex) {
logger.debug(LOG_MESSAGE + "Failed to get movie info using TMDB ID: " + tmdbID + " - " + ex.getMessage());
moviedb = null;
}
}
if (moviedb == null && StringTools.isValidString(imdbID)) {
// Search based on IMDb ID
logger.debug(LOG_MESSAGE + "Using IMDb ID (" + imdbID + ") for " + movie.getBaseFilename());
try {
moviedb = TMDb.getMovieInfoImdb(imdbID, languageCode);
tmdbID = String.valueOf(moviedb.getId());
if (StringTools.isNotValidString(tmdbID)) {
logger.debug(LOG_MESSAGE + "No TMDb ID found for movie!");
}
} catch (MovieDbException ex) {
logger.debug(LOG_MESSAGE + "Failed to get movie info using IMDB ID: " + imdbID + " - " + ex.getMessage());
moviedb = null;
}
}
if (moviedb == null) {
try {
// Search using movie name
int movieYear = 0;
if (StringTools.isValidString(movie.getYear()) && StringUtils.isNumeric(movie.getYear())) {
movieYear = Integer.parseInt(movie.getYear());
}
TmdbResultsList result = TMDb.searchMovie(movie.getTitle(), movieYear, languageCode, INCLUDE_ADULT, 0);
movieList = result.getResults();
// Iterate over the list until we find a match
for (MovieDb m : movieList) {
logger.debug(LOG_MESSAGE + "Checking " + m.getTitle() + " (" + m.getReleaseDate().substring(0, 4) + ")");
if (TheMovieDbApi.compareMovies(m, movie.getTitle(), Integer.toString(movieYear), SEARCH_MATCH)) {
moviedb = m;
break;
}
// See if the original title is different and then compare it too
if (!movie.getTitle().equals(movie.getOriginalTitle())
&& TheMovieDbApi.compareMovies(m, movie.getOriginalTitle(), Integer.toString(movieYear))) {
moviedb = m;
break;
}
}
} catch (MovieDbException ex) {
logger.debug(LOG_MESSAGE + "Failed to get movie info for " + movie.getTitle() + " - " + ex.getMessage());
moviedb = null;
}
}
if (moviedb == null) {
logger.debug(LOG_MESSAGE + "Movie " + movie.getBaseName() + " not found!");
logger.debug(LOG_MESSAGE + "Try using a NFO file to specify the movie");
} else {
try {
// Get the full information on the film
moviedb = TMDb.getMovieInfo(moviedb.getId(), languageCode);
} catch (MovieDbException ex) {
logger.debug(LOG_MESSAGE + "Failed to download remaining information for " + movie.getBaseName());
}
logger.debug(LOG_MESSAGE + "Found id (" + moviedb.getId() + ") for " + moviedb.getTitle());
}
try {
// Get the release information
movieReleaseInfo = TMDb.getMovieReleaseInfo(moviedb.getId(), countryCode).getResults();
} catch (MovieDbException ex) {
logger.debug(LOG_MESSAGE + "Failed to get release information");
}
try {
// Get the cast information
moviePeople = TMDb.getMovieCasts(moviedb.getId()).getResults();
} catch (MovieDbException ex) {
logger.debug(LOG_MESSAGE + "Failed to get cast information");
}
} finally {
// the rest is not web search anymore
ThreadExecutor.leaveIO();
}
if (moviedb != null) {
if (moviedb.getId() > 0) {
movie.setMovieType(Movie.TYPE_MOVIE);
}
if (StringTools.isValidString(moviedb.getTitle())) {
copyMovieInfo(moviedb, movie);
retval = true;
}
// Set the release information
if (movieReleaseInfo.size() > 0 && OverrideTools.checkOverwriteCertification(movie, TMDB_PLUGIN_ID)) {
logger.debug(LOG_MESSAGE + "Found release information: " + movieReleaseInfo.get(0).toString());
movie.setCertification(movieReleaseInfo.get(0).getCertification(), TMDB_PLUGIN_ID);
}
// Add the cast information
// TODO: Add the people to the cast/crew
if (moviePeople.size() > 0) {
List newActors = new ArrayList();
List newDirectors = new ArrayList();
List newWriters = new ArrayList();
logger.debug(LOG_MESSAGE + "Adding " + moviePeople.size() + " people to the cast list");
for (Person person : moviePeople) {
if (person.getPersonType() == PersonType.CAST) {
logger.trace(LOG_MESSAGE + "Adding cast member " + person.toString());
newActors.add(person.getName());
} else if (person.getPersonType() == PersonType.CREW) {
logger.trace(LOG_MESSAGE + "Adding crew member " + person.toString());
if ("Director".equalsIgnoreCase(person.getJob())) {
logger.trace(LOG_MESSAGE + person.getName() + " is a Director");
newDirectors.add(person.getName());
} else if ("Author".equalsIgnoreCase(person.getJob())) {
logger.trace(LOG_MESSAGE + person.getName() + " is a Writer");
newWriters.add(person.getName());
continue;
} else {
logger.trace(LOG_MESSAGE + "Unknown job " + person.getJob() + " for " + person.toString());
}
} else {
logger.trace(LOG_MESSAGE + "Unknown person type " + person.getPersonType() + " for " + person.toString());
}
}
if (OverrideTools.checkOverwriteDirectors(movie, TMDB_PLUGIN_ID)) {
movie.setDirectors(newDirectors, TMDB_PLUGIN_ID);
}
if (OverrideTools.checkOverwritePeopleDirectors(movie, TMDB_PLUGIN_ID)) {
movie.setPeopleDirectors(newDirectors, TMDB_PLUGIN_ID);
}
if (OverrideTools.checkOverwriteWriters(movie, TMDB_PLUGIN_ID)) {
movie.setWriters(newWriters, TMDB_PLUGIN_ID);
}
if (OverrideTools.checkOverwritePeopleWriters(movie, TMDB_PLUGIN_ID)) {
movie.setPeopleWriters(newWriters, TMDB_PLUGIN_ID);
}
if (OverrideTools.checkOverwriteActors(movie, TMDB_PLUGIN_ID)) {
movie.setCast(newActors, TMDB_PLUGIN_ID);
}
if (OverrideTools.checkOverwritePeopleActors(movie, TMDB_PLUGIN_ID)) {
movie.setPeopleCast(newActors, TMDB_PLUGIN_ID);
}
} else {
logger.debug(LOG_MESSAGE + "No cast or crew members found");
}
// Update TheMovieDb Id if needed
if (StringTools.isNotValidString(movie.getId(TMDB_PLUGIN_ID))) {
movie.setId(TMDB_PLUGIN_ID, moviedb.getId());
}
// Update IMDb Id if needed
if (StringTools.isNotValidString(movie.getId(IMDB_PLUGIN_ID))) {
movie.setId(IMDB_PLUGIN_ID, moviedb.getImdbID());
}
}
if (downloadFanart && StringTools.isNotValidString(movie.getFanartURL())) {
movie.setFanartURL(getFanartURL(movie));
if (StringTools.isValidString(movie.getFanartURL())) {
movie.setFanartFilename(movie.getBaseName() + fanartToken + "." + fanartExtension);
}
}
return retval;
}
/**
* Copy the movie info from the MovieDB bean to the YAMJ movie bean
*
* @param moviedb The MovieDB source
* @param movie The YAMJ target
* @return The altered movie bean
*/
private void copyMovieInfo(MovieDb moviedb, Movie movie) {
// TMDb ID
movie.setId(TMDB_PLUGIN_ID, moviedb.getId());
// IMDb ID
movie.setId(IMDB_PLUGIN_ID, moviedb.getImdbID());
// title
if (OverrideTools.checkOverwriteTitle(movie, TMDB_PLUGIN_ID)) {
movie.setTitle(moviedb.getTitle(), TMDB_PLUGIN_ID);
}
// original title
if (OverrideTools.checkOverwriteOriginalTitle(movie, TMDB_PLUGIN_ID)) {
movie.setOriginalTitle(moviedb.getOriginalTitle(), TMDB_PLUGIN_ID);
}
// plot
if (OverrideTools.checkOverwritePlot(movie, TMDB_PLUGIN_ID)) {
movie.setPlot(moviedb.getOverview(), TMDB_PLUGIN_ID);
}
// outline
if (OverrideTools.checkOverwriteOutline(movie, TMDB_PLUGIN_ID)) {
movie.setOutline(moviedb.getOverview(), TMDB_PLUGIN_ID);
}
// rating
if (overwriteCheck(String.valueOf(moviedb.getVoteAverage()), String.valueOf(movie.getRating()))) {
try {
float rating = moviedb.getVoteAverage() * 10; // Convert rating to integer
movie.addRating(TMDB_PLUGIN_ID, (int) rating);
} catch (Exception error) {
logger.debug(LOG_MESSAGE + "Error converting rating for " + movie.getBaseName());
}
}
// release date
if (OverrideTools.checkOverwriteReleaseDate(movie, TMDB_PLUGIN_ID)) {
movie.setReleaseDate(moviedb.getReleaseDate(), TMDB_PLUGIN_ID);
}
// year
if (OverrideTools.checkOverwriteYear(movie, TMDB_PLUGIN_ID)) {
String year = moviedb.getReleaseDate();
// Check if this is the default year and skip it
if (!"1900-01-01".equals(year)) {
year = (new DateTime(year)).toString("yyyy");
movie.setYear(year, TMDB_PLUGIN_ID);
}
}
// runtime
if (OverrideTools.checkOverwriteRuntime(movie, TMDB_PLUGIN_ID)) {
movie.setRuntime(String.valueOf(moviedb.getRuntime()), TMDB_PLUGIN_ID);
}
// tagline
if (OverrideTools.checkOverwriteTagline(movie, TMDB_PLUGIN_ID)) {
movie.setTagline(moviedb.getTagline(), TMDB_PLUGIN_ID);
}
// Country
if (OverrideTools.checkOverwriteCountry(movie, TMDB_PLUGIN_ID)) {
List countries = moviedb.getProductionCountries();
if (!countries.isEmpty()) {
movie.setCountry(countries.get(0).getName(), TMDB_PLUGIN_ID);
}
}
// Company
if (OverrideTools.checkOverwriteCompany(movie, TMDB_PLUGIN_ID)) {
List studios = moviedb.getProductionCompanies();
if (!studios.isEmpty()) {
String studio = studios.get(0).getName();
movie.setCompany(studio, TMDB_PLUGIN_ID);
}
}
// Language
if (moviedb.getSpokenLanguages().size() > 0 && OverrideTools.checkOverwriteLanguage(movie, TMDB_PLUGIN_ID)) {
StringBuilder movieLanguage = new StringBuilder();
String isoCode = moviedb.getSpokenLanguages().get(0).getIsoCode();
if (StringTools.isValidString(isoCode)) {
movieLanguage.append(MovieFilenameScanner.determineLanguage(isoCode));
}
if (moviedb.getSpokenLanguages().size() > 1) {
for (Language lang : moviedb.getSpokenLanguages()) {
if (movieLanguage.length() > 0) {
movieLanguage.append(LANGUAGE_DELIMITER);
}
movieLanguage.append(MovieFilenameScanner.determineLanguage(lang.getIsoCode()));
}
}
movie.setLanguage(movieLanguage.toString(), TMDB_PLUGIN_ID);
}
// Genres
if (OverrideTools.checkOverwriteGenres(movie, TMDB_PLUGIN_ID)) {
List newGenres = new ArrayList();
for (Genre genre : moviedb.getGenres()) {
newGenres.add(genre.getName());
}
movie.setGenres(newGenres, TMDB_PLUGIN_ID);
}
}
/**
* Checks to see if the source string is null or "UNKNOWN" and that target string ISN'T null or "UNKNOWN"
*
* @param sourceString The source string to check
* @param targetString The destination string to check
* @return True if valid to overwrite
*/
private boolean overwriteCheck(String sourceString, String targetString) {
// false if the source is null or UNKNOWN
if (StringTools.isValidString(sourceString)) {
// sourceString is valid, check target string IS null OR UNKNOWN
if (StringTools.isNotValidString(targetString) || targetString.equals("-1")) {
return true;
} else {
return false;
}
} else {
return false;
}
}
@Override
public boolean scanNFO(String nfo, Movie movie) {
int beginIndex;
boolean result = Boolean.FALSE;
if (StringTools.isValidString(movie.getId(TMDB_PLUGIN_ID))) {
logger.debug(LOG_MESSAGE + "TheMovieDb ID exists for " + movie.getBaseName() + " = " + movie.getId(TMDB_PLUGIN_ID));
result = Boolean.TRUE;
} else {
logger.debug(LOG_MESSAGE + "Scanning NFO for TheMovieDb ID");
beginIndex = nfo.indexOf("/movie/");
if (beginIndex != -1) {
StringTokenizer st = new StringTokenizer(nfo.substring(beginIndex + 7), "/ \n,:!&é\"'(--è_çà)=$");
movie.setId(TMDB_PLUGIN_ID, st.nextToken());
logger.debug(LOG_MESSAGE + "TheMovieDb ID found in NFO = " + movie.getId(TMDB_PLUGIN_ID));
result = Boolean.TRUE;
} else {
logger.debug(LOG_MESSAGE + "No TheMovieDb ID found in NFO!");
}
}
// We might as well look for the IMDb ID as well
if (StringTools.isValidString(movie.getId(IMDB_PLUGIN_ID))) {
logger.debug(LOG_MESSAGE + "IMDB ID exists for " + movie.getBaseName() + " = " + movie.getId(IMDB_PLUGIN_ID));
result = Boolean.TRUE;
} else {
beginIndex = nfo.indexOf("/tt");
if (beginIndex != -1) {
StringTokenizer st = new StringTokenizer(nfo.substring(beginIndex + 1), "/ \n,:!&é\"'(--è_çà)=$");
movie.setId(ImdbPlugin.IMDB_PLUGIN_ID, st.nextToken());
logger.debug(LOG_MESSAGE + "IMDB ID found in nfo = " + movie.getId(ImdbPlugin.IMDB_PLUGIN_ID));
} else {
beginIndex = nfo.indexOf("/Title?");
if (beginIndex != -1 && beginIndex + 7 < nfo.length()) {
StringTokenizer st = new StringTokenizer(nfo.substring(beginIndex + 7), "/ \n,:!&é\"'(--è_çà)=$");
movie.setId(ImdbPlugin.IMDB_PLUGIN_ID, "tt" + st.nextToken());
logger.debug(LOG_MESSAGE + "IMDB ID found in NFO = " + movie.getId(ImdbPlugin.IMDB_PLUGIN_ID));
}
}
}
return result;
}
@Override
public void scanTVShowTitles(Movie movie) {
// TheMovieDB.org does not have any TV Shows, so just return
}
/**
* Locate the FanartURL for the movie. This should probably be skipped as this uses TheMovieDb.org anyway
*
* @param movie Movie bean for the movie to locate
* @return The URL of the fanart
*/
protected String getFanartURL(Movie movie) {
return FanartScanner.getFanartURL(movie);
}
public boolean scan(Person person) {
return false;
}
@Override
public boolean scan(com.moviejukebox.model.Person person) {
// TODO Auto-generated method stub
return false;
}
}