support UIC codes

3 files changed, 6 insertions, 0 deletions

diff --git a/Changelog b/Changelog
index 4d4c20c..6110b96 100644
--- a/Changelog
+++ b/Changelog
@@ -1,6 +1,7 @@
 git HEAD
 
     * Result: Add qos code 79 ("Ohne Mehrzweckabteil")
+    * db-iris: Accept UIC station codes
 
 Travel::Status::DE::IRIS 1.03 - Tue Sep 15 2015
 
diff --git a/bin/db-iris b/bin/db-iris
index 2aa94b7..ecbf376 100755
--- a/bin/db-iris
+++ b/bin/db-iris
@@ -141,6 +141,10 @@ sub get_departure {
 sub get_station {
 	my ($input_name) = @_;
 
+	if ( $input_name =~ m{ ^ [[:digit:]]+ $ }x ) {
+		return $input_name;
+	}
+
 	my @stations = Travel::Status::DE::IRIS::Stations::get_station($input_name);
 
 	if ( @stations == 0 ) {
diff --git a/lib/Travel/Status/DE/IRIS.pm b/lib/Travel/Status/DE/IRIS.pm
index 984c209..f9300f0 100644
--- a/lib/Travel/Status/DE/IRIS.pm
+++ b/lib/Travel/Status/DE/IRIS.pm
@@ -67,6 +67,7 @@ sub new {
 
 	$self->{station}      = $opt{station};
 	$self->{station_code} = $station_node->getAttribute('eva');
+	$self->{station_name} = $station_node->getAttribute('name');
 
 	my $dt_req = $self->{datetime}->clone;
 	for ( 1 .. 3 ) {