add dumpstops command

This commit is contained in:
Birte Kristina Friesel 2024-02-09 19:09:16 +01:00
parent fe0a873899
commit 39ebc84e5f
No known key found for this signature in database
GPG key ID: 19E6E524EBB177BA
2 changed files with 56 additions and 0 deletions

View 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.

View file

@ -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;