File Coverage

blib/lib/Text/CSV/UniqueColumns.pm

Criterion	Covered	Total	%
statement	9	79	11.3
branch	0	12	0.0
condition	0	4	0.0
subroutine	3	10	30.0
pod	0	7	0.0
total	12	112	10.7

package Text::CSV::UniqueColumns;

use 5.008008;
use strict;
use warnings;

require Exporter;

our @ISA = qw(Exporter);


our $VERSION = '0.3';

my (%headerHash, @cleanup);

sub new {
    my ($class) = shift;
    my ($sFile) = shift || die "Please provide csv file as argument\n";
    my ($sCols) = shift;
    die "$sFile not found" if (!-e $sFile);

    my $self = {
        '_file' => $sFile,
        '_cols' => $sCols,
        '_headers' => ""
        };

    bless $self, $class;

    getHeaders($self,$sFile);
    die "Could not get headers\n" if (!$self->{'_headers'});

   return $self;
}

sub checkUniq {
    my ($self) = shift;
    my $sCols = shift || return "Provide cols as arguments\n";
    $self->{'_cols'} = $sCols;
    my $iCount = 1;
    my $sPasteOutput;
    my $sPasteFiles = " ";
    my @CompositeCols = split(',',$sCols);
    foreach my $sCol (@CompositeCols) {
        chomp($sCol);
        if ($headerHash{$sCol}) {
            $headerHash{"Composite$iCount"} = `cut -f$headerHash{$sCol} -d , $self->{'_file'}`;
            open (FILE , ">Composite$iCount") or return "Cannot write Composite$iCount $! \n";
            push (@cleanup, "Composite$iCount");
            print FILE $headerHash{"Composite$iCount"};
            $sPasteFiles .= " Composite$iCount";
            close(FILE);
            $iCount++;
        }
        else {
            return  "Column - $sCol not found\n INFO - Use \"-l\" option to list columns in file\n";
        }
    }
    my $sCmd = "paste  -d , $sPasteFiles > pasteOutput ";
    push (@cleanup, 'pasteOutput');
    $sPasteOutput = `$sCmd`;
    my $iCount1 =  `cat pasteOutput | sed s/' '//g | wc -l`;
    my $iCount2 =  `cat pasteOutput | sed s/' '//g | sort | uniq |  wc -l`;
    cleanUp();
    if ($iCount1 == $iCount2) {
       return "1"; #unique
    }
    else {
       return "0";
    }
}

sub getColumnList {
    my ($self) = @_;
    foreach  (@{$self->{'_headers'}}) {
        return join(',', @{$self->{'_headers'}});
        print "$_\n";
    }
}


sub getUniqCols {
    my ($self) = @_;
    my ($sUniqCols) = " ";
    foreach my $iNo ( 0 .. (scalar(@{$self->{'_headers'}}) - 1)) {
        my $iField = $iNo + 1;
        my $sCmd = "cut -f$iField -d , $self->{'_file'} | sed s/' '//g | wc -l;";
        $sCmd .= "cut -f$iField -d ,  $self->{'_file'} | sed s/' '//g | sort | uniq | wc -l";
        my ($iCount1, $iCount2) = split("\n",`$sCmd`);

        if ( $iCount1 == $iCount2) {
            $sUniqCols .=  $self->{'_headers'}->[$iNo].",";
        }
        else {
            next;
        }
    }
    chop ($sUniqCols);
    return $sUniqCols;

}

sub buildHeaderHash {
    my ($self) = @_;
    my $iColNo = 1;
    foreach my $sCol (@{$self->{'_headers'}}) {
        $sCol =~ s/\s+//g;
        $sCol =~ s/\n//g;
        $headerHash{$sCol} = $iColNo;
        $iColNo++;
    }
}


sub getHeaders {
    my ($self,$sFile) = @_;
    print "file is $sFile \n";
    my @headers = split(',', `head -1 $sFile`);
    $self->{'_headers'} = \@headers;
    buildHeaderHash($self);
}

sub cleanUp {
    foreach my $sFile (@cleanup){
        `rm -f $sFile`;
    }
}


1;



__END__