filebot/website/scripts/artwork.tmdb.groovy

83 lines
2.3 KiB
Groovy
Raw Normal View History

// filebot -script "http://filebot.sf.net/scripts/artwork.tmdb.groovy" -trust-script /path/to/media/
// EXPERIMENTAL // HERE THERE BE DRAGONS
2012-01-04 04:09:17 +00:00
if (net.sourceforge.filebot.Settings.applicationRevisionNumber < 838) throw new Exception("Application revision too old")
/*
2011-12-28 14:15:39 +00:00
* Fetch movie artwork. The movie is determined using the parent folders name.
*/
def fetchArtwork(outputFile, movieInfo, artworkType, artworkSize) {
// select and fetch artwork
def artwork = movieInfo.images.find { it.type == artworkType && it.size == artworkSize }
if (artwork == null) {
println "Artwork not found: $outputFile"
return null
}
println "Fetching $outputFile => $artwork"
return artwork.url.saveAs(outputFile)
}
def fetchNfo(outputFile, movieInfo) {
movieInfo.applyXmlTemplate('''<movie>
<title>$name</title>
<year>$released.year</year>
<rating>$rating</rating>
<votes>$votes</votes>
<plot>$overview</plot>
<runtime>$runtime</runtime>
<mpaa>$certification</mpaa>
<genre>${!genres.empty ? genres[0] : ''}</genre>
<id>tt${imdbId.pad(7)}</id>
</movie>
''').saveAs(outputFile)
}
def fetchMovieArtworkAndNfo(movieDir, movie) {
println "Fetch nfo and artwork for $movie"
def movieInfo = TheMovieDB.getMovieInfo(movie, Locale.ENGLISH)
2011-12-29 00:51:00 +00:00
println movieInfo
movieInfo.images.each {
println "Available artwork: $it.url => $it"
}
// fetch nfo
fetchNfo(movieDir['movie.nfo'], movieInfo)
// fetch series banner, fanart, posters, etc
fetchArtwork(movieDir['folder.jpg'], movieInfo, 'poster', 'original')
fetchArtwork(movieDir['backdrop.jpg'], movieInfo, 'backdrop', 'original')
}
2011-12-29 04:05:10 +00:00
args.eachMediaFolder { dir ->
def videos = dir.listFiles{ it.isVideo() }
def query = _args.query ?: dir.name
2012-01-04 04:09:17 +00:00
def options = TheMovieDB.searchMovie(query, _args.locale)
if (options.isEmpty()) {
println "Movie not found: $query"
return null
}
2012-01-03 13:44:31 +00:00
// sort by relevance
options = options.sortBySimilarity(query, { it.name })
// auto-select series
2012-01-03 13:44:31 +00:00
def movie = options[0]
2012-01-04 04:09:17 +00:00
// maybe require user input
if (options.size() != 1 && !_args.nonStrict && !java.awt.GraphicsEnvironment.headless) {
movie = javax.swing.JOptionPane.showInputDialog(null, "Please select Movie:", dir.path, 3, null, options.toArray(), movie);
2012-01-03 13:44:31 +00:00
if (movie == null) return null
}
2011-12-29 00:51:00 +00:00
println "$dir => $movie"
fetchMovieArtworkAndNfo(dir, movie)
}