From: Ariel Glenn Date: Sun, 10 Jul 2011 14:20:17 +0000 (+0000) Subject: fix issues after failed renameUser; this script will ask for lots of confirmation... X-Git-Tag: 1.31.0-rc.0~28953 X-Git-Url: https://git.cyclocoop.org/%7B%24www_url%7Dadmin/compta/exercices/journal.php?a=commitdiff_plain;h=3e7256ed64e238a6d4e2018a5e73ba32c3315c65;p=lhc%2Fweb%2Fwiklou.git fix issues after failed renameUser; this script will ask for lots of confirmation, better safe than sorry. --- diff --git a/maintenance/renameUserCleanup.php b/maintenance/renameUserCleanup.php new file mode 100644 index 0000000000..892555b005 --- /dev/null +++ b/maintenance/renameUserCleanup.php @@ -0,0 +1,185 @@ + + */ + +require_once( dirname( __FILE__ ) . '/Maintenance.php' ); + +class RenameUserCleanup extends Maintenance { + const BATCH_SIZE = 1000; + + public function __construct() { + parent::__construct(); + $this->mDescription = "Maintenance script to finish incomplete rename user, in particular to reassign edits that were missed"; + $this->addOption( 'olduser', 'Old user name', true, true ); + $this->addOption( 'newuser', 'New user name', true, true ); + } + + public function execute() { + $this->output( "Rename User Cleanup starting...\n\n" ); + $olduser = User::newFromName( $this->getOption( 'olduser' ) ); + $newuser = User::newFromName( $this->getOption( 'newuser' ) ); + if ( !$newuser->getId() ) { + $this->error( "No such user: " . $this->getOption( 'newuser' ), true ); + exit(1); + } + if ($olduser->getId() ) { + print( "WARNING!!: Old user still exists: " . $this->getOption( 'olduser' ) . "\n"); + print("proceed anyways? We'll only re-attribute edits that have the new user uid (or 0) and the old user name. [N/y] "); + $stdin = fopen ("php://stdin","rt"); + $line = fgets($stdin); + fclose($stdin); + if ( $line[0] != "Y" && $line[0] != "y" ) { + print("Exiting at user's request\n"); + exit(0); + } + } + + $dbr = wfGetDB( DB_SLAVE ); + $result = $dbr->select( 'logging', '*', + array( 'log_type' => 'renameuser', + 'log_action' => 'renameuser', + 'log_title' => $olduser->getName(), + 'log_params' => $newuser->getName() + ), + __METHOD__ + ); + if (! $result ) { + print("No log entry found for a rename of ".$olduser->getName()." to ".$newuser->getName().", giving up\n"); + exit(1); + } + foreach ( $result as $row ) { + print("Found log entry of the rename: ".$olduser->getName()." to ".$newuser->getName()." on $row->log_timestamp\n"); + } + if ($result->numRows() > 1) { + print("More than one rename entry found in the log, not sure what to do. Continue anyways? [N/y] "); + $stdin = fopen ("php://stdin","rt"); + $line = fgets($stdin); + fclose($stdin); + if ( $line[0] != "Y" && $line[0] != "y" ) { + print("Exiting at user's request\n"); + exit(1); + } + } + $dbw = wfGetDB( DB_MASTER ); + + $this->updateTable('revision', 'rev_user_text', 'rev_user', 'rev_timestamp', $olduser, $newuser, $dbw); + $this->updateTable('archive', 'ar_user_text', 'ar_user', 'ar_timestamp', $olduser, $newuser, $dbw); + $this->updateTable('logging', 'log_user_text', 'log_user', 'log_timestamp', $olduser, $newuser, $dbw); + $this->updateTable('image', 'img_user_text', 'img_user', 'img_timestamp', $olduser, $newuser, $dbw); + $this->updateTable('oldimage', 'oi_user_text', 'oi_user', 'oi_timestamp', $olduser, $newuser, $dbw); +# FIXME: updateTable('filearchive', 'fa_user_text','fa_user', 'fa_timestamp', $olduser, $newuser, $dbw); (not indexed yet) + print "Done!\n"; + exit(0); + } + + public function updateTable($table,$usernamefield,$useridfield,$timestampfield,$olduser,$newuser,$dbw) { + $doUid = 0; + + $contribs = $dbw->selectField( $table, 'count(*)', + array( $usernamefield => $olduser->getName(), $useridfield => $newuser->getId() ), __METHOD__ ); + if ($contribs == 0) { + $contribs = $dbw->selectField( $table, 'count(*)', + array( $usernamefield => $olduser->getName(), $useridfield => 0 ), __METHOD__ ); + if ($contribs > 0) { + print("Found $contribs edits to be re-attributed from table $table but the uid present is 0 (should be ".$newuser->getId().")\n"); + print("If you proceed, the uid field will be set to that of the new user name (i.e. ".$newuser->getId().") in these rows.\n"); + $doUid = 1; + } + else { + print("No edits to be re-attributed from table $table\n"); + return(0); + } + } + else { + print("total number of edits to be re-attributed from table $table: $contribs\n"); + } + print("proceed? [N/y] "); + $stdin = fopen ("php://stdin","rt"); + $line = fgets($stdin); + fclose($stdin); + if ( $line[0] != "Y" && $line[0] != "y" ) { + print("skipping at user's request\n"); + return(0); + } + $selectConds = array( $usernamefield => $olduser->getName() ); + $updateFields = array( $usernamefield => $newuser->getName() ); + $updateConds = array( $usernamefield => $olduser->getName() ); + + $extraConds = array( $useridfield => $newuser->getId() ); + $extraCondsNoUid = array( $useridfield => 0 ); + # uid in rows is set properly, use as cond to find rows, don't bother to update it + if (! $doUid) { + $selectConds = array_merge( $selectConds, $extraConds ); + $updateConds = array_merge( $updateConds, $extraConds ); + } + # uid in edit rows is 0, we will set it and we will only update rows with 0 uid and the old user name + else { + $selectConds = array_merge( $selectConds, $extraCondsNoUid ); + $updateConds = array_merge( $updateConds, $extraCondsNoUid ); + $updateFields = array_merge( $updateFields, $extraConds ); + } + + while ($contribs > 0) { + print("doing batch of up to approximately ".self::BATCH_SIZE."\n"); + print("do this batch? [N/y] "); + $stdin = fopen ("php://stdin","rt"); + $line = fgets($stdin); + fclose($stdin); + if ( $line[0] != "Y" && $line[0] != "y" ) { + print("skipping at user's request\n"); + return(0); + } + $dbw->begin(); + $result = $dbw->select( $table, $timestampfield, $selectConds , __METHOD__, + array( 'ORDER BY' => $timestampfield.' DESC', 'LIMIT' => self::BATCH_SIZE ) ); + if (! $result) { + print("There were rows for updating but now they are gone. Skipping.\n"); + $db->rollback(); + return(0); + } + $result->seek($result->numRows() -1 ); + $row = $result->fetchObject(); + $timestamp = $row->$timestampfield; + $updateCondsWithTime = array_merge( $updateConds, array ("$timestampfield >= $timestamp") ); + $success = $dbw->update( $table, $updateFields, $updateCondsWithTime, __METHOD__ ); + if ($success) { + $rowsDone = $dbw->affectedRows(); + $dbw->commit(); + } + else { + print("problem with the update, rolling back and exiting\n"); + $db->rollback(); + exit(1); + } + $contribs = User::edits( $olduser->getId() ); + $contribs = $dbw->selectField( $table, 'count(*)', $selectConds, __METHOD__ ); + print("updated $rowsDone edits; $contribs edits remaining to be re-attributed\n"); + } + return(0); + } + +} + +$maintClass = "RenameUserCleanup"; +require_once( RUN_MAINTENANCE_IF_MAIN );