1 files changed, 6 insertions, 44 deletions

diff --git a/LedgerSMB/DBObject/Customer.pm b/LedgerSMB/DBObject/Customer.pm
index 3805762f..a9bcffc7 100644
--- a/LedgerSMB/DBObject/Customer.pm
+++ b/LedgerSMB/DBObject/Customer.pm
@@ -1,8 +1,6 @@
 package LedgerSMB::DBObject::Customer;
 
-use base qw(LedgerSMB::DBObject);
-use LedgerSMB::DBObject;
-use LedgerSMB::Entity;
+use base qw(LedgerSMB::DBObject::Company);
 use strict;
 
 my $CUSTOMER_ENTITY_CLASS = 2;
@@ -10,66 +8,30 @@ my $CUSTOMER_ENTITY_CLASS = 2;
 sub save {
     my $self = shift @_;
 
-    # This saves both the entity and the credit account. -- CT
     $self->{entity_class} = $CUSTOMER_ENTITY_CLASS;
-    
-    my ($ref) = $self->exec_method(funcname => 'entity_credit_save');
-    $self->{entity_id} = $ref->{entity_credit_save};
-    $self->{dbh}->commit;
+    $self->save_credit(); # inherited from Company    
 }
 
-sub get_metadata {
-    my $self = shift @_;
-
-    @{$self->{location_class_list}} = 
-         $self->exec_method(funcname => 'location_list_class');
-
-    @{$self->{country_list}} = 
-         $self->exec_method(funcname => 'location_list_country');
-
-    @{$self->{contact_class_list}} = 
-         $self->exec_method(funcname => 'entity_list_contact_class');
-}
 
 sub save_location {
     my $self = shift @_;
     $self->{entity_class} = $CUSTOMER_ENTITY_CLASS;
     $self->{country_id} = $self->{country};
-    $self->exec_method(funcname => 'customer_location_save');
+    $self->exec_method(funcname => 'company__location_save');
 
     $self->{dbh}->commit;
 }
 
-sub save_contact {
-    my ($self) = @_;
-    $self->exec_method(funcname => 'company__save_contact');
-    $self->{dbh}->commit;
-}
 
-sub save_bank_account {
-    my $self = shift @_;
-    $self->exec_method(funcname => 'entity__save_bank_account');
-    $self->{dbh}->commit;
-}
 
 sub get {
     my $self = shift @_;
-    my ($ref) = $self->exec_method(funcname => 'customer__retrieve');
+    $self->{entity_class} = $CUSTOMER_ENTITY_CLASS;
+    my ($ref) = $self->exec_method(funcname => 'entity__retrieve_credit');
     $self->merge($ref);
 
     $self->{name} = $self->{legal_name};
-
-    @{$self->{locations}} = $self->exec_method(
-		funcname => 'company__list_locations');
-
-    @{$self->{contacts}} = $self->exec_method(
-		funcname => 'company__list_contacts');
-
-    @{$self->{bank_account}} = $self->exec_method(
-		funcname => 'company__list_bank_account');
-
-    @{$self->{notes}} = $self->exec_method(
-		funcname => 'company__list_notes');
+    $self->get_company();
 }