X-Git-Url: http://git.freeside.biz/gitweb/?p=freeside.git;a=blobdiff_plain;f=FS%2FFS%2Fotaker_Mixin.pm;h=af61a85ac12156ea5238e68ceb229ab154b33fbb;hp=9b86dc6d1a7785279e1974ee773eb7ebabb67e2d;hb=b49c1bd5510a5f10b05bffacc6cc8b6a1b2153e8;hpb=fabb47880497f35b385485d48b9b6a99db38cb4c diff --git a/FS/FS/otaker_Mixin.pm b/FS/FS/otaker_Mixin.pm index 9b86dc6d1..af61a85ac 100644 --- a/FS/FS/otaker_Mixin.pm +++ b/FS/FS/otaker_Mixin.pm @@ -4,13 +4,22 @@ 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 @@ -33,11 +42,13 @@ 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; @@ -45,14 +56,17 @@ sub _upgrade_otaker { eval { $record->otaker($record->otaker) }; if ( $@ ) { my $username = $record->otaker; + my($lastname, $firstname) = ( 'User', 'Legacy' ); if ( $username =~ /^(.+), (.+)$/ ) { - $username = lc($2.$1); + ($lastname, $firstname) = ($1, $2); + $username = lc($firstname.$lastname); + $username =~ s/ //g; } my $access_user = new FS::access_user { 'username' => $username, '_password' => 'CHANGEME', - 'first' => 'Legacy', - 'last' => 'User', + 'first' => $firstname, + 'last' => $lastname, 'disabled' => 'Y', }; my $error = $access_user->insert; @@ -63,6 +77,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; + } }