RT# 83450 - fixed rateplan export

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

diff --git a/FS/FS/type_pkgs.pm b/FS/FS/type_pkgs.pm
index bf34e7c..8cc2345 100644 (file)
--- a/FS/FS/type_pkgs.pm
+++ b/FS/FS/type_pkgs.pm
@@ -1,12 +1,7 @@
 package FS::type_pkgs;
+use base qw( FS::Record );
 
 use strict;
-use vars qw( @ISA );
-use FS::Record qw( qsearchs );
-use FS::agent_type;
-use FS::part_pkg;
-
-@ISA = qw( FS::Record );
 
 =head1 NAME
 
@@ -83,17 +78,11 @@ sub check {
 
   my $error = 
        $self->ut_numbern('typepkgnum')
-    || $self->ut_number('typenum')
-    || $self->ut_number('pkgpart')
+    || $self->ut_foreign_key('typenum', 'agent_type', 'typenum' )
+    || $self->ut_foreign_key('pkgpart', 'part_pkg',   'pkgpart' )
   ;
   return $error if $error;
 
-  return "Unknown typenum"
-    unless qsearchs( 'agent_type', { 'typenum' => $self->typenum } );
-
-  return "Unknown pkgpart"
-    unless qsearchs( 'part_pkg', { 'pkgpart' => $self->pkgpart } );
-
   $self->SUPER::check;
 }
 
@@ -101,12 +90,9 @@ sub check {
 
 Returns the FS::part_pkg object associated with this record.
 
-=cut
+=item agent_type
 
-sub part_pkg {
-  my $self = shift;
-  qsearchs( 'part_pkg', { 'pkgpart' => $self->pkgpart } );
-}
+Returns the FS::agent_type object associated with this record.
 
 =cut
 
@@ -116,7 +102,7 @@ sub part_pkg {
 
 =head1 SEE ALSO
 
-L<FS::Record>, L<FS::agent_type>, L<FS::part_pkgs>, schema.html from the base
+L<FS::Record>, L<FS::agent_type>, L<FS::part_pkg>, schema.html from the base
 documentation.
 
 =cut