Add --first-class option

2 files changed, 4 insertions, 2 deletions

diff --git a/bin/dbris b/bin/dbris
index d131378..6cc6ee9 100755
--- a/bin/dbris
+++ b/bin/dbris
@@ -16,7 +16,7 @@ use Travel::Routing::DE::DBRIS;
 
 my ( $date, $time, $from, $to, $language );
 my $mots;
-my $discounts;
+my ( $first_class, $discounts );
 my $developer_mode;
 my ( $json_output, $raw_json_output );
 my $use_cache = 1;
@@ -38,6 +38,7 @@ GetOptions(
 	'D|discounts=s'        => \$discounts,
 	'h|help'               => sub { show_help(0) },
 	'f|full-route'         => \$show_full_route,
+	'first-class!'         => \$first_class,
 	'm|modes-of-transit=s' => \$mots,
 	'l|language=s'         => \$language,
 	't|time=s'             => \$time,
@@ -115,6 +116,7 @@ my %opt = (
 	to             => get_stop( $to_raw,   0 ),
 	via            => [ map { get_stop( $_, 1 ) } @via_raw ],
 	language       => $language,
+	first_class    => $first_class,
 	cache          => $cache,
 	developer_mode => $developer_mode,
 );
diff --git a/lib/Travel/Routing/DE/DBRIS.pm b/lib/Travel/Routing/DE/DBRIS.pm
index 6f57dcf..8bf6bbe 100644
--- a/lib/Travel/Routing/DE/DBRIS.pm
+++ b/lib/Travel/Routing/DE/DBRIS.pm
@@ -63,7 +63,7 @@ sub new {
 		ankunftsHalt     => $conf{to}->id,
 		anfrageZeitpunkt => $dt->strftime('%Y-%m-%dT%H:%M:00'),
 		ankunftSuche     => 'ABFAHRT',
-		klasse           => 'KLASSE_2',
+		klasse           => $conf{first_class} ? 'KLASSE_1' : 'KLASSE_2',
 		produktgattungen => \@mots,
 		reisende         => [
 			{