diff options
| -rwxr-xr-x | bin/iris-delay-stats | 67 |
1 files changed, 60 insertions, 7 deletions
diff --git a/bin/iris-delay-stats b/bin/iris-delay-stats index 32c38a8..37cce8a 100755 --- a/bin/iris-delay-stats +++ b/bin/iris-delay-stats @@ -4,19 +4,72 @@ use strict; use warnings; use 5.010; +use DBI; use List::Util qw(first); +use Travel::Status::DE::IRIS; our $VERSION = '0.00'; -sub table_exists { - my $db = shift; - my $table = shift; - my @tables = $db->tables('','','','TABLE'); +my $station = shift or die("Usage: $0 <station> <database>\n"); +my $dbname = shift or die("Usage: $0 <station> <database>\n"); + +my $first_run = not -e $dbname; + +my $dbh = DBI->connect("dbi:SQLite:dbname=$dbname", q{}, q{}); + +if ($first_run) { + my $query = qq{ + create table departures ( + train_id int not null, + station char(10) not null, + raw_id char(64) not null primary key, + scheduled_time int not null, + delay int, + is_canceled boolean, + destination char(64) not null, + train_type char(6) not null, + train_no int not null, + line_no int, + platform char(16), + } . join(', ', map { "msg_$_ boolean" } (1..99)) . ')'; + $dbh->do($query); +} - if (first { $_ eq $table } @tables) { - return 1; +my $status = Travel::Status::DE::IRIS->new( + station => $station, + lookahead => 60); + +my @fields = (qw(train_id station raw_id scheduled_time delay + is_canceled destination train_type train_no line_no platform)); +push(@fields, map { "msg_$_" } (1..99)); +my $fieldlist = join(', ', @fields); +my $field_placeholders = join(', ', ('?') x @fields); +my $insert_query = qq{ + insert or replace into departures ( $fieldlist ) values ( $field_placeholders ) +}; +my $sth = $dbh->prepare($insert_query); + +for my $r ($status->results) { + + my @msgtypes = (0) x 99; + for my $m ($r->raw_messages) { + $msgtypes[ $m->[1] ] = 1; } - return 0; + + $sth->execute( + $r->train_id, + $station, + $r->raw_id, + $r->datetime->epoch, + $r->delay, + $r->is_cancelled, + $r->destination, + $r->type, + $r->train_no, + $r->line, + $r->sched_platform, + @msgtypes + ); } __END__ |