File Coverage

blib/lib/Python/Serialize/Pickle/InlinePython.pm

Criterion	Covered	Total	%
statement	7	9	77.7
branch			n/a
condition			n/a
subroutine	3	3	100.0
pod			n/a
total	10	12	83.3

package Python::Serialize::Pickle::InlinePython;
use strict;
use warnings;
our $VERSION = '0.01';

use Inline Python => <<'...';
from pickle import Pickler, Unpickler

def __pickler(fname):
    return Pickler(open(fname, 'wb'))

def __unpickler(fname):
    return Unpickler(open(fname, 'rb'))

def __load(unp):
    try:
        return unp.load()
    except EOFError:
        return "EOFErrorEOFErrorEOFErrorEOFError" # wtf?
...

sub new {
    my ($class, $fname) = @_;
    if ($fname =~ s/^>//g) {
        bless { pickler => __pickler($fname) }, 'Python::Serialize::Pickle::InlinePython';
    } else {
        bless { unpickler => __unpickler($fname) }, 'Python::Serialize::Pickle::InlinePython';
    }
}

sub dump {
    my ($self, $stuff) = @_;
    die "this is not a writer-obj" unless $self->{pickler};
    $self->{pickler}->dump($stuff);
}

sub load {
    my $self = shift;
    die "this is not a reader-obj" unless $self->{unpickler};
    my $ret = __load($self->{unpickler});
    if ($ret eq 'EOFErrorEOFErrorEOFErrorEOFError') {
        return undef; ## no critic
    } else {
        return $ret;
    }
}

sub close {
    my $self = shift;
    delete $self->{$_} for keys %$self;
}

1;
__END__