mirror of
https://github.com/mitb-archive/filebot
synced 2024-12-25 09:18:51 -05:00
73 lines
1.9 KiB
Groovy
73 lines
1.9 KiB
Groovy
// filebot -script "http://filebot.sf.net/scripts/artwork.tmdb.groovy" -trust-script /path/to/media/
|
|
|
|
// EXPERIMENTAL // HERE THERE BE DRAGONS
|
|
if (net.sourceforge.filebot.Settings.applicationRevisionNumber < 802) throw new Exception("Application revision too old")
|
|
|
|
|
|
/*
|
|
* Fetch series and season banners for all tv shows
|
|
*/
|
|
|
|
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.size() > 0 ? genres.get(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)
|
|
|
|
// 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')
|
|
}
|
|
|
|
|
|
def jobs = args.getFolders().findResults { dir ->
|
|
def videos = dir.listFiles{ it.isVideo() }
|
|
if (videos.isEmpty()) {
|
|
return null
|
|
}
|
|
|
|
def query = _args.query ?: dir.name
|
|
def options = TheMovieDB.searchMovie(query, Locale.ENGLISH)
|
|
if (options.isEmpty()) {
|
|
println "Movie not found: $query"
|
|
return null
|
|
}
|
|
|
|
// auto-select series
|
|
def movie = options.sortBySimilarity(query, { it.name })[0]
|
|
|
|
return { fetchMovieArtworkAndNfo(dir, movie) }
|
|
}
|
|
|
|
parallel(jobs, 10)
|