mirror of
https://github.com/derf/travelynx
synced 2024-12-05 10:49:13 +00:00
add dumpstops command
This commit is contained in:
parent
fe0a873899
commit
39ebc84e5f
2 changed files with 56 additions and 0 deletions
50
lib/Travelynx/Command/dumpstops.pm
Normal file
50
lib/Travelynx/Command/dumpstops.pm
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
package Travelynx::Command::dumpstops;
|
||||||
|
|
||||||
|
# Copyright (C) 2024 Birte Kristina Friesel
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
|
||||||
|
use Mojo::Base 'Mojolicious::Command';
|
||||||
|
use List::Util qw();
|
||||||
|
use Text::CSV;
|
||||||
|
|
||||||
|
has description => 'Export HAFAS/IRIS stops to CSV';
|
||||||
|
|
||||||
|
has usage => sub { shift->extract_usage };
|
||||||
|
|
||||||
|
sub run {
|
||||||
|
my ( $self, $command, $filename ) = @_;
|
||||||
|
my $db = $self->app->pg->db;
|
||||||
|
|
||||||
|
if ( not $command or not $filename ) {
|
||||||
|
$self->help;
|
||||||
|
}
|
||||||
|
elsif ( $command eq 'csv' ) {
|
||||||
|
open( my $fh, '>', $filename ) or die("open($filename): $!\n");
|
||||||
|
|
||||||
|
my $csv = Text::CSV->new( { eol => "\r\n" } );
|
||||||
|
$csv->combine(qw(name eva lat lon source archived));
|
||||||
|
print $fh $csv->string;
|
||||||
|
|
||||||
|
my $iter = $self->app->stations->get_db_iterator;
|
||||||
|
while ( my $row = $iter->hash ) {
|
||||||
|
$csv->combine( @{$row}{qw{name eva lat lon source archived}} );
|
||||||
|
print $fh $csv->string;
|
||||||
|
}
|
||||||
|
close($fh);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
$self->help;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
1;
|
||||||
|
|
||||||
|
__END__
|
||||||
|
|
||||||
|
=head1 SYNOPSIS
|
||||||
|
|
||||||
|
Usage: index.pl dumpstops <format> <filename>
|
||||||
|
|
||||||
|
Exports known stops to <filename>.
|
||||||
|
Right now, only the "csv" format is supported.
|
|
@ -71,6 +71,12 @@ sub add_meta {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
sub get_db_iterator {
|
||||||
|
my ($self) = @_;
|
||||||
|
|
||||||
|
return $self->{pg}->db->select( 'stations', '*' );
|
||||||
|
}
|
||||||
|
|
||||||
sub get_meta {
|
sub get_meta {
|
||||||
my ( $self, %opt ) = @_;
|
my ( $self, %opt ) = @_;
|
||||||
my $db = $opt{db} // $self->{pg}->db;
|
my $db = $opt{db} // $self->{pg}->db;
|
||||||
|
|
Loading…
Reference in a new issue