X-Git-Url: http://git.freeside.biz/gitweb/?p=freeside.git;a=blobdiff_plain;f=FS%2FFS%2Fotaker_Mixin.pm;h=186c7d1bac4e517ed6675553c35e3894617e07f0;hp=071958c333a0ba3cd7fc5dff58317065315b24b4;hb=3adb46fccf9f631e188ea5383bd147b340477639;hpb=b516a7cb6c072871568bb0f217cc2bab6ebddd4a

diff --git a/FS/FS/otaker_Mixin.pm b/FS/FS/otaker_Mixin.pm
index 071958c33..186c7d1ba 100644
--- a/FS/FS/otaker_Mixin.pm
+++ b/FS/FS/otaker_Mixin.pm
@@ -4,18 +4,35 @@ use strict;
 use Carp qw( croak ); #confess );
 use FS::Record qw( qsearch qsearchs );
 use FS::access_user;
+use FS::UID qw( dbh );
 
 sub otaker {
   my $self = shift;
   if ( scalar(@_) ) { #set
     my $otaker = shift;
-    my $access_user = qsearchs('access_user', { 'username' => $otaker } )
-      or croak "can't set otaker: $otaker not found!"; #confess?
+    my $access_user = qsearchs('access_user', { 'username' => $otaker } );
+    if ( !$access_user && $otaker =~ /^(.+), (.+)$/ ) { #same as below..
+      my($lastname, $firstname) = ($1, $2);
+      $otaker = lc($firstname.$lastname);
+      $otaker =~ s/ //g;
+      $access_user =  qsearchs('access_user', { 'first' => $firstname, 
+                                                'last'  => $lastname  } )
+                   || qsearchs('access_user', { 'username' => $otaker } );
+    }
+    croak "can't set otaker: $otaker not found!" unless $access_user; #confess?
     $self->usernum( $access_user->usernum );
     $otaker; #not sure return is used anywhere, but just in case
   } else { #get
     if ( $self->usernum ) {
-      $self->access_user->username;
+      # avoid a common failure mode: this should work even when the table 
+      # isn't foreign-keyed to access_user
+      my $access_user = FS::access_user->by_key($self->usernum);
+      if (!$access_user) {
+        croak "otaker called on ".$self->table."#".
+              $self->get($self->primary_key).
+              " but user does not exist";
+      }
+      return $access_user->username;
     } elsif ( length($self->get('otaker')) ) {
       $self->get('otaker');
     } else {
@@ -24,31 +41,35 @@ sub otaker {
   }
 }
 
-sub access_user {
-  my $self = shift;
-  qsearchs('access_user', { 'usernum' => $self->usernum } );
-}
-
 sub _upgrade_otaker {
   my $class = shift;
   my $table = $class->table;
 
+  my $limit = ( $table eq 'cust_attachment' ? 10 : 1000 );
+
   while ( 1 ) {
     my @records = qsearch({
                     'table'     => $table,
                     'hashref'   => {},
-                    'extra_sql' => 'WHERE otaker IS NOT NULL LIMIT 1000',
+                    'extra_sql' => "WHERE otaker IS NOT NULL LIMIT $limit",
                   });
     last unless @records;
 
     foreach my $record (@records) {
       eval { $record->otaker($record->otaker) };
       if ( $@ ) {
+        my $username = $record->otaker;
+        my($lastname, $firstname) = ( 'User', 'Legacy' );
+        if ( $username =~ /^(.+), (.+)$/ ) {
+          ($lastname, $firstname) = ($1, $2);
+          $username = lc($firstname.$lastname);
+          $username =~ s/ //g;
+        }
         my $access_user = new FS::access_user {
-          'username'  => $record->otaker,
+          'username'  => $username,
           '_password' => 'CHANGEME',
-          'first'     => 'Legacy',
-          'last'      => 'User',
+          'first'     => $firstname,
+          'last'      => $lastname,
           'disabled'  => 'Y',
         };
         my $error = $access_user->insert;
@@ -59,6 +80,11 @@ sub _upgrade_otaker {
       my $error = $record->replace;
       die $error if $error;
     }
+        
+    if ( $table eq 'cust_attachment' ) {
+	warn "  committing (cust_attachment) \n";
+	dbh->commit or die dbh->errstr;
+    }
 
   }