File Coverage

blib/lib/SPOPS/Import/DBI/GenericOperation.pm

Criterion	Covered	Total	%
statement	6	6	100.0
branch			n/a
condition			n/a
subroutine	2	2	100.0
pod			n/a
total	8	8	100.0

package SPOPS::Import::DBI::GenericOperation;

# $Id: GenericOperation.pm,v 1.2 2004/06/02 00:33:20 lachoy Exp $

use strict;
use base qw( SPOPS::Import );
use SPOPS::Exception qw( spops_error );
use SPOPS::SQLInterface;

$SPOPS::Import::DBI::GenericOperation::VERSION  = sprintf("%d.%02d", q$Revision: 1.2 $ =~ /(\d+)\.(\d+)/);

my @FIELDS = qw( table db where );
SPOPS::Import::DBI::GenericOperation->mk_accessors( @FIELDS );

sub add_where_params {
    my ( $self, @args ) = @_;
    push @{ $self->{where_params} }, @args;
}

sub where_params {
    my ( $self, $all_args ) = @_;
    if ( $all_args ) {
        $self->{where_params} = ( ref $all_args )
                                  ? $all_args : [ $all_args ];
    }
    return $self->{where_params};
}

########################################
# Core API

sub get_fields { return ( $_[0]->SUPER::get_fields(), @FIELDS ) }

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

    unless ( $self->db )     { spops_error "Cannot run without a database handle available" }
    unless ( $self->table )  { spops_error "Cannot run without table defined" }
    unless ( $self->where )  { spops_error "Cannot run without where clause defined" }

    my %op_args = (
        db    => $self->db,
        table => $self->table,
        where => $self->where,
        value => $self->where_params,
    );
    my $rv = eval { $self->_run_operation( \%op_args ) };
    return ( $@ ) ? [ [ undef, $rv, $@ ] ]
                  : [ [ 1, $rv, undef ] ];
}

sub _run_operation {
    my ( $self ) = @_;
    die "'", ref( $self ), "' does not implement the required ",
        "method '_run_operation()'";
}

########################################
# For subclasses...

sub data_from_file {
    my ( $self, $filename ) = @_;
    $self->assign_data( $self->raw_data_from_file( $filename ) );
}


sub data_from_fh {
    my ( $self, $fh ) = @_;
    $self->assign_data( $self->raw_data_from_fh( $fh ) );
}


sub assign_raw_data {
    my ( $self, $metadata ) = @_;
    for ( qw( where table where_params ) ) {
        $self->$_( $metadata->{$_} );
        delete $metadata->{ $_ };
    }
    $self->extra_metadata( $metadata );
    return $self;
}

1;

__END__