X-Git-Url: http://git.freeside.biz/gitweb/?a=blobdiff_plain;f=FS%2FFS%2Fsched_item.pm;fp=FS%2FFS%2Fsched_item.pm;h=de792a3eb19cab10e8df2acd727be12804623279;hb=a57c537ff9ef2a57de3225fbd49772be4b81e74c;hp=5f55188e61fb1eff8c93c3c9b3d3a8de039cfbb2;hpb=7c0748c7e439cdbc14ced38c21e782f4669f9b15;p=freeside.git

diff --git a/FS/FS/sched_item.pm b/FS/FS/sched_item.pm
index 5f55188e6..de792a3eb 100644
--- a/FS/FS/sched_item.pm
+++ b/FS/FS/sched_item.pm
@@ -2,7 +2,8 @@ package FS::sched_item;
 use base qw( FS::Record );
 
 use strict;
-#use FS::Record qw( qsearch qsearchs );
+use FS::Record qw( dbh ); # qsearch qsearchs );
+use FS::sched_avail;
 
 =head1 NAME
 
@@ -109,6 +110,43 @@ sub name {
   $access_user ? $access_user->name : $self->itemname;
 }
 
+=item replace_sched_avail SCHED_AVAIL, ...
+
+Replaces the existing availability schedule with the list of passed-in
+FS::sched_avail objects
+
+=cut
+
+sub replace_sched_avail {
+  my( $self, @new_sched_avail ) = @_;
+
+  my $oldAutoCommit = $FS::UID::AutoCommit;
+  local $FS::UID::AutoCommit = 0;
+  my $dbh = dbh;
+
+  foreach my $old_sched_avail ( $self->sched_avail ) {
+    my $error = $old_sched_avail->delete;
+    if ( $error ) {
+      $dbh->rollback if $oldAutoCommit;
+      return $error;
+    }
+  }
+
+  foreach my $new_sched_avail ( @new_sched_avail ) {
+    $new_sched_avail->itemnum( $self->itemnum );
+    my $error = $new_sched_avail->insert;
+    if ( $error ) {
+      $dbh->rollback if $oldAutoCommit;
+      return $error;
+    }
+  }
+
+  $dbh->commit or die $dbh->errstr if $oldAutoCommit;
+
+  '';
+
+}
+
 =back
 
 =head1 BUGS