SHOW:
|
|
- or go back to the newest paste.
| 1 | // HIT ENTER TO KILL SCRIPT | |
| 2 | Thread.startDaemon {
| |
| 3 | System.in.read() | |
| 4 | System.exit(0) | |
| 5 | } | |
| 6 | ||
| 7 | - | input.each{ f ->
|
| 7 | + | |
| 8 | - | def match = [movie:null, series:null, docu:null, anime:null] |
| 8 | + | |
| 9 | ||
| 10 | // ignore clutter files | |
| 11 | - | def movie = detectMovie(f, false) |
| 11 | + | |
| 12 | ||
| 13 | ||
| 14 | ||
| 15 | def isEpisodeFile(f) {
| |
| 16 | parseEpisodeNumber(f.name) || parseDate(f.name) || f.name =~ /\b\d+(?i:of)\d+\b/ || (parseEpisodeNumber(f.name, false) && f.name =~ /\b\p{XDigit}{8}\b/)
| |
| 17 | } | |
| 18 | ||
| 19 | ||
| 20 | def groups = input.groupBy{ f ->
| |
| 21 | def match = [movie:null, series:null, anime:null] | |
| 22 | ||
| 23 | // try as movie | |
| 24 | def movie = isEpisodeFile(f) ? null : detectMovie(f, false, Locale.ENGLISH, null, TheMovieDB) | |
| 25 | if (movie) {
| |
| 26 | match.movie = movie | |
| 27 | } | |
| 28 | ||
| 29 | // try as tv series | |
| 30 | def seriesName = detectSeriesName(f) ?: detectSeriesName(f.dir.listFiles{ it.isVideo() })
| |
| 31 | if (seriesName) {
| |
| 32 | def resultSet = TheTVDB.search(seriesName, Locale.ENGLISH) | |
| 33 | if (resultSet.size() > 0) {
| |
| 34 | - | // try as docu |
| 34 | + | |
| 35 | - | def docuName = detectSeriesName(f) ?: detectSeriesName(f.dir.listFiles{ it.isVideo() })
|
| 35 | + | |
| 36 | - | if (docuName) {
|
| 36 | + | |
| 37 | - | docuName = null // TODO docu search |
| 37 | + | |
| 38 | // try as anime | |
| 39 | def animeName = detectSeriesName(f) ?: detectSeriesName(f.dir.listFiles{ it.isVideo() })
| |
| 40 | - | println "$f => $match" |
| 40 | + | |
| 41 | - | // TODO DECIDE movie VS series VS anime VS docu |
| 41 | + | |
| 42 | if (resultSet.size() > 0) {
| |
| 43 | match.anime = resultSet | |
| 44 | } | |
| 45 | } | |
| 46 | ||
| 47 | println "$f.name => $match" | |
| 48 | return match | |
| 49 | } | |
| 50 | ||
| 51 | groups.each{ match, files ->
| |
| 52 | def score = [movie:0, series:0, anime:0] | |
| 53 | ||
| 54 | files.each{ f ->
| |
| 55 | if (parseEpisodeNumber(f.name, true) || parseDate(f.name) || f.name =~ /\b\d+(?i:of)\d+\b/) | |
| 56 | score.series+=1 | |
| 57 | ||
| 58 | if (parseEpisodeNumber(f.name, false) || f.name =~ /\b\p{XDigit}{8}\b/)
| |
| 59 | score.anime+=1 | |
| 60 | ||
| 61 | if (f =~ /\b(?:19|20)\\d{2}\b/)
| |
| 62 | score.movie+=1 | |
| 63 | ||
| 64 | // boost best match based on filename similarity | |
| 65 | score.entrySet().sort{ match[it.key] ? [[f.name, f.parentFile.name], match[it.key] as List].combinations().collect{ similarity(it[0], it[1]) }.max() : 0 }.last().value+=1
| |
| 66 | } | |
| 67 | ||
| 68 | println "$match => $score" | |
| 69 | ||
| 70 | // select best match | |
| 71 | def m = match.entrySet().sort{ score[it.key] }.last()
| |
| 72 | switch (m.key) {
| |
| 73 | case 'movie' : return rename(file:files, format:'Movies/{n} ({y})/{n} ({y}){" CD$pi"}', db:'TheMovieDB')
| |
| 74 | case 'series': return rename(file:files, format:'TV Shows/{n}/{episode.special ? "Special" : "Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB')
| |
| 75 | case 'anime' : return rename(file:files, format:'Anime/{n}/{n} - {sxe} - {t}', db:'AniDB')
| |
| 76 | } | |
| 77 | } |