Unverified Commit 39ebc84e authored by Birte Kristina Friesel's avatar Birte Kristina Friesel
Browse files

add dumpstops command

parent fe0a8738
Loading
Loading
Loading
Loading
+50 −0
Original line number Diff line number Diff line
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.
+6 −0
Original line number Diff line number Diff line
@@ -71,6 +71,12 @@ sub add_meta {
	}
}

sub get_db_iterator {
	my ($self) = @_;

	return $self->{pg}->db->select( 'stations', '*' );
}

sub get_meta {
	my ( $self, %opt ) = @_;
	my $db  = $opt{db} // $self->{pg}->db;