Changeset 1985

Timestamp:

04/20/17 18:30:00 (7 years ago)

Author:

nanardon

Message:

Add tools to create multiple object from csv file

Location:

trunk/LATMOS-Accounts

Files:

• MANIFEST (modified) (1 diff)
• Makefile.PL (modified) (1 diff)
• lib/LATMOS/Accounts/Bases/OCHelper.pm (modified) (1 diff)
• lib/LATMOS/Accounts/Utils.pm (modified) (3 diffs)
• t/05_utils.t (modified) (3 diffs)

trunk/LATMOS-Accounts/MANIFEST

@@ -53 +53 @@
 bin/la-warn-expire
 bin/la2xls
+bin/la-load-csv
 etc/cron.d/latmos-accounts
 etc/init.d/la-sync-manager

trunk/LATMOS-Accounts/Makefile.PL

@@ -83 +83 @@
         bin/la-ban-passwd
         bin/la-renewEmp
+        bin/la-load-csv
         ) ],
         macro => {

trunk/LATMOS-Accounts/lib/LATMOS/Accounts/Bases/OCHelper.pm

@@ -112 +112 @@
 }
 
+=head2 $ochelper->Automate($info)
+
+Try to create object from C<$info> w/o interacting with user.
+If given infomation does not allow to create object, it failed.
+
+Return 1 on success.
+
+=cut
+
+sub Automate {
+    my ($self, $info) = @_;
+
+    for (my $count = 0; $count < 3; $count++) {
+        my $status;
+        ($status, $info) = $self->step($info);
+        if ($status eq 'CREATED') {
+            return 1;
+        } elsif ($status eq 'ERROR') {
+            return;
+        }
+    }
+}
+
 1;
 

trunk/LATMOS-Accounts/lib/LATMOS/Accounts/Utils.pm

@@ -9 +9 @@
 use File::Temp qw(tempfile);
 use Crypt::Cracklib;
+use Text::CSV;
 
 our $VERSION = (q$Rev$ =~ /^Rev: (\d+) /)[0];
@@ -21 +22 @@
 
 @ISA = qw(Exporter);
-@EXPORT = qw(to_ascii exec_command switch_user run_via_sudo buildLogin yesno);
-@EXPORT_OK = qw(to_ascii exec_command switch_user run_via_sudo buildLogin yesno);
+@EXPORT = qw(to_ascii exec_command switch_user run_via_sudo buildLogin loadCSV yesno);
+@EXPORT_OK = qw(to_ascii exec_command switch_user run_via_sudo buildLogin loadCSV yesno);
 
 =head2 to_ascii($text)
@@ -175 +176 @@
 }
 
+=head2 loadCSV($fh, $callback, $initcallback)
+
+Parse CVS files and return an array for each parsed line.
+
+If defined call C<$callback> for parsed line.
+
+C<$initcallback>, if defined, is called just after the first line.
+
+=cut
+
+sub loadCSV {
+    my ($fh, $cb, $initcb) = @_;
+
+    my $csv = Text::CSV->new({
+            blank_is_undef => 1,
+            binary => 1,
+    });
+
+    binmode($fh, ":encoding(UTF-8)");
+
+    # First line contains attribute
+    my $columns = $csv->getline( $fh );
+
+    $csv->column_names($columns);
+
+    if ($initcb) {
+        $initcb->($csv);
+    }
+
+    my $all = [];
+    my $linecount = 1;
+    while ( my $row = $csv->getline_hr( $fh ) ) {
+        $linecount++;
+        if ($cb) {
+            if (! $cb->($row, $linecount)) {
+                return;
+            }
+        }
+        push(@{ $all }, $row);
+    }
+    $csv->eof () or do {
+        return;
+    };
+
+    return $all;
+}
+
 =head2 check_oid_validity($name)
 

trunk/LATMOS-Accounts/t/05_utils.t

@@ -1 +1 @@
 use strict;
 use warnings;
-use Test::More tests => 30;
+use Test::More tests => 32;
 use File::Temp qw(mkstemp);
 
@@ -10 +10 @@
 ok(la_log(LA_NOTICE, "a notice"), "can run la_log");
 
+{
 my ($fh, $file) = mkstemp( "tmpfileXXXXX" );
 print $fh <<EOF;
@@ -32 +33 @@
 ok(LATMOS::Accounts::Utils::dump_read_temp_file(
         sub { 1; }, sub { 1; }), "Can edit temp file");
+
+}
+{
+my ($fh, $file) = mkstemp( "tmpfileXXXXX" );
+binmode($fh, ":encoding(UTF-8)");
+print $fh <<EOF;
+sn,givenName
+Nom,Prénom
+EOF
+
+seek($fh, 0, 0);
+
+my $res = LATMOS::Accounts::Utils::loadCSV($fh);
+$fh = undef;
+unlink($file);
+
+# now testing
+is(scalar(@{ $res }), 1, "loadCSV return same number result");
+is($res->[0]{givenName}, "Prénom", "Attribute givenName is properly return");
+
+}
 
 ok(!LATMOS::Accounts::Utils::check_oid_validity('toto'), '"toto" is valid oid');