bin/merge-user

   1 #!/usr/bin/perl -w
   2
   3 use strict;
   4 use FS::UID qw(adminsuidsetup dbh);
   5 use FS::Schema;
   6 use FS::Record qw(qsearch qsearchs);
   7
   8 my $DRY_RUN = 1;
   9 $FS::UID::AutoCommit = 0;
  10
  11 my ($user, $from_usernum, $to_usernum, $go) = @ARGV;
  12 die usage() if not ($user and $from_usernum and $to_usernum);
  13 $DRY_RUN = 0 if $go eq 'go';
  14
  15 my $dbh = adminsuidsetup($user);
  16
  17 # Sanity checks.
  18 die "Can't merge a user to itself." if $from_usernum == $to_usernum;
  19 my $from_user = FS::access_user->by_key($from_usernum) or
  20     die "Usernum '$from_usernum' not found.\n";
  21 my $to_user = FS::access_user->by_key($to_usernum) or
  22     die "Usernum '$to_usernum' not found.\n";
  23
  24 my $tables = FS::Schema::tables_hashref;
  25 foreach my $table (keys %$tables) {
  26   next if $table =~ /^access/; # Don't try to merge these.
  27   if( grep /^usernum$/, FS::Record::real_fields($table) ) {
  28     foreach ($table, "h_$table") {
  29       print "$_: ";
  30       my $sql =
  31         "UPDATE $_ SET usernum = $to_usernum WHERE usernum = $from_usernum";
  32       #print $sql;
  33       my $sth = $dbh->prepare($sql);
  34       $sth->execute;
  35       if($dbh->err) {
  36         print $dbh->errstr."\n";
  37         $dbh->rollback;
  38         exit(1);
  39       }
  40       print $sth->rows, "\n";
  41     }
  42   }
  43 }
  44
  45 if($DRY_RUN) {
  46   warn "Dry run complete.  Reverting all changes.\n";
  47   $dbh->rollback;
  48 }
  49 else {
  50 # Warning: access_user->delete does not transactionize because of
  51 # htpasswd issues.
  52   print "Deleting merged user.\n";
  53   my $error = $from_user->delete;
  54   die $error if $error;
  55
  56   warn "Committing changes.\n";
  57   $dbh->commit;
  58 }
  59 exit(0);
  60
  61 sub usage {
  62   "Usage:\n    merge-user admin_user from_usernum to_usernum [ 'go' ]\n
  63   (Specify 'go' to actually apply changes.)\n\n";
  64 }