residential prospects, RT#7111

[freeside.git] / FS / FS / prospect_main.pm

diff --git a/FS/FS/prospect_main.pm b/FS/FS/prospect_main.pm
index 369029b..079ad25 100644 (file)
--- a/FS/FS/prospect_main.pm
+++ b/FS/FS/prospect_main.pm
@@ -41,14 +41,13 @@ from FS::Record.  The following fields are currently supported:
 
 primary key
 
-=item company
-
-company
+=item agentnum
 
-=item locationnum
+Agent
 
-locationnum
+=item company
 
+company
 
 =back
 
@@ -204,13 +203,27 @@ sub check {
   my $error = 
     $self->ut_numbern('prospectnum')
     || $self->ut_foreign_key('agentnum', 'agent', 'agentnum' )
-    || $self->ut_text('company')
+    || $self->ut_textn('company')
   ;
   return $error if $error;
 
   $self->SUPER::check;
 }
 
+=item name
+
+=cut
+
+sub name {
+  my $self = shift;
+  return $self->company if $self->company;
+
+  my $contact = ($self->contact)[0]; #first contact?  good enough for now
+  return $contact->line if $contact;
+
+  $self->prospectnum;
+}
+
 =item contact
 
 Returns the contacts (see L<FS::contact>) associated with this prospect.
@@ -222,6 +235,17 @@ sub contact {
   qsearch( 'contact', { 'prospectnum' => $self->prospectnum } );
 }
 
+=item cust_location
+
+Returns the locations (see L<FS::cust_location>) associated with this prospect.
+
+=cut
+
+sub cust_location {
+  my $self = shift;
+  qsearch( 'cust_location', { 'prospectnum' => $self->prospectnum } );
+}
+
 =item search HASHREF
 
 (Class method)