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 unless (-d $destdir) {
  26         $errors{'NODIR'} = 1;
  27         warn "patronimages directory not present";
  28 }
  29 if ( %errors ) {
  30     $template->param( ERRORS => [ \%errors ] );
  31 }
  32 my $uploadfilename = $input->param( 'uploadfile' );
  33 my $uploadfile = $input->upload( 'uploadfile' );
  34 my ( $total, $handled, @counts );
  35
  36 if ( $uploadfile ) {
  37     my $dirname = File::Temp::tempdir( CLEANUP => 1);
  38     my ( $tfh, $tempfile ) = File::Temp::tempfile( SUFFIX => '.zip', UNLINK => 1 );
  39     my ( @directories, %errors );
  40
  41     $errors{'NOTZIP'} = 1 unless ( $uploadfilename =~ /\.zip$/i );
  42     $errors{'NOWRITETEMP'} = 1 unless ( -w $dirname );
  43     $errors{'NOWRITEDEST'} = 1 unless ( -w $destdir );
  44     $errors{'EMPTYUPLOAD'} = 1 unless ( length( $uploadfile ) > 0 );
  45
  46     if ( %errors ) {
  47         $template->param( ERRORS => [ \%errors ] );
  48     } else {
  49         while ( <$uploadfile> ) {
  50             print $tfh $_;
  51         }
  52
  53         close $tfh;
  54
  55         system("unzip $tempfile -d $dirname");
  56
  57         push @directories, "$dirname";
  58         foreach $recursive_dir ( @directories ) {
  59             opendir $dir, $recursive_dir;
  60             while ( my $entry = readdir $dir ) {
  61                         push @directories, "$recursive_dir/$entry" if ( -d "$recursive_dir/$entry" and $entry !~ /^\./ );
  62             }
  63             closedir $dir;
  64         }
  65
  66         foreach my $dir ( @directories ) {
  67             $handled += handle_dir( $dir );
  68         }
  69
  70         $total = scalar @directories;
  71
  72         $template->param(
  73                          TOTAL => $total,
  74                          HANDLED => $handled,
  75                          COUNTS => \@counts,
  76                          );
  77     }
  78 }
  79
  80 output_html_with_http_headers $input, $cookie, $template->output;
  81
  82 sub handle_dir {
  83     my ( $dir ) = @_;
  84     my ( %count );
  85     $count{filenames} = ();
  86
  87     my $file = ( -r "$dir/IDLINK.TXT" ) ? "$dir/IDLINK.TXT" : "$dir/DATALINK.TXT";
  88     unless (open (FILE, $file)) {
  89                 print "Openning $dir/$file failed!\n";
  90                 return 0;
  91         };
  92
  93     while (my $line = <FILE>) {
  94                 chomp $line;
  95                 my ( $filename, $cardnumber );
  96                 my $delim = ($line =~ /\t/) ? "\t" : ",";
  97                 ($cardnumber, $filename) = split $delim, $line;
  98                 $cardnumber =~ s/[\"\r\n]//g;  # remove offensive characters
  99                 $filename   =~ s/[\"\r\n]//g;
 100
 101                 if ($cardnumber && $filename) {
 102                 my $result = move ( "$dir/$filename", "$destdir/$cardnumber.jpg" );
 103                         if ( $result ) {
 104                                 $count{count}++;
 105                                 push @{ $count{filenames} }, { source => $filename, dest => $cardnumber .".jpg" };
 106                         }
 107                 }
 108     }
 109     $count{source} = $dir;
 110     $count{dest} = $destdir;
 111     push @counts, \%count;
 112     close FILE;
 113     return 1;
 114 }