switched api to http://mobile.defas-fgi.de/beg/ and added tests for the corresponding functions

This commit is contained in:
Johannes Loher 2017-04-11 18:57:10 +02:00
parent 82f134f565
commit 0f95b5777e
2 changed files with 210 additions and 144 deletions

View file

@ -6,22 +6,29 @@ import std.getopt : defaultGetoptPrinter, getopt;
import std.json : JSONValue;
import std.stdio : File, writeln;
import requests : postContent;
import requests : getContent;
import fahrplanparser;
import substitution;
// find stops: http://mobile.defas-fgi.de/beg/XML_STOPFINDER_REQUEST?outputFormat=XML&stateless=1&locationServerActive=1&regionID_sf=1&type_sf=stop&name_sf=Regensburg%20Universit%C3%A4t
enum baseURL = "http://mobile.defas-fgi.de/beg/";
enum departureMonitorRequest = "XML_DM_REQUEST";
enum stopFinderRequest = "XML_STOPFINDER_REQUEST";
void main(string[] args)
{
string fileName;
string busStop = "4014080";
string substitutionFileName = "replacement.txt";
// dfmt off
auto helpInformation = getopt(args,
"file|f", "The file that the data is written to.", &fileName,
"stop|s", "The bus stop for which to fetch data.", &busStop,
"replacement-file|r", "The file that contais the direction name replacement info.", &substitutionFileName);
// dfmt on
if (helpInformation.helpWanted)
{
@ -29,13 +36,20 @@ void main(string[] args)
return;
}
auto content = postContent("http://txt.bayern-fahrplan.de/textversion/bcl_abfahrtstafel",
["limit" : "20",
"useRealtime" : "1",
// dfmt off
auto content = getContent(baseURL ~ departureMonitorRequest,
["outputFormat" : "XML",
"language" : "de",
"stateless" : "1",
"type_dm" : "stop",
"name_dm" : busStop,
"useRealtime" : "1",
"mode" : "direct",
"type_dm" : "any",
"itdLPxx_bcl" : "true"]);
"ptOptionActive" : "1",
"mergeDep" : "1",
"limit" : "20",
"deleteAssignedStops_dm" : "1"]);
// dfmt on
if (substitutionFileName.exists && substitutionFileName.isFile)
{
@ -44,6 +58,7 @@ void main(string[] args)
auto currentTime = Clock.currTime;
JSONValue j = ["time" : "%02s:%02s".format(currentTime.hour, currentTime.minute)];
j.object["departures"] = (cast(string) content.data).parsedFahrplan.array.JSONValue;
auto output = j.toPrettyString.replace("\\/", "/");
if (fileName !is null)
@ -57,5 +72,4 @@ void main(string[] args)
{
output.writeln;
}
}

View file

@ -1,12 +1,10 @@
module fahrplanparser;
import std.algorithm : filter, map;
import std.array : empty, front, replace;
import std.algorithm : map;
import std.array : front;
import std.conv : to;
import std.datetime : dur, TimeOfDay;
import std.regex : ctRegex, matchAll;
import std.string : strip;
import std.typecons : tuple, Tuple;
import std.datetime : dur, TimeOfDay, DateTimeException;
import std.string : format;
import kxml.xml : readDocument, XmlNode;
@ -14,14 +12,18 @@ import substitution;
private:
enum ScheduleHeadings
enum departureNodeName = "dp";
enum timeNodeName = "t";
enum realTimeNodeName = "rt";
enum departuresXPath = "/efa/dps/" ~ departureNodeName;
template timeXPath(string _timeNodeName = timeNodeName)
{
date,
departure,
line,
direction,
platform
enum timeXPath = "/st/" ~ _timeNodeName;
}
enum useRealTimeXPath = "/realtime";
enum lineXPath = "/m/nu";
enum directionXPath = "/m/des";
public:
@ -29,132 +31,182 @@ auto parsedFahrplan(in string data)
{
// dfmt off
return data.readDocument
.parseXPath(`//table[@id="departureMonitor"]/tbody/tr`)[1 .. $]
.getRowContents
.filter!(row => !row.empty)
.map!(a => ["departure" : a[0].parseTime[0].to!string[0 .. $ - 3],
"delay" : a[0].parseTime[1].total!"minutes".to!string,
"line" : a[1],
"direction" : a[2].substitute]);
.parseXPath(departuresXPath)
.map!(dp => ["departure" : "%02s:%02s".format(dp.departureTime.hour, dp.departureTime.minute),
"delay" : dp.delay.total!"minutes".to!string,
"line": dp.parseXPath(lineXPath).front.getCData,
"direction": dp.parseXPath(directionXPath).front.getCData.substitute]);
// dfmt on
}
private:
class BadTimeInputException : Exception
{
this(string msg) @safe pure nothrow @nogc
{
super(msg);
}
this() @safe pure nothrow @nogc
{
this("");
}
}
auto parseTime(in string input) @safe
{
auto matches = matchAll(input, ctRegex!(`(?P<hours>\d{1,2}):(?P<minutes>\d{2})`));
if (matches.empty)
throw new BadTimeInputException();
auto actualTime = TimeOfDay(matches.front["hours"].to!int, matches.front["minutes"].to!int);
matches.popFront;
if (!matches.empty)
{
auto expectedTime = TimeOfDay(matches.front["hours"].to!int,
matches.front["minutes"].to!int);
auto timeDiff = actualTime - expectedTime;
if (timeDiff < dur!"minutes"(0))
timeDiff = dur!"hours"(24) + timeDiff;
return tuple(expectedTime, timeDiff);
}
return tuple(actualTime, dur!"minutes"(0));
}
@safe unittest
{
import std.exception : assertThrown;
assertThrown(parseTime(""));
assertThrown(parseTime("lkeqf"));
assertThrown(parseTime(":"));
assertThrown(parseTime("00:0"));
assert("00:00".parseTime == tuple(TimeOfDay(0, 0), dur!"minutes"(0)));
assert("0:00".parseTime == tuple(TimeOfDay(0, 0), dur!"minutes"(0)));
assert("00:00 00:00".parseTime == tuple(TimeOfDay(0, 0), dur!"minutes"(0)));
assert("00:00 00:00 12:00".parseTime == tuple(TimeOfDay(0, 0), dur!"minutes"(0)));
assert("12:3412:34".parseTime == tuple(TimeOfDay(12, 34), dur!"minutes"(0)));
assert("ölqjfo12:34oieqf12:31ölqjf".parseTime == tuple(TimeOfDay(12, 31), dur!"minutes"(3)));
assert("17:53 (planmäßig 17:51 Uhr)".parseTime == tuple(TimeOfDay(17, 51), dur!"minutes"(2)));
assert("00:00 23:59".parseTime == tuple(TimeOfDay(23, 59), dur!"minutes"(1)));
}
auto getRowContents(XmlNode[] rows)
{
return rows.map!(x => getRowContent(x));
}
auto getRowContent(XmlNode row)
{
return row.parseXPath("//td")[ScheduleHeadings.departure .. ScheduleHeadings.direction + 1].map!(
cell => stripLinks(cell));
}
auto stripLinks(XmlNode cell)
{
auto links = cell.parseXPath("//a");
if (links.empty)
{
return cell.getCData;
}
else
{
return links.front.getCData.replace("...", "");
}
}
@system unittest
{
auto foo = new XmlNode("foo");
assert(foo.stripLinks == "");
import std.stdio;
import std.array: array;
auto xml = "";
assert(xml.parsedFahrplan.array == []);
auto link = new XmlNode("a");
link.setCData("test");
foo.addChild(link);
assert(foo.stripLinks == "test");
xml = "<efa><dps></dps></efa>";
assert(xml.parsedFahrplan.array == []);
link.setCData("test2...");
assert(foo.stripLinks == "test2");
xml = "<efa><dps><dp><realtime>1</realtime><st><t>1224</t><rt>1242</rt></st><m><nu>6</nu><des>Wernerwerkstraße</des></m></dp></dps></efa>";
assert(xml.parsedFahrplan.array == [["direction":"Wernerwerkstraße", "line":"6", "departure":"12:24", "delay":"18"]]);
auto bar = new XmlNode("bar");
bar.setCData("test3");
assert(bar.stripLinks == "test3");
xml = "<efa><dps><dp><realtime>0</realtime><st><t>1224</t></st><m><nu>6</nu><des>Wernerwerkstraße</des></m></dp></dps></efa>";
assert(xml.parsedFahrplan.array == [["direction":"Wernerwerkstraße", "line":"6", "departure":"12:24", "delay":"0"]]);
bar.addChild(link);
assert(bar.stripLinks == "test2");
auto baz = new XmlNode("baz");
auto subNode = new XmlNode("subNode");
baz.addChild(subNode);
assert(baz.stripLinks == "");
baz.addChild(link);
assert(baz.stripLinks == "test2");
baz.addCData("test4");
assert(baz.stripLinks == "test2");
baz.removeChild(link);
assert(baz.stripLinks == "test4");
xml = "<efa><dps><dp><realtime>0</realtime><st><t>1224</t></st><m><nu>6</nu><des>Wernerwerkstraße</des></m></dp><dp><realtime>1</realtime><st><t>1353</t><rt>1356</rt></st><m><nu>11</nu><des>Burgweinting</des></m></dp></dps></efa>";
assert(xml.parsedFahrplan.array == [["direction":"Wernerwerkstraße", "line":"6", "departure":"12:24", "delay":"0"], ["direction":"Burgweinting", "line":"11", "departure":"13:53", "delay":"3"]]);
}
private:
class UnexpectedValueException(T) : Exception
{
this(T t, string node) @safe pure
{
super(`Unexpected value "%s" for node "%s"`.format(t, node));
}
}
auto departureTime(string _timeNodeName = timeNodeName)(XmlNode dp)
in
{
assert(dp.getName == departureNodeName);
}
body
{
return TimeOfDay.fromISOString(dp.parseXPath(timeXPath!_timeNodeName).front.getCData ~ "00");
}
@system unittest
{
import std.exception : assertThrown;
auto xml = "<dp><st><t>0000</t></st></dp>".readDocument.parseXPath("/dp").front;
assert(xml.departureTime == TimeOfDay(0, 0));
xml = "<dp><st><t>0013</t></st></dp>".readDocument.parseXPath("/dp").front;
assert(xml.departureTime == TimeOfDay(0, 13));
xml = "<dp><st><t>1100</t></st></dp>".readDocument.parseXPath("/dp").front;
assert(xml.departureTime == TimeOfDay(11, 00));
xml = "<dp><st><t>1242</t></st></dp>".readDocument.parseXPath("/dp").front;
assert(xml.departureTime == TimeOfDay(12, 42));
xml = "<dp><st><t>2359</t></st></dp>".readDocument.parseXPath("/dp").front;
assert(xml.departureTime == TimeOfDay(23, 59));
assertThrown!DateTimeException("<dp><st><t>2400</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>0061</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>2567</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t></t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>0</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>00</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>000000</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>00:00</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
assertThrown!DateTimeException("<dp><st><t>abcd</t></st></dp>".readDocument.parseXPath("/dp")
.front.departureTime);
}
auto delay(XmlNode dp)
in
{
assert(dp.getName == departureNodeName);
}
body
{
auto useRealtimeString = dp.parseXPath(useRealTimeXPath).front.getCData;
if (useRealtimeString == "0")
return dur!"minutes"(0);
else if (useRealtimeString == "1")
{
auto expectedTime = dp.departureTime;
auto realTime = dp.departureTime!realTimeNodeName;
auto timeDiff = realTime - expectedTime;
if (timeDiff < dur!"minutes"(0))
timeDiff = dur!"hours"(24) + timeDiff;
return timeDiff;
}
else
throw new UnexpectedValueException!string(useRealtimeString, "realtime");
}
@system unittest
{
import std.exception : assertThrown;
import core.exception : AssertError;
auto xml = "<dp><realtime>0</realtime></dp>".readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(0));
xml = "<dp><realtime></realtime></dp>".readDocument.parseXPath("/dp").front;
assertThrown!(UnexpectedValueException!string)(xml.delay);
xml = "<dp><realtime>2</realtime></dp>".readDocument.parseXPath("/dp").front;
assertThrown!(UnexpectedValueException!string)(xml.delay);
xml = "<dp><realtime>a</realtime></dp>".readDocument.parseXPath("/dp").front;
assertThrown!(UnexpectedValueException!string)(xml.delay);
xml = "<dp><realtime>1</realtime></dp>".readDocument.parseXPath("/dp").front;
assertThrown!AssertError(xml.delay);
xml = "<dp><realtime>1</realtime><st><t></t></st></dp>".readDocument.parseXPath("/dp").front;
assertThrown!DateTimeException(xml.delay);
xml = "<dp><realtime>1</realtime><st><rt></rt></st></dp>".readDocument.parseXPath("/dp").front;
assertThrown!AssertError(xml.delay);
xml = "<dp><st><rt></rt><t></t></st></dp>".readDocument.parseXPath("/dp").front;
assertThrown!AssertError(xml.delay);
xml = "<dp><realtime>1</realtime><st><rt></rt><t></t></st></dp>".readDocument.parseXPath("/dp")
.front;
assertThrown!DateTimeException(xml.delay);
xml = "<dp><realtime>1</realtime><st><rt>0000</rt><t></t></st></dp>".readDocument.parseXPath("/dp")
.front;
assertThrown!DateTimeException(xml.delay);
xml = "<dp><realtime>1</realtime><st><rt></rt><t>0000</t></st></dp>".readDocument.parseXPath("/dp")
.front;
assertThrown!DateTimeException(xml.delay);
xml = "<dp><realtime>1</realtime><st><rt>0000</rt><t>0000</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(0));
xml = "<dp><realtime>1</realtime><st><rt>0001</rt><t>0000</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(1));
xml = "<dp><realtime>1</realtime><st><rt>1753</rt><t>1751</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(2));
xml = "<dp><realtime>1</realtime><st><rt>1010</rt><t>1000</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(10));
xml = "<dp><realtime>1</realtime><st><rt>1301</rt><t>1242</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(19));
xml = "<dp><realtime>1</realtime><st><rt>0000</rt><t>1242</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(678));
xml = "<dp><realtime>1</realtime><st><rt>0000</rt><t>2359</t></st></dp>"
.readDocument.parseXPath("/dp").front;
assert(xml.delay == dur!"minutes"(1));
}