tools/picture-upload.pl

   1 #!/usr/bin/perl
   2
   3 use File::Temp;
   4 use File::Copy;
   5 use CGI;
   6 use C4::Context;
   7 use C4::Auth;
   8 use C4::Output;
   9
  10 #my $destdir = "/usr/local/koha/intranet/htdocs/intranet-tmpl/images/patronpictures";
  11 #my $uploadfile = shift @ARGV;
  12 my $input = new CGI;
  13 my $destdir = C4::Context->config('intrahtdocs') . "/patronimages";
  14
  15 warn "DEST : $destdir";
  16 my ($template, $loggedinuser, $cookie)
  17         = get_template_and_user({template_name => "tools/picture-upload.tmpl",
  18                                         query => $input,
  19                                         type => "intranet",
  20                                         authnotrequired => 0,
  21                                         flagsrequired => {management => 1, tools => 1},
  22                                         debug => 0,
  23                                         });
  24
  25 my $uploadfilename = $input->param( 'uploadfile' );
  26 my $uploadfile = $input->upload( 'uploadfile' );
  27 my ( $total, $handled, @counts );
  28
  29 if ( $uploadfile ) {
  30     my $dirname = File::Temp::tempdir( CLEANUP => 1);
  31     my ( $tfh, $tempfile ) = File::Temp::tempfile( SUFFIX => '.zip', UNLINK => 1 );
  32     my ( @directories, %errors );
  33
  34     $errors{'NOTZIP'} = 1 unless ( $uploadfilename =~ /\.zip$/i );
  35     $errors{'NOWRITETEMP'} = 1 unless ( -w "$dirname" );
  36     $errors{'NOWRITEDEST'} = 1 unless ( -w "$destdir" );
  37     $errors{'EMPTYUPLOAD'} = 1 unless ( length( $uploadfile ) > 0 );
  38
  39     if ( %errors ) {
  40         $template->param( ERRORS => [ \%errors ] );
  41     } else {
  42         while ( <$uploadfile> ) {
  43             print $tfh $_;
  44         }
  45
  46         close $tfh;
  47
  48         `unzip $tempfile -d $dirname`;
  49
  50         push @directories, "$dirname";
  51         foreach $recursive_dir ( @directories ) {
  52             opendir $dir, $recursive_dir;
  53             while ( my $entry = readdir $dir ) {
  54                 push @directories, "$recursive_dir/$entry" if ( -d "$recursive_dir/$entry" and $entry !~ /^\./ );
  55             }
  56             closedir $dir;
  57         }
  58
  59         foreach my $dir ( @directories ) {
  60             $handled += handle_dir( $dir );
  61         }
  62
  63         $total = scalar @directories;
  64
  65         $template->param(
  66                          TOTAL => $total,
  67                          HANDLED => $handled,
  68                          COUNTS => \@counts,
  69                          );
  70     }
  71 }
  72
  73 output_html_with_http_headers $input, $cookie, $template->output;
  74
  75
  76 sub handle_dir {
  77     my ( $dir ) = @_;
  78     my ( %count );
  79     $count{filenames} = ();
  80
  81     return 0 unless ( -r "$dir/IDLINK.TXT" or -r "$dir/DATALINK.TXT" );
  82
  83     my $file = ( -r "$dir/IDLINK.TXT" ) ? "$dir/IDLINK.TXT" : "$dir/DATALINK.TXT";
  84
  85     open $fh, $file or { print "Openning $dir/$filename failed!\n" and return 0 };
  86
  87     while (my $line = <$fh>) {
  88         chomp $line;
  89
  90         my ( $filename, $cardnumber );
  91
  92         my $delim = ($line =~ /\t/) ? "\t" : ",";
  93
  94         ($cardnumber, $filename) = split $delim, $line;
  95         $cardnumber =~ s/[\"\r\n]//g;  # remove offensive characters
  96         $filename =~ s/[\"\r\n]//g;
  97
  98         if ($cardnumber && $filename) {
  99             my $result = move ( "$dir/$filename", "$destdir/$cardnumber.jpg" );
 100             if ( $result ) {
 101                 $count{count}++;
 102                 push @{ $count{filenames} }, { source => $filename, dest => $cardnumber .".jpg" };
 103             }
 104         }
 105     }
 106     $count{source} = $dir;
 107     $count{dest} = $destdir;
 108     push @counts, \%count;
 109
 110     close $fh;
 111
 112     return 1;
 113 }