InTransit->add: dbris: ensure that train_no is always set2.11.23

1 files changed, 5 insertions, 3 deletions

diff --git a/lib/Travelynx/Model/InTransit.pm b/lib/Travelynx/Model/InTransit.pm
index 9f9dd9b..0795179 100644
--- a/lib/Travelynx/Model/InTransit.pm
+++ b/lib/Travelynx/Model/InTransit.pm
@@ -196,11 +196,13 @@ sub add {
 	elsif ( $journey and $stop ) {
 
 		# DBRIS
+		my $number = $journey->train_no // $journey->number // $train_suffix;
+
 		my $line;
-		if ( $journey->line_no ne $journey->train_no ) {
+		if ( defined $journey->line_no and $journey->line_no ne $number ) {
 			$line = $journey->line_no;
 		}
-		elsif ( $train_suffix ne $journey->train_no ) {
+		elsif ( defined $train_suffix and $train_suffix ne $number ) {
 			$line = $train_suffix;
 		}
 
@@ -254,7 +256,7 @@ sub add {
 				dep_platform => $stop->platform,
 				train_type   => $journey->type // q{},
 				train_line   => $line,
-				train_no     => $journey->train_no,
+				train_no     => $number,
 				train_id     => $data->{trip_id},
 				sched_departure => $stop->sched_dep,
 				real_departure  => $stop->rt_dep // $stop->sched_dep,