check size if playlist before import

git-svn-id: https://svn.helsinki.at/rhimport/trunk@59 7c65635b-ec39-4f67-a626-873dbafdd612

1 files changed, 23 insertions, 4 deletions

diff --git a/rhimport.pm b/rhimport.pm
index 6597fb4..7ecc2fe 100644
--- a/rhimport.pm
+++ b/rhimport.pm
@@ -108,6 +108,14 @@ sub get_used_carts
   return @carts;
 }
 
+sub get_num_carts
+{
+  my ($dbh, $group) = @_;
+
+  my ($low_cart, $high_cart) = get_cart_range($dbh, $group);
+  return $high_cart - $low_cart + 1;
+}
+
 sub clear_carts
 {
   my ($dbh, $group, $to_cart, $progress_cb) = @_;
@@ -150,7 +158,7 @@ sub clear_carts
 
 sub import_playlist
 {
-  my ($playlist, $dropbox, $user, $progress_cb, $error_cb) = @_;
+  my ($playlist, $dropbox, $user, $num_carts, $progress_cb, $error_cb) = @_;
 
   print "Starting import from playlist $playlist to $dropbox";
   my @entries;
@@ -165,10 +173,21 @@ sub import_playlist
   }
   close(FILE);
   my $num_entries = scalar(@entries);
-  print " ($num_entries Files in list)\n";
-  
+  print " ($num_entries Files in list -> $num_carts available)\n";
+
+  if($num_carts < $num_entries) {
+    if($error_cb) {
+      my $ret = $error_cb->("Achtung!\nDie Playlist beinhaltet mehr als $num_carts Dateien. Es werden nur die ersten $num_carts Dateien importiert.");
+      if(!$ret) {
+        return 0, "too many files in playlist";
+      }
+
+      $#entries = $num_carts-1;
+      $num_entries = scalar(@entries);
+    }
+  }
+  print "will import $num_entries files\n";
 
-  return (0, "");
   my $cnt = 0;
   my $ret = 0;
   my $log = "";