Merge branch 'master' of https://github.com/tylerforesthauser/scripts

Author: tylerforesthauser

.gitignore

@@ -1 +1,4 @@
-build
+build
+
+# Ignore Mac DS_Store files
+.DS_Store

README.md

@@ -7,8 +7,8 @@ and much more powerful. The provided functions are the same as
 in the CLI and parameter usage is also exactly the same.
 
 ## Links
-* [Usage and Examples](https://www.filebot.net/forums/viewtopic.php?f=4&t=5)
-* [AMC Script Manual](https://www.filebot.net/forums/viewtopic.php?f=4&t=215)
+* [Usage and Examples](https://www.filebot.net/forums/viewtopic.php?t=5)
+* [AMC Script Manual](https://www.filebot.net/amc.html)
 
 ## Clone
 If you want to run scripts from a local file or if you want to make your own modifications just clone the repository into your local filesystem:

amc.groovy

@@ -1,49 +1,94 @@
 #!/usr/bin/env filebot -script
 
 
-def fetchMovieArtwork(file, movie, language) {
-	fetch movie.getArtwork('posters', language)
-	fetch movie.getArtwork('backdrops', null)
+def fetchMovieArtwork(movie, language, movieFolder) {
+	fetchArtwork(movie, 'posters', language, movieFolder / 'poster.jpg')
+	fetchArtwork(movie, 'backdrops', null, movieFolder / 'backdrop.jpg')
 }
 
 
-def fetchSeriesArtwork(file, series, language) {
-	switch(series.database) {
-		case ~/TheTVDB/:
-			fetch series.getArtwork('poster', language)
-			fetch series.getArtwork('series', language)
-			fetch series.getArtwork('season', language)
-			fetch series.getArtwork('seasonwide', language)
-			break
-		case ~/TheMovieDB/:
-			fetch series.getArtwork('posters', language)
-			fetch series.getArtwork('backdrops', null)
-			break
-		default:
-			fetch series.getArtwork()
-			break
+def fetchSeriesArtwork(series, language, seriesFolder, seasonFolder) {
+	if (series.database == 'TheTVDB') {
+		fetchArtwork(series, 'poster', language, seriesFolder / 'poster.jpg')
+		fetchArtwork(series, 'series', language, seriesFolder / 'series.jpg')
+		fetchArtwork(series, 'fanart', language, seriesFolder / 'fanart.jpg')
+		fetchArtwork(series, 'season', language, seasonFolder / 'season.jpg')
+		fetchArtwork(series, 'seasonwide', language, seasonFolder / 'seasonwide.jpg')
+		return
+	}
+
+	if (series.database == 'TheMovieDB::TV') {
+		fetchArtwork(series, 'posters', language, seriesFolder / 'poster.jpg')
+		fetchArtwork(series, 'backdrops', null, seriesFolder / 'backdrop.jpg')
+		return
 	}
+
+	log.warning "Artwork not supported: $series"
+	return
 }
 
 
-def fetch(artwork) {
-	artwork.eachWithIndex{ a, i ->
-		log.fine "$i: $a"
+def fetchEpisodeArtwork(episode, episodeFile) {
+	def thumbnailFile = episodeFile.dir / episodeFile.nameWithoutExtension + '.jpg'
+	if (thumbnailFile.exists()) {
+		log.finest "[SKIP] Artwork already exists: $thumbnailFile"
+		return
+	}
+
+	def i = episode.info
+	if (i == null) {
+		log.warning "Artwork not supported: $episode.seriesInfo"
+		return
+	}
+	if (i.image == null) {
+		log.warning "Artwork not found: $episode [$thumbnailFile]"
+		return
+	}
+
+	log.fine "Fetch $i.image [$thumbnailFile]"
+	try {
+		i.image.saveAs(thumbnailFile)
+	} catch(e) {
+		log.severe "Failed to fetch artwork: $e.message"
 	}
 }
 
 
+def fetchArtwork(object, category, language, file) {
+	if (file.exists()) {
+		log.finest "[SKIP] Artwork already exists: $file"
+		return
+	}
+
+	def artwork = object.getArtwork(category, language)
+	if (artwork == null || artwork.empty) {
+		log.warning "Artwork not found: $object [$file]"
+		return
+	}
+
+	def a = artwork[0]
+	log.fine "Fetch $a [$file]"
+	try {
+		a.url.saveAs(file)
+	} catch(e) {
+		log.severe "Failed to fetch artwork: $e.message"
+	}
+}
+
+
+
+
 args.getFiles{ it.video }.each{ f ->
 	def m = f.metadata
-
 	switch(m) {
 		case Movie:
 			log.finest "[MOVIE] $m [$f]"
-			fetchMovieArtwork(f, m, m.language)
+			fetchMovieArtwork(m, m.language, f.dir)
 			break
 		case Episode:
 			log.finest "[EPISODE] $m [$f]"
-			fetchSeriesArtwork(f, m.seriesInfo, m.seriesInfo.language)
+			fetchSeriesArtwork(m.seriesInfo, m.seriesInfo.language, f.dir.dir, f.dir)
+			fetchEpisodeArtwork(m, f)
 			break;
 		default:
 			log.finest "[XATTR NOT FOUND] $f"

artwork.groovy

@@ -12,25 +12,25 @@ include('lib/htpc')
 
 args.eachMediaFolder{ dir ->
 	// fetch only missing artwork by default
-	if (dir.hasFile{it.name == 'movie.nfo'} && dir.hasFile{it.name == 'poster.jpg'} && dir.hasFile{it.name == 'fanart.jpg'}) {
-		log.finest "Skipping $dir"
+	if (dir.hasFile{ it.name == 'poster.jpg' }) {
+		log.finest "Skip [$dir] because [poster.jpg] already exists"
 		return
 	}
 
 	def videos = dir.listFiles{ it.isVideo() }
 	def query = _args.query
-	def locale = any{ _args.language.locale }{ Locale.ENGLISH }
+	def locale = _args.language.locale
 	def options = []
 
 	if (query) {
 		// manual search & sort by relevance
-		options = TheMovieDB.searchMovie(query, locale).sortBySimilarity(query, { it.name })
+		options = TheMovieDB.searchMovie(query, locale).sortBySimilarity(query){ it.name }
 	} else if (videos.size() > 0) {
 		// run movie auto-detection for video files
 		options = MediaDetection.detectMovie(videos[0], TheMovieDB, locale, true)
 	}
 
-	if (options.isEmpty()) {
+	if (!options) {
 		log.warning "$dir ${videos.name} => movie not found"
 		return
 	}

artwork.tmdb.groovy

@@ -1,7 +1,7 @@
 #!/usr/bin/env filebot -script
 
 
-// sanity checks
+// require at least 1 input folder
 if (args.size() == 0) {
 	die "Illegal usage: no input"
 }
@@ -12,48 +12,57 @@ include('lib/htpc')
 
 args.eachMediaFolder{ dir ->
 	// fetch only missing artwork by default
-	if (dir.hasFile{it.name == 'banner.jpg'}) {
-		log.finest "Skipping $dir"
+	if (dir.hasFile{ it.name == 'poster.jpg' }) {
+		log.finest "Skip [$dir] because [poster.jpg] already exists"
 		return
 	}
 
 	def videos = dir.listFiles{ it.isVideo() }
-	def query = _args.query ?: detectSeriesName(videos)
 	def sxe = videos.findResult{ parseEpisodeNumber(it) }
-	def locale = any{ _args.language.locale }{ Locale.ENGLISH }
+	def locale = _args.language.locale
 
-	if (query == null) {
+	// use --q option value
+	def query = _args.query
+	// use series detection
+	if (!query) {
+		def s = detectSeries(videos)
+		if (s) {
+			query = (s.getExternalId('TheTVDB') as String) ?: s.name
+		}
+	}
+	// use series folder name
+	if (!query) {
 		query = dir.dir.hasFile{ it.name =~ /Season/ && it.isDirectory() } ? dir.dir.name : dir.name
 	}
 
-	log.finest "$dir => Search by $query"
-	def options = TheTVDB.search(query, locale)
+	log.finest "$dir => Lookup by $query"
+	def options = TheTVDB.lookup(query, locale)
 	if (options.isEmpty()) {
 		log.warning "TV Series not found: $query"
 		return
 	}
 
 	// sort by relevance
-	options = options.sortBySimilarity(query, { it.name })
+	options = options.sortBySimilarity(query){ it.name }
 
 	// auto-select series
-	def series = options[0]
+	def id = options.first()
 
 	// maybe require user input
 	if (options.size() > 1 && _args.strict) {
-		series = showInputDialog(options, query, 'Select TV Series')
+		id = showInputDialog(options, query, 'Select TV Series')
 	}
 
-	if (series == null) {
+	if (id == null) {
 		return null
 	}
 
 	// auto-detect structure
-	def seriesDir = [dir.dir, dir].sortBySimilarity(series.name, { it.name })[0]
+	def seriesInfo = TheTVDB.getSeriesInfo(id, locale)
+
+	def seriesDir = [dir.dir, dir].sortBySimilarity(seriesInfo.name){ it.name }.first()
 	def season = sxe && sxe.season > 0 ? sxe.season : 1
 
-	log.fine "$dir => $series"
-	tryLogCatch {
-		fetchSeriesArtworkAndNfo(seriesDir, dir, series.id, season, false, locale)
-	}
+	log.fine "$dir => $seriesInfo.name [$seriesInfo]"
+	fetchSeriesArtworkAndNfo(seriesDir, dir, seriesInfo, season, false, locale)
 }

artwork.tvdb.groovy

@@ -4,10 +4,10 @@
 def deleteRootFolder = any{ root.toBoolean() }{ false }
 
 def ignore  = any{ ignore }{ /extrathumbs/ }
-def exts    = any{ exts }{ /jpg|jpeg|png|gif|ico|nfo|info|xml|htm|html|log|m3u|cue|ffp|srt|sub|idx|smi|sup|md5|sfv|txt|rtf|url|db|dna|exe|log|tgmd|json|data|ignore|srv|srr|nzb|vbs|ini|vsmeta/ }
-def terms   = any{ terms }{ /sample|trailer|extras|deleted.scenes|music.video|scrapbook|DS_Store/ }
+def exts    = any{ exts }{ /jpg|jpeg|png|gif|ico|nfo|info|xml|htm|html|log|m3u|cue|ffp|srt|sub|idx|smi|sup|md5|sfv|txt|rtf|url|website|db|dna|exe|log|tgmd|json|data|ignore|srv|srr|nzb|vbs|ini|vsmeta/ }
+def terms   = any{ terms }{ /sample|trailer|extras|deleted.scenes|music.video|scrapbook|DS_Store|eaDir/ }
 def minsize = any{ minsize.toLong() }{ 20 * 1024 * 1024 }
-def maxsize = any{ maxsize.toLong() }{ 100 * 1024 * 1024 }
+def maxsize = any{ maxsize.toLong() }{ 120 * 1024 * 1024 }
 
 
 def testRun = _args.action.equalsIgnoreCase('test')
@@ -24,17 +24,17 @@ if (args.size() == 0) {
 // delete orphaned "clutter" files like nfo, jpg, etc and sample files
 def isClutter = { f ->
 	// whitelist
-	if (f.path.findMatch(ignore))
+	if (f.path.findWordMatch(ignore))
 		return false
 
 	// file is either too small to have meaning, or to large to be considered clutter
 	def fsize = f.length()
 
 	// path contains blacklisted terms or extension is blacklisted
-	if (f.extension ==~ exts && fsize < maxsize)
+	if (f.extension?.findWordMatch(exts) && fsize < maxsize)
 		return true
 
-	if (f.path.findMatch(/\b(/ + terms + /)\b/) && fsize < maxsize)
+	if (f.path.findWordMatch(terms) && fsize < maxsize)
 		return true
 
 	// NOTE: some smb filesystem implementations are buggy and known to incorrectly return filesize 0 for all files
@@ -58,15 +58,43 @@ def clean = { f ->
 
 
 // memoize media folder status for performance
-def hasMediaFiles = { dir -> dir.isDirectory() && dir.getFiles().find{ (it.isVideo() || it.isAudio()) && !isClutter(it) } }.memoize()
+def hasMediaFiles = { dir ->
+	return dir.isDirectory() && dir.getFiles().any{ f -> (f.isVideo() || f.isAudio()) && !isClutter(f) }
+}.memoize()
+
 
 // delete clutter files in orphaned media folders
-args.getFiles{ isClutter(it) && !hasMediaFiles(it.dir) }.each{ clean(it) }
+args.files.each{ f ->
+	// keep non-clutter files
+	if (!isClutter(f)) {
+		log.finest "Keep $f (not clutter)"
+		return
+	}
+
+	// keep sibling files of non-clutter files
+	if (hasMediaFiles(f.dir)) {
+		log.finest "Keep $f (parent folder contains media files)"
+		return
+	}
+
+	clean(f)
+}
+
 
 // delete empty folders but exclude given args
-args.getFolders().toSorted().reverse().each{
-	if (it.isDirectory() && !it.hasFile{ it.isDirectory() || !isClutter(it) }) {
-		if (deleteRootFolder || !args.contains(it))
-			clean(it)
+args.folders.toSorted().reverse().each{ d ->
+	// skip input folder
+	if (!deleteRootFolder && args.contains(d)) {
+		log.finest "Keep $d (root folder)"
+		return
 	}
+
+	// skip non-empty folder
+	if (d.hasFile{ f -> f.isDirectory() || !isClutter(f) }) {
+		log.finest "Keep $d (not empty)"
+		return
+	}
+
+	// delete folder
+	clean(d)
 }