moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-11-16 06:15:02 -05:00
Code Issues Releases Wiki Activity

make TVMazeClient more robust to unexpected json response data

Browse Source
This commit is contained in:
Reinhard Pointner 2016-01-17 15:59:00 +00:00
parent 8ca67b9b64
commit 8e7bf88b06
1 changed files with 27 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
source/net/filebot/web/TVMazeClient.java
View File

@ -10,6 +10,7 @@ import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.stream.Collectors;
@ -68,8 +69,8 @@ public class TVMazeClient extends AbstractEpisodeListProvider {
for (Object result : response.getArray()) {
Map<?, ?> show = (Map<?, ?>) ((Map<?, ?>) result).get("show");
int id = Integer.parseInt(show.get("id").toString());
String name = show.get("name").toString();
Integer id = getValue(show, "id", Integer::new);
String name = getValue(show, "name", String::new);
// FUTURE WORK: consider adding TVmaze aka titles for each result, e.g. http://api.tvmaze.com/shows/1/akas
results.add(new TVMazeSearchResult(id, name));
@ -83,9 +84,9 @@ public class TVMazeClient extends AbstractEpisodeListProvider {
// e.g. http://api.tvmaze.com/shows/1
JsonObject<?, ?> response = request("shows/" + show.getId());
String status = (String) response.get("status");
String premiered = (String) response.get("premiered");
String runtime = response.get("runtime").toString();
String status = getValue(response, "status", String::new);
SimpleDate premiered = getValue(response, "premiered", s -> SimpleDate.parse(s, "yyyy-MM-dd"));
Integer runtime = getValue(response, "runtime", Integer::new);
JsonObject<?, ?> genres = (JsonObject<?, ?>) response.get("genres");
JsonObject<?, ?> rating = (JsonObject<?, ?>) response.get("rating");
@ -93,14 +94,14 @@ public class TVMazeClient extends AbstractEpisodeListProvider {
seriesInfo.setName(show.getName());
seriesInfo.setAliasNames(show.getEffectiveNames());
seriesInfo.setStatus(status);
seriesInfo.setRuntime(new Integer(runtime));
seriesInfo.setStartDate(SimpleDate.parse(premiered, "yyyy-MM-dd"));
seriesInfo.setRuntime(runtime);
seriesInfo.setStartDate(premiered);
if (genres != null && genres.isArray()) {
seriesInfo.setGenres(Arrays.stream(genres.getArray()).map(Objects::toString).collect(Collectors.toList()));
}
if (rating != null && !rating.isEmpty()) {
seriesInfo.setRating(new Double(rating.get("average").toString()));
seriesInfo.setRating(getValue(rating, "average", Double::new));
}
return seriesInfo;
@ -120,22 +121,31 @@ public class TVMazeClient extends AbstractEpisodeListProvider {
if (response.isArray()) {
for (Object element : response.getArray()) {
JsonObject<?, ?> episode = (JsonObject<?, ?>) element;
try {
String episodeTitle = episode.get("name").toString();
Integer seasonNumber = Integer.parseInt(episode.get("season").toString());
Integer episodeNumber = Integer.parseInt(episode.get("number").toString());
SimpleDate airdate = SimpleDate.parse(episode.get("airdate").toString(), "yyyy-MM-dd");
String episodeTitle = getValue(episode, "name", String::new);
Integer seasonNumber = getValue(episode, "season", Integer::new);
Integer episodeNumber = getValue(episode, "number", Integer::new);
SimpleDate airdate = getValue(episode, "airdate", s -> SimpleDate.parse(s, "yyyy-MM-dd"));
episodes.add(new Episode(seriesInfo.getName(), seasonNumber, episodeNumber, episodeTitle, null, null, airdate, new SeriesInfo(seriesInfo)));
} catch (Exception e) {
Logger.getLogger(TVMazeClient.class.getName()).log(Level.WARNING, "Illegal episode data: " + e + ": " + episode);
}
}
}
return new SeriesData(seriesInfo, episodes);
}
private <V> V getValue(Map<?, ?> json, String key, Function<String, V> converter) {
try {
Object value = json.get(key);
if (value != null) {
return converter.apply(value.toString());
}
} catch (Exception e) {
Logger.getLogger(TVMazeClient.class.getName()).log(Level.WARNING, "Illegal " + key + " value: " + json);
}
return null;
}
public JsonObject<?, ?> request(String resource) throws IOException {
return new CachedJsonResource("http://api.tvmaze.com/" + resource).getJSON();
}