moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-08-13 17:03:45 -04:00
Code Issues Releases Wiki Activity

Improved behavior for -rename --q commands

Browse Source
This commit is contained in:
Reinhard Pointner 2017-02-06 00:26:24 +08:00
parent 2efc3c17fa
commit ddf40be144
3 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
source/net/filebot/cli/CmdlineOperations.java
View File

@ -278,7 +278,7 @@ public class CmdlineOperations implements CmdlineInterface {
// select search result
if (results.size() > 0) {
List<SearchResult> selectedSearchResults = selectSearchResult(query, results, true, strict, limit);
List<SearchResult> selectedSearchResults = selectSearchResult(query, results, true, true, strict, limit);
if (selectedSearchResults != null) {
for (SearchResult it : selectedSearchResults) {
@ -389,7 +389,7 @@ public class CmdlineOperations implements CmdlineInterface {
}
// force all mappings
Movie movie = (Movie) selectSearchResult(query, options);
Movie movie = selectSearchResult(query, options);
for (File file : files) {
movieByFile.put(file, movie);
}
@ -424,6 +424,8 @@ public class CmdlineOperations implements CmdlineInterface {
// reduce options to perfect matches if possible
List<Movie> perfectMatches = matchMovieByWordSequence(getName(file), options, 0);
// narrow down options if possible
if (perfectMatches.size() > 0) {
options = perfectMatches;
}
@ -431,13 +433,13 @@ public class CmdlineOperations implements CmdlineInterface {
try {
// select first element if matches are reliable
if (options.size() > 0) {
movie = (Movie) selectSearchResult(null, options);
movie = selectSearchResult(stripReleaseInfo(getName(file)), options);
// make sure to get the language-specific movie object for the selected option
movie = getLocalizedMovie(service, movie, locale);
}
} catch (Exception e) {
log.log(Level.WARNING, format("%s: [%s] %s", e.getClass().getSimpleName(), getStructurePathTail(file), e.getMessage()));
log.warning(cause(e));
}
}
@ -844,18 +846,18 @@ public class CmdlineOperations implements CmdlineInterface {
return output;
}
protected SearchResult selectSearchResult(String query, Collection<? extends SearchResult> options) throws Exception {
List<SearchResult> matches = selectSearchResult(query, options, false, false, 1);
protected <T extends SearchResult> T selectSearchResult(String query, Collection<T> options) throws Exception {
List<T> matches = selectSearchResult(query, options, false, false, false, 1);
return matches.size() > 0 ? matches.get(0) : null;
}
protected List<SearchResult> selectSearchResult(String query, Collection<? extends SearchResult> options, boolean alias, boolean strict, int limit) throws Exception {
List<SearchResult> probableMatches = getProbableMatches(query, options, alias, strict);
protected <T extends SearchResult> List<T> selectSearchResult(String query, Collection<T> options, boolean sort, boolean alias, boolean strict, int limit) throws Exception {
List<T> probableMatches = getProbableMatches(sort ? query : null, options, alias, strict);
if (probableMatches.isEmpty() || (strict && probableMatches.size() != 1)) {
// allow single search results to just pass through in non-strict mode even if match confidence is low
if (options.size() == 1 && !strict) {
return new ArrayList<SearchResult>(options);
return options.stream().collect(toList());
}
if (strict) {
@ -863,8 +865,8 @@ public class CmdlineOperations implements CmdlineInterface {
}
// just pick the best N matches
if (query != null) {
probableMatches = new ArrayList<SearchResult>(sortBySimilarity(options, singleton(query), getSeriesMatchMetric()));
if (sort) {
probableMatches = sortBySimilarity(options, singleton(query), getSeriesMatchMetric()).stream().collect(toList());
}
}

8
source/net/filebot/cli/CmdlineOperationsTextUI.java
View File

@ -95,15 +95,15 @@ public class CmdlineOperationsTextUI extends CmdlineOperations {
}
@Override
protected List<SearchResult> selectSearchResult(String query, Collection<? extends SearchResult> options, boolean alias, boolean strict, int limit) throws Exception {
List<SearchResult> matches = getProbableMatches(query, options, alias, false);
protected <T extends SearchResult> List<T> selectSearchResult(String query, Collection<T> options, boolean sort, boolean alias, boolean strict, int limit) throws Exception {
List<T> matches = getProbableMatches(sort ? query : null, options, alias, false);
if (matches.size() < 2) {
if (matches.size() <= 1) {
return matches;
}
// manually select option if there is more than one
SearchResult selection = showInputDialog(matches, "Multiple Options", String.format("Select best match for \"%s\"", query));
T selection = showInputDialog(matches, "Multiple Options", String.format("Select best match for \"%s\"", query));
if (selection == null) {
return emptyList();

8
source/net/filebot/media/MediaDetection.java
View File

@ -1290,16 +1290,16 @@ public class MediaDetection {
return tvdbId.isEmpty() ? null : WebServices.TheTVDB.getSeriesInfo(tvdbId.get(0), locale);
}
public static List<SearchResult> getProbableMatches(String query, Collection<? extends SearchResult> options, boolean alias, boolean strict) {
public static <T extends SearchResult> List<T> getProbableMatches(String query, Collection<T> options, boolean alias, boolean strict) {
if (query == null) {
return options.stream().distinct().collect(toList());
}
// check all alias names, or just the primary name
Function<SearchResult, Collection<String>> names = alias ? SearchResult::getEffectiveNames : (it) -> singleton(it.getName());
Function<SearchResult, Collection<String>> names = alias ? SearchResult::getEffectiveNames : f -> singleton(f.getName());
// auto-select most probable search result
List<SearchResult> probableMatches = new ArrayList<SearchResult>();
List<T> probableMatches = new ArrayList<T>();
// use name similarity metric
SimilarityMetric metric = new NameSimilarityMetric();
@ -1310,7 +1310,7 @@ public class MediaDetection {
String q = removeTrailingBrackets(query).toLowerCase();
// find probable matches using name similarity > 0.8 (or > 0.6 in non-strict mode)
for (SearchResult option : options) {
for (T option : options) {
float f = 0;
for (String n : names.apply(option)) {
n = removeTrailingBrackets(n).toLowerCase();