X-Git-Url: http://git.freeside.biz/gitweb/?p=freeside.git;a=blobdiff_plain;f=FS%2FFS%2Fpart_referral.pm;h=c0858c0eda558eda67dd4115619469e5224aa4d7;hp=23885dffd6fc3d8bd73cb7cf919f1dde17bd742c;hb=f11341c6ca588f1ff5e46a857540e88d49d7647a;hpb=0ebeec96313dd7edfca340f01f8fbbbac1f4aa1d

diff --git a/FS/FS/part_referral.pm b/FS/FS/part_referral.pm
index 23885dffd..c0858c0ed 100644
--- a/FS/FS/part_referral.pm
+++ b/FS/FS/part_referral.pm
@@ -38,6 +38,8 @@ The following fields are currently supported:
 
 =item referral - Text name of this advertising source
 
+=item disabled - Disabled flag, empty or 'Y'
+
 =back
 
 =head1 NOTE
@@ -91,9 +93,17 @@ replace methods.
 sub check {
   my $self = shift;
 
-  $self->ut_numbern('refnum')
+  my $error = $self->ut_numbern('refnum')
     || $self->ut_text('referral')
   ;
+  return $error if $error;
+
+  if ( $self->dbdef_table->column('disabled') ) {
+    $error = $self->ut_enum('disabled', [ '', 'Y' ] );
+    return $error if $error;
+  }
+
+  $self->SUPER::check;
 }
 
 =back