mirror of
https://github.com/mitb-archive/filebot
synced 2024-12-23 16:28:51 -05:00
* added fetch *.artwork function to the utorrent-postprocess script
This commit is contained in:
parent
a7104877bb
commit
de73d1ac31
@ -19,7 +19,7 @@ def fetchBanner(outputFile, series, bannerType, bannerType2 = null, season = nul
|
||||
|
||||
def fetchNfo(outputFile, series) {
|
||||
def info = TheTVDB.getSeriesInfo(series, _args.locale)
|
||||
println info
|
||||
println info
|
||||
info.applyXmlTemplate('''<tvshow xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
|
||||
<title>$name</title>
|
||||
<year>$firstAired.year</year>
|
||||
|
@ -1,8 +1,14 @@
|
||||
// filebot -script "http://filebot.sf.net/scripts/utorrent-postprocess.groovy" "%D\%N" --output "X:/media" --action copy --conflict override -non-strict -trust-script -Xxbmc=localhost
|
||||
println "Input: $args"
|
||||
println "Parameters: $_args.parameters"
|
||||
// filebot -script "http://filebot.sf.net/scripts/utorrent-postprocess.groovy" --output "X:/media" --action copy --conflict override -non-strict -trust-script -Xxbmc=localhost "-Xut_dir=%D" "-Xut_file=%F" "-Xut_label=%L" "-Xut_state=%S" "-Xut_kind=%K"
|
||||
def input = []
|
||||
|
||||
def input = args.getFiles()
|
||||
// print input parameters
|
||||
_args.parameters.each{ k, v -> println "Parameter: $k = $v" }
|
||||
|
||||
if (ut_kind == "multi") {
|
||||
input += new File(ut_dir).getFiles() // multi-file torrent
|
||||
} else {
|
||||
input += new File(ut_dir, ut_file) // single-file torrent
|
||||
}
|
||||
|
||||
// extract archives if necessary
|
||||
input += extract(file:input)
|
||||
@ -10,9 +16,11 @@ input += extract(file:input)
|
||||
// process only media files
|
||||
input = input.findAll{ it.isVideo() || it.isSubtitle() }
|
||||
|
||||
// print input fileset
|
||||
input.each{ println "Input: $it" }
|
||||
|
||||
// group episodes/movies and rename according to XBMC standards
|
||||
def groups = input.groupBy {
|
||||
def groups = input.groupBy{
|
||||
def tvs = detectSeriesName(it)
|
||||
def mov = detectMovie(it, false)
|
||||
println "$it.name [series: $tvs, movie: $mov]"
|
||||
@ -33,12 +41,37 @@ def groups = input.groupBy {
|
||||
groups.each{ group, files ->
|
||||
// EPISODE MODE
|
||||
if (group.tvs && !group.mov) {
|
||||
return rename(file:files, format:'TV Shows/{n}/{episode.special ? "Special" : "Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB')
|
||||
def output = rename(file:files, format:'TV Shows/{n}{episode.special ? "/Special" : "/Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB')
|
||||
|
||||
output*.dir.unique().each{ dir ->
|
||||
println "Fetching artwork for $dir from TheTVDB"
|
||||
def query = group.tvs
|
||||
def sxe = output.findResult{ parseEpisodeNumber(it) }
|
||||
def options = TheTVDB.search(query, _args.locale)
|
||||
if (options.isEmpty()) {
|
||||
println "TV Series not found: $query"
|
||||
return
|
||||
}
|
||||
options = options.sortBySimilarity(query, { it.name })
|
||||
def series = options[0]
|
||||
def seriesDir = [dir.dir, dir].sortBySimilarity(series.name, { it.name })[0]
|
||||
def season = sxe && sxe.season > 0 ? sxe.season : 1
|
||||
fetchSeriesBannersAndNfo(seriesDir, dir, series, season)
|
||||
}
|
||||
}
|
||||
|
||||
// MOVIE MODE
|
||||
if (group.mov && !group.tvs) {
|
||||
return rename(file:files, format:'Movies/{n} ({y}){" CD$pi"}{".$lang"}', db:'TheMovieDB')
|
||||
def output = rename(file:files, format:'Movies/{n} ({y})/{n} ({y}){" CD$pi"}', db:'TheMovieDB')
|
||||
|
||||
output*.dir.unique().each{ dir ->
|
||||
println "Fetching artwork for $dir from TheMovieDB"
|
||||
try {
|
||||
fetchMovieArtworkAndNfo(dir, group.mov)
|
||||
} catch(e) {
|
||||
println "${e.class.simpleName}: ${e.message}"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -46,10 +79,123 @@ groups.each{ group, files ->
|
||||
|
||||
// make XBMC scan for new content
|
||||
try {
|
||||
telnet(xbmc, 9090) { writer, reader ->
|
||||
def msg = '{"id":1,"method":"VideoLibrary.Scan","params":[],"jsonrpc":"2.0"}'
|
||||
writer.println(msg)
|
||||
xbmc.split(/[\s,|]+/).each{
|
||||
println "Notify XBMC: $it"
|
||||
telnet(it, 9090) { writer, reader ->
|
||||
def msg = '{"id":1,"method":"VideoLibrary.Scan","params":[],"jsonrpc":"2.0"}'
|
||||
writer.println(msg)
|
||||
}
|
||||
}
|
||||
} catch(e) {
|
||||
println "${e.class.simpleName}: ${e.message}"
|
||||
}
|
||||
// END OF SCRIPT
|
||||
|
||||
|
||||
|
||||
|
||||
// FUNCTIONS for TMDB and TVDB artwork/nfo
|
||||
def fetchBanner(outputFile, series, bannerType, bannerType2 = null, season = null) {
|
||||
// select and fetch banner
|
||||
def banner = [_args.locale.language, null].findResult { TheTVDB.getBanner(series, [BannerType:bannerType, BannerType2:bannerType2, Season:season, Language:it]) }
|
||||
if (banner == null) {
|
||||
println "Banner not found: $outputFile / $bannerType:$bannerType2"
|
||||
return null
|
||||
}
|
||||
println "Fetching $outputFile => $banner"
|
||||
return banner.url.saveAs(outputFile)
|
||||
}
|
||||
def fetchSeriesNfo(outputFile, series) {
|
||||
def info = TheTVDB.getSeriesInfo(series, _args.locale)
|
||||
info.applyXmlTemplate('''<tvshow xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
|
||||
<title>$name</title>
|
||||
<year>$firstAired.year</year>
|
||||
<top250></top250>
|
||||
<seasons>-1</seasons>
|
||||
<episode></episode>
|
||||
<episodeguideurl></episodeguideurl>
|
||||
<displayseason>-1</displayseason>
|
||||
<displayepisode>-1</displayepisode>
|
||||
<rating>$rating</rating>
|
||||
<votes>$ratingCount</votes>
|
||||
<outline></outline>
|
||||
<plot>$overview</plot>
|
||||
<tagline></tagline>
|
||||
<runtime>$runtime</runtime>
|
||||
<mpaa>$contentRating</mpaa>
|
||||
<playcount></playcount>
|
||||
<lastplayed></lastplayed>
|
||||
<id>$id</id>
|
||||
<episodeguide><url cache="${id}.xml">http://www.thetvdb.com/api/1D62F2F90030C444/series/${id}/all/''' + _args.locale.language + '''.zip</url></episodeguide>
|
||||
<genre>${!genres.empty ? genres[0] : ''}</genre>
|
||||
<set></set>
|
||||
<credits></credits>
|
||||
<director></director>
|
||||
<thumb>$bannerUrl</thumb>
|
||||
<premiered>$firstAired</premiered>
|
||||
<status>$status</status>
|
||||
<studio>$network</studio>
|
||||
<trailer></trailer>
|
||||
<gsp:scriptlet> actors.each { </gsp:scriptlet>
|
||||
<actor>
|
||||
<name>$it</name>
|
||||
<role></role>
|
||||
</actor>
|
||||
<gsp:scriptlet> } </gsp:scriptlet>
|
||||
<artist></artist>
|
||||
</tvshow>
|
||||
''')
|
||||
.replaceAll(/\t|\r|\n/, '') // xbmc can't handle leading/trailing whitespace properly
|
||||
.saveAs(outputFile)
|
||||
}
|
||||
def fetchSeriesBannersAndNfo(seriesDir, seasonDir, series, season) {
|
||||
println "Fetch nfo and banners for $series / Season $season"
|
||||
// fetch nfo
|
||||
fetchSeriesNfo(seriesDir['tvshow.nfo'], series)
|
||||
// fetch series banner, fanart, posters, etc
|
||||
["680x1000", null].findResult{ fetchBanner(seriesDir['folder.jpg'], series, "poster", it) }
|
||||
["graphical", null].findResult{ fetchBanner(seriesDir['banner.jpg'], series, "series", it) }
|
||||
// fetch highest resolution fanart
|
||||
["1920x1080", "1280x720", null].findResult{ fetchBanner(seriesDir["fanart.jpg"], series, "fanart", it) }
|
||||
// fetch season banners
|
||||
if (seasonDir != seriesDir) {
|
||||
fetchBanner(seasonDir["folder.jpg"], series, "season", "season", season)
|
||||
fetchBanner(seasonDir["banner.jpg"], series, "season", "seasonwide", season)
|
||||
}
|
||||
}
|
||||
|
||||
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 fetchMovieNfo(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>
|
||||
''')
|
||||
.replaceAll(/\t|\r|\n/, '') // xbmc can't handle leading/trailing whitespace properly
|
||||
.saveAs(outputFile)
|
||||
}
|
||||
def fetchMovieArtworkAndNfo(movieDir, movie) {
|
||||
println "Fetch nfo and artwork for $movie"
|
||||
def movieInfo = TheMovieDB.getMovieInfo(movie, Locale.ENGLISH)
|
||||
// fetch nfo
|
||||
fetchMovieNfo(movieDir['movie.nfo'], movieInfo)
|
||||
// fetch series banner, fanart, posters, etc
|
||||
fetchArtwork(movieDir['folder.jpg'], movieInfo, 'poster', 'original')
|
||||
fetchArtwork(movieDir['backdrop.jpg'], movieInfo, 'backdrop', 'original')
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user