moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-12-21 15:28:52 -05:00
Code Issues Releases Wiki Activity

Add anime-lists parser

Browse Source 
* https://github.com/ScudLee/anime-lists
This commit is contained in:
Reinhard Pointner 2019-06-09 00:24:43 +07:00
parent 50057a6c8c
commit abad2989f9
2 changed files with 76 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGES.md
View File

@ -1,6 +1,7 @@
Next Release (4.8.6)
====================
* Enhanced `Selection Dialog` with thumbnails and tooltips
* Improved support for mapping episode information between different databases and numbering schemes
* Added `{history}` binding for looking up the original file path of `{f}` (e.g. useful for `-exec` post-processing commands)
* Evaluate `{closures}` automatically in `String.plus(Closure)` constructs (e.g. `{"[" + {n} + " " + {s00e00} + "]"}`)
* Ensure that `ActionPopup` is always displayed on top of the Windows Task Bar
@ -10,7 +11,7 @@ Next Release (4.8.6)
* Allow `@file.groovy` syntax in `Format Editor` and `Preset Editor` (e.g. `@/path/to/MyFormat.groovy`)
* Added `--mapper` option (e.g. `--mapper XEM.TheTVDB`)
* Allow `*.groovy` files as argument value for `--format`, `--filter`, `--mapper` and `--file-filter` CLI options (e.g. `--format /path/to/MyFormat.groovy`)
* Support [bash completion](installer/bash/filebot_completion)
* Support [bash completion](installer/bash/filebot_completion)
FileBot 4.8.5

87
source/net/filebot/web/AnimeLists.java
View File

@ -4,15 +4,20 @@ import static java.util.Arrays.*;
import static java.util.stream.Collectors.*;
import static net.filebot.CachedResource.*;
import static net.filebot.Logging.*;
import static net.filebot.util.RegularExpressions.*;
import static net.filebot.util.StringUtilities.*;
import java.io.ByteArrayInputStream;
import java.io.StringWriter;
import java.net.URL;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Stream;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlElementWrapper;
@ -47,6 +52,18 @@ public enum AnimeLists {
Integer s = destination == TheTVDB ? a.defaulttvdbseason : null;
Integer e = episode.getEpisode();
// additional custom mapping
if (a.mapping != null) {
for (Mapping m : a.mapping) {
if (s == m.tvdbseason) {
Integer mappedEpisodeNumber = m.numbers.get(e);
if (mappedEpisodeNumber != null) {
return episode.derive(m.tvdbseason, mappedEpisodeNumber);
}
}
}
}
if (a.episodeoffset != null) {
e = destination == TheTVDB ? e + a.episodeoffset : e - a.episodeoffset;
}
@ -76,11 +93,7 @@ public enum AnimeLists {
return Cache.getCache("animelists", CacheType.Persistent);
}
protected static final Resource<Model> MODEL = Resource.lazy(AnimeLists::fetchModel);
protected static Model fetchModel() throws Exception {
return (Model) JAXBContext.newInstance(Model.class).createUnmarshaller().unmarshal(new ByteArrayInputStream(request("anime-list.xml")));
}
protected static final Resource<Model> MODEL = Resource.lazy(() -> unmarshal(request("anime-list.xml"), Model.class));
protected static byte[] request(String file) throws Exception {
// NOTE: GitHub only supports If-None-Match (If-Modified-Since is ignored)
@ -98,29 +111,41 @@ public enum AnimeLists {
@XmlElement
public Entry[] anime;
@Override
public String toString() {
return marshal(this, Model.class);
}
}
@XmlRootElement(name = "anime")
public static class Entry {
@XmlAttribute
public Integer anidbid;
@XmlJavaTypeAdapter(OptionalIntegerAdapter.class)
@XmlJavaTypeAdapter(NumberAdapter.class)
@XmlAttribute
public Integer tvdbid;
@XmlJavaTypeAdapter(OptionalIntegerAdapter.class)
@XmlJavaTypeAdapter(NumberAdapter.class)
@XmlAttribute
public Integer defaulttvdbseason;
@XmlJavaTypeAdapter(OptionalIntegerAdapter.class)
@XmlJavaTypeAdapter(NumberAdapter.class)
@XmlAttribute
public Integer episodeoffset;
@XmlElementWrapper(name = "mapping-list")
public Mapping[] mapping;
@Override
public String toString() {
return marshal(this, Entry.class);
}
}
@XmlRootElement(name = "mapping")
public static class Mapping {
@XmlAttribute
@ -138,11 +163,34 @@ public enum AnimeLists {
@XmlAttribute
public Integer offset;
@XmlJavaTypeAdapter(NumberMapAdapter.class)
@XmlValue
public String value;
public Map<Integer, Integer> numbers;
@Override
public String toString() {
return marshal(this, Mapping.class);
}
}
private static class OptionalIntegerAdapter extends XmlAdapter<String, Integer> {
private static <T> T unmarshal(byte[] bytes, Class<T> type) throws Exception {
return (T) JAXBContext.newInstance(type).createUnmarshaller().unmarshal(new ByteArrayInputStream(bytes));
}
private static <T> String marshal(T object, Class<T> type) {
try {
StringWriter buffer = new StringWriter();
Marshaller marshaller = JAXBContext.newInstance(type).createMarshaller();
marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
marshaller.marshal(object, buffer);
return buffer.toString();
} catch (Exception e) {
return e.toString();
}
}
private static class NumberAdapter extends XmlAdapter<String, Integer> {
@Override
public Integer unmarshal(String s) throws Exception {
@ -155,6 +203,19 @@ public enum AnimeLists {
}
}
private static class NumberMapAdapter extends XmlAdapter<String, Map<Integer, Integer>> {
@Override
public Map<Integer, Integer> unmarshal(String s) throws Exception {
return tokenize(s, SEMICOLON).map(m -> matchIntegers(m)).filter(m -> m.size() >= 2).collect(toMap(m -> m.get(0), m -> m.get(1)));
}
@Override
public String marshal(Map<Integer, Integer> m) throws Exception {
return m.entrySet().stream().map(e -> join(Stream.of(e.getKey(), e.getValue()), "-")).collect(joining(";"));
}
}
public static List<String> names() {
return stream(values()).map(Enum::name).collect(toList());
}
@ -170,11 +231,11 @@ public enum AnimeLists {
}
public static void main(String[] args) throws Exception {
System.out.println(AnimeLists.AniDB.map(14444, AnimeLists.TheTVDB));
System.out.println(AnimeLists.AniDB.map(9183, AnimeLists.TheTVDB));
List<Episode> episodes = WebServices.AniDB.getEpisodeList(14444, SortOrder.Absolute, Locale.ENGLISH);
List<Episode> episodes = WebServices.AniDB.getEpisodeList(9183, SortOrder.Absolute, Locale.ENGLISH);
for (Episode episode : episodes) {
System.out.println(AnimeLists.AniDB.map(episode, AnimeLists.TheTVDB));
System.out.println(AnimeLists.AniDB.map(episode, AnimeLists.TheTVDB).get());
}
System.exit(0);