moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-12-24 16:58:51 -05:00
Code Issues Releases Wiki Activity

* utorrent integration: + fancy notification mails + force movie/series/anime + basic anime support (no auto-detection, only if forced)

Browse Source
This commit is contained in:
Reinhard Pointner 2012-07-30 16:59:09 +00:00
parent 14e4b86344
commit 0cb56f905d
4 changed files with 107 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
source/net/sourceforge/filebot/cli/ScriptShell.lib.groovy
View File

@ -95,8 +95,19 @@ import groovy.text.GStringTemplateEngine
import net.sourceforge.filebot.format.PropertyBindings
import net.sourceforge.filebot.format.UndefinedObject
Object.metaClass.applyXmlTemplate = { template -> new XmlTemplateEngine("\t", false).createTemplate(template).make(new PropertyBindings(delegate, new UndefinedObject(""))).toString() }
Object.metaClass.applyTextTemplate = { template -> new GStringTemplateEngine().createTemplate(template).make(new PropertyBindings(delegate, new UndefinedObject(""))).toString() }
Object.metaClass.applyXml = { template -> new XmlTemplateEngine("\t", false).createTemplate(template).make(new PropertyBindings(delegate, new UndefinedObject(""))).toString() }
Object.metaClass.applyText = { template -> new GStringTemplateEngine().createTemplate(template).make(new PropertyBindings(delegate, new UndefinedObject(""))).toString() }
// MarkupBuilder helper
import groovy.xml.MarkupBuilder
def XML(bc) {
def out = new StringWriter()
def xmb = new MarkupBuilder(out)
bc.rehydrate(bc.delegate, xmb, xmb).call() // call closure in MarkupBuilder context
return out.toString()
}
// Shell helper
@ -156,7 +167,7 @@ List.metaClass.watch = { c -> createWatchService(c, delegate, true) }
def getRenameLog(complete = false) {
def spooler = net.sourceforge.filebot.HistorySpooler.getInstance()
def history = complete ? spooler.completeHistory : spooler.sessionHistory
return history.sequences*.elements.flatten().collectEntries{ [new File(it.dir, it.from), new File(it.dir, it.to)] }
return history.sequences*.elements.flatten().collectEntries{ [new File(it.dir, it.from), new File(it.to).isAbsolute() ? new File(it.to) : new File(it.dir, it.to)] }
}
// Season / Episode helpers

8
website/scripts/lib/ant.groovy
View File

@ -18,7 +18,7 @@ def sshexec(param) {
* e.g.
* mail(mailhost:'smtp.gmail.com', mailport:'587', ssl:'no', enableStartTLS:'yes', user:'rednoah@gmail.com', password:'correcthorsebatterystaple', from:'rednoah@gmail.com', to:'someone@gmail.com', subject:'Hello Ant World', message:'Dear Ant, ...')
*/
def mail(param) {
def sendmail(param) {
def sender = param.remove('from')
def recipient = param.remove('to')
@ -35,12 +35,12 @@ def mail(param) {
* e.g.
* gmail(subject:'Hello Ant World', message:'Dear Ant, ...', to:'someone@gmail.com', user:'rednoah', password:'correcthorsebatterystaple')
*/
def gmail(param) {
def sendGmail(param) {
param << [mailhost:'smtp.gmail.com', mailport:'587', ssl:'no', enableStartTLS:'yes']
param << [user:param.username ? param.remove('username') + "@gmail.com" : param.user]
param << [user:param.username ? param.remove('username') + '@gmail.com' : param.user]
param << [from: param.from ?: param.user]
mail(param)
sendmail(param)
}

4
website/scripts/lib/htpc.groovy
View File

@ -66,7 +66,7 @@ def fetchSeriesFanart(outputFile, series, type, season, locale) {
def fetchSeriesNfo(outputFile, series, locale) {
def info = TheTVDB.getSeriesInfo(series, locale)
info.applyXmlTemplate('''<tvshow xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
info.applyXml('''<tvshow xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
<title>$name</title>
<year>$firstAired.year</year>
<rating>$rating</rating>
@ -188,7 +188,7 @@ def createFileInfoXml(file) {
}
def fetchMovieNfo(outputFile, movieInfo, movieFile) {
movieInfo.applyXmlTemplate('''<movie xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
movieInfo.applyXml('''<movie xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
<title>$name</title>
<originaltitle>$originalName</originaltitle>
<set>$collection</set>

100
website/scripts/utorrent-postprocess.groovy
View File

@ -1,9 +1,9 @@
// filebot -script "fn:utorrent-postprocess" --output "X:/media" --action copy --conflict override --def subtitles=true artwork=true xbmc=localhost plex=10.0.0.3 "ut_dir=%D" "ut_file=%F" "ut_kind=%K" "ut_label=%L" "ut_state=%S"
// filebot -script "fn:utorrent-postprocess" --output "X:/media" --action copy --conflict override --def subtitles=true artwork=true xbmc=localhost plex=10.0.0.1 gmail=username:password "ut_dir=%D" "ut_file=%F" "ut_kind=%K" "ut_title=%N" "ut_label=%L" "ut_state=%S"
def input = []
def failOnError = _args.conflict == 'fail'
// print input parameters
_args.bindings?.each{ println "Parameter: $it.key = $it.value" }
_args.bindings?.each{ _log.finest("Parameter: $it.key = $it.value") }
// disable enable features as specified via --def parameters
def subtitles = tryQuietly{ subtitles.toBoolean() }
@ -13,11 +13,31 @@ def artwork = tryQuietly{ artwork.toBoolean() }
def xbmc = tryQuietly{ xbmc.split(/[\s,|]+/) }
def plex = tryQuietly{ plex.split(/[\s,|]+/) }
// email notifications
def gmail = tryQuietly{ gmail.split(':', 2) }
// force movie/series/anime logic
def forceMovie(f) {
tryQuietly{ ut_label } =~ /^(?i:Movie|Couch.Potato)/
}
def forceSeries(f) {
parseEpisodeNumber(f) || parseDate(f) || tryQuietly{ ut_label } =~ /^(?i:TV)/
}
def forceAnime(f) {
tryQuietly{ ut_label } =~ /^(?i:Anime)/
}
def forceIgnore(f) {
tryQuietly{ ut_label } =~ /^(?i:Music|Ebook|other)/
}
// collect input fileset as specified by the given --def parameters
if (args.empty) {
// assume we're called with utorrent parameters
if (ut_kind == "single") {
if (ut_kind == 'single') {
input += new File(ut_dir, ut_file) // single-file torrent
} else {
input += new File(ut_dir).getFiles() // multi-file torrent
@ -37,15 +57,26 @@ input = input.findAll{ it.isVideo() || it.isSubtitle() }
input = input.findAll{ !(it.path =~ /\b(?i:sample|trailer|extras|deleted.scenes|music.video|scrapbook)\b/) }
// print input fileset
input.each{ println "Input: $it" }
input.each{ f -> _log.finest("Input: $f") }
// artwork/nfo utility
include("lib/htpc")
// group episodes/movies and rename according to XBMC standards
def groups = input.groupBy{ f ->
// skip auto-detection if possible
if (forceIgnore(f))
return []
if (forceMovie(f))
return [mov: detectMovie(f, false)]
if (forceSeries(f))
return [tvs: detectSeriesName(f)]
if (forceAnime(f))
return [anime: detectSeriesName(f)]
def tvs = detectSeriesName(f)
def mov = (parseEpisodeNumber(f) || parseDate(f)) ? null : detectMovie(f, false) // skip movie detection if we can already tell it's an episode
def mov = detectMovie(f, false)
println "$f.name [series: $tvs, movie: $mov]"
// DECIDE EPISODE VS MOVIE (IF NOT CLEAR)
@ -71,9 +102,13 @@ def groups = input.groupBy{ f ->
throw new Exception("Media detection failed")
}
return [tvs:tvs, mov:mov]
return [tvs: tvs, mov: mov, anime: null]
}
// log movie/series/anime detection results
groups.each{ group -> _log.finest("Group: $group") }
// process each batch
groups.each{ group, files ->
// fetch subtitles
if (subtitles) {
@ -81,28 +116,31 @@ groups.each{ group, files ->
}
// EPISODE MODE
if (group.tvs && !group.mov) {
def dest = rename(file:files, format:'TV Shows/{n}/{episode.special ? "Special" : "Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB')
if ((group.tvs || group.anime) && !group.mov) {
// choose series / anime config
def config = group.tvs ? [name: group.tvs, format:'TV Shows/{n}/{episode.special ? "Special" : "Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB']
: [name: group.anime, format:'Anime/{n}/{n} - {e.pad(2)} - {t}', db:'AniDB']
def dest = rename(file: files, format: config.format, db: config.db)
if (dest && artwork) {
dest.mapByFolder().each{ dir, fs ->
println "Fetching artwork for $dir from TheTVDB"
def sxe = fs.findResult{ eps -> parseEpisodeNumber(eps) }
def options = TheTVDB.search(group.tvs)
def options = TheTVDB.search(config.name)
if (options.isEmpty()) {
println "TV Series not found: $group.tvs"
println "TV Series not found: $config.name"
return
}
options = options.sortBySimilarity(group.tvs, { opt -> opt.name })
options = options.sortBySimilarity(config.name, { s -> s.name })
fetchSeriesArtworkAndNfo(dir.dir, dir, options[0], sxe && sxe.season > 0 ? sxe.season : 1)
}
}
if (dest == null && failOnError) {
throw new Exception("Failed to rename series: $group.tvs")
throw new Exception("Failed to rename series: $config.name")
}
}
// MOVIE MODE
if (group.mov && !group.tvs) {
if (group.mov && !group.tvs && !group.anime) {
def dest = rename(file:files, format:'Movies/{n} ({y})/{n} ({y}){" CD$pi"}{".$lang"}', db:'TheMovieDB')
if (dest && artwork) {
dest.mapByFolder().each{ dir, fs ->
@ -128,3 +166,39 @@ plex?.each{
println "Notify Plex: $it"
refreshPlexLibrary(it)
}
// send status email
if (gmail && getRenameLog().size() > 0) {
// ant/mail utility
include('lib/ant')
// send html mail
def renameLog = getRenameLog()
sendGmail(
subject: '[FileBot] ' + ut_title,
message: XML {
html {
body {
p("FileBot finished processing ${ut_title} (${renameLog.size()} files).");
hr(); table {
th("Parameter"); th("Value")
_args.bindings.findAll{ param -> param.key =~ /^ut_/ }.each{ param ->
tr { [param.key, param.value].each{ td(it)} }
}
}
hr(); table {
th("Original Name"); th("New Name"); th("New Location")
renameLog.each{ from, to ->
tr { [from.name, to.name, to.parent].each{ cell -> td { code(cell) } } }
}
}
hr(); small("// Generated by ${net.sourceforge.filebot.Settings.applicationIdentifier} on ${new Date().dateString} at ${new Date().timeString}")
}
}
},
messagemimetype: "text/html",
to: tryQuietly{ gmail2 } ?: gmail[0] =~ /@/ ? gmail[0] : gmail[0] + '@gmail.com',
user: gmail[0], password: gmail[1]
)
}