moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-12-21 23:38:50 -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 02:32:09 +07:00
parent 64db545ab6
commit d6b562c77f
2 changed files with 105 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -37,48 +37,109 @@ public enum AnimeLists {
public Optional<Episode> map(Episode episode, AnimeLists destination) throws Exception {
return find(episode.getSeriesInfo().getId()).map(a -> {
if (destination == TheTVDB && a.defaulttvdbseason == null) {
// auto-align mode
if (a.defaulttvdbseason == null) {
try {
return WebServices.TheTVDB.getEpisodeList(a.tvdbid, SortOrder.Airdate, Locale.ENGLISH).stream().filter(e -> {
return episode.getEpisode() != null && episode.getEpisode().equals(e.getAbsolute());
}).findFirst().orElse(null);
return destination.mapAutoAligned(a, episode);
} catch (Exception e) {
debug.warning(e::toString);
}
return null;
}
} else {
// offset mode
Integer s = destination == TheTVDB ? a.defaulttvdbseason : null;
Integer e = episode.getEpisode();
// additional custom mapping
// offset mode
int s = getSeasonNumber(episode);
int e = getEpisodeNumber(episode);
// check explicit episode 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 (s == getSeason(m)) {
Optional<Integer> episodeMapping = destination.getEpisodeNumber(m, e);
if (episodeMapping.isPresent()) {
return derive(episode, destination.getSeason(m), episodeMapping.get());
}
}
}
}
if (a.episodeoffset != null) {
e = destination == TheTVDB ? e + a.episodeoffset : e - a.episodeoffset;
}
// apply default season
s = destination.getSeason(a, episode);
return episode.derive(s, e);
}
// apply episode offset
e += destination.getEpisodeNumberOffset(a);
return derive(episode, s, e);
});
}
private Episode derive(Episode episode, int s, int e) {
if (s == 0) {
return episode.derive(null, null, e); // special episode
} else {
return episode.derive(s, e, null); // regular episode
}
}
public Optional<Integer> map(int id, AnimeLists destination) throws Exception {
return find(id).map(destination::getId);
}
public Optional<Entry> find(int id) throws Exception {
return stream(MODEL.get().anime).filter(this::isValid).filter(a -> id == getId(a)).findFirst();
protected Episode mapAutoAligned(Entry a, Episode episode) throws Exception {
switch (this) {
case TheTVDB:
return WebServices.TheTVDB.getEpisodeList(a.tvdbid, SortOrder.Airdate, Locale.ENGLISH).stream().filter(e -> {
return episode.getEpisode() != null && episode.getEpisode().equals(e.getAbsolute());
}).findFirst().orElse(null);
case AniDB:
return WebServices.AniDB.getEpisodeList(a.anidbid, SortOrder.Absolute, Locale.ENGLISH).stream().filter(e -> {
return episode.getAbsolute() != null && episode.getAbsolute().equals(e.getEpisode());
}).findFirst().orElse(null);
}
return null;
}
protected Optional<Integer> getEpisodeNumber(Mapping m, Integer e) {
if (m.numbers != null) {
switch (this) {
case TheTVDB:
return Optional.ofNullable(m.numbers.get(e));
case AniDB:
return m.numbers.entrySet().stream().filter(it -> e.equals(it.getValue())).map(it -> it.getKey()).findFirst();
}
}
return Optional.empty();
}
protected int getEpisodeNumberOffset(Entry a) {
return a.episodeoffset == null ? 0 : this == TheTVDB ? a.episodeoffset : -a.episodeoffset;
}
protected int getSeasonNumber(Episode e) {
// special episode
if (e.getSpecial() != null) {
return 0;
}
// regular absolute episode
if (e.getSeason() == null) {
return this == AniDB ? 1 : -1;
}
// regular SxE episode
return e.getSeason();
}
protected int getEpisodeNumber(Episode e) {
return e.getSpecial() != null ? e.getSpecial() : e.getEpisode();
}
protected int getSeason(Mapping m) {
return this == AniDB ? m.anidbseason : m.tvdbseason;
}
protected int getSeason(Entry a, Episode e) {
return e.getSpecial() != null ? 0 : this == AniDB ? 1 : a.defaulttvdbseason;
}
protected int getId(Entry a) {
@ -89,6 +150,10 @@ public enum AnimeLists {
return a.anidbid != null && a.tvdbid != null;
}
public Optional<Entry> find(int id) throws Exception {
return stream(MODEL.get().anime).filter(this::isValid).filter(a -> id == getId(a)).findFirst();
}
protected static Cache getCache() {
return Cache.getCache("animelists", CacheType.Persistent);
}
@ -136,6 +201,9 @@ public enum AnimeLists {
@XmlAttribute
public Integer episodeoffset;
@XmlElement
public String name;
@XmlElementWrapper(name = "mapping-list")
public Mapping[] mapping;
@ -235,9 +303,18 @@ public enum AnimeLists {
List<Episode> episodes = WebServices.AniDB.getEpisodeList(9183, SortOrder.Absolute, Locale.ENGLISH);
for (Episode episode : episodes) {
System.out.println("\n" + episode);
System.out.println(AnimeLists.AniDB.map(episode, AnimeLists.TheTVDB).get());
}
System.out.println("----------------------------");
List<Episode> episodes2 = WebServices.TheTVDB.getEpisodeList(102261, SortOrder.Airdate, Locale.ENGLISH);
for (Episode episode : episodes2) {
System.out.println("\n" + episode);
System.out.println(AnimeLists.TheTVDB.map(episode, AnimeLists.AniDB).get());
}
System.exit(0);
}

8
source/net/filebot/web/Episode.java
View File

@ -139,12 +139,8 @@ public class Episode implements Serializable {
return new Episode(this);
}
public Episode derive(Integer season, Integer episode) {
return new Episode(getSeriesName(), season, episode, getTitle(), getAbsolute(), getSpecial(), getAirdate(), getId(), getSeriesInfo());
}
public Episode derive(Integer season, Integer episode, Integer absolute, Integer special) {
return new Episode(getSeriesName(), season, episode, getTitle(), absolute, special, getAirdate(), getId(), getSeriesInfo());
public Episode derive(Integer season, Integer episode, Integer special) {
return new Episode(getSeriesName(), season, episode, getTitle(), getAbsolute(), special, getAirdate(), getId(), getSeriesInfo());
}
public Episode derive(String seriesName, Integer season, Integer episode, Integer absolute, Integer special) {