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

* support {genres} for AniDB seriesInfo

Browse Source
This commit is contained in:
Reinhard Pointner 2014-12-20 18:46:54 +00:00
parent 07a5402a84
commit eaa0b9a5bb
2 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
source/net/filebot/util/XPathUtilities.java
View File

@ -90,6 +90,14 @@ public final class XPathUtilities {
return null;
}
public static Integer getIntegerAttribute(String attribute, Node node) {
try {
return new Scanner(getAttribute(attribute, node)).useDelimiter("\\D+").nextInt();
} catch (Exception e) {
return null;
}
}
/**
* Get text content of the first child node matching the given node name. Use this method instead of {@link #selectString(String, Object)} whenever xpath support is not required, because it is much faster, especially for large documents.
*

16
source/net/filebot/web/AnidbClient.java
View File

@ -8,6 +8,7 @@ import static net.filebot.web.WebRequest.*;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.AbstractMap.SimpleImmutableEntry;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
@ -23,6 +24,7 @@ import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import java.util.zip.GZIPInputStream;
import javax.swing.Icon;
@ -124,6 +126,20 @@ public class AnidbClient extends AbstractEpisodeListProvider {
seriesInfo.setRatingCount(new Integer(selectString("anime/ratings/permanent/@count", dom)));
seriesInfo.setStartDate(SimpleDate.parse(selectString("anime/startdate", dom), "yyyy-MM-dd"));
// add categories ordered by weight as genres
// * only use categories with weight >= 400
// * sort by weight (descending)
// * limit to 5 genres
seriesInfo.setGenres(selectNodes("anime/categories/category", dom).stream().map(categoryNode -> {
String name = getTextContent("name", categoryNode);
Integer weight = getIntegerAttribute("weight", categoryNode);
return new SimpleImmutableEntry<String, Integer>(name, weight);
}).filter(nw -> {
return nw.getKey() != null && nw.getValue() != null && nw.getKey().length() > 0 && nw.getValue() >= 400;
}).sorted((a, b) -> {
return b.getValue().compareTo(a.getValue());
}).map(it -> it.getKey()).limit(5).collect(Collectors.toList()));
// parse episode data
String animeTitle = selectString("anime/titles/title[@type='official' and @lang='" + locale.getLanguage() + "']", dom);
if (animeTitle == null || animeTitle.length() == 0) {