1 files changed, 41 insertions, 20 deletions
diff --git a/maintenance/fixUserRegistration.php b/maintenance/fixUserRegistration.php
index eb5b7f7d..d3305358 100644
--- a/maintenance/fixUserRegistration.php
+++ b/maintenance/fixUserRegistration.php
@@ -3,32 +3,53 @@
  * Fix the user_registration field.
  * In particular, for values which are NULL, set them to the date of the first edit
  *
- * @file
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
  * @ingroup Maintenance
  */
 
-require_once( 'commandLine.inc' );
+require_once( dirname(__FILE__) . '/Maintenance.php' );
 
-$fname = 'fixUserRegistration.php';
-
-$dbr = wfGetDB( DB_SLAVE );
-$dbw = wfGetDB( DB_MASTER );
+class FixUserRegistration extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->mDescription = "Fix the user_registration field";
+	}
 
-// Get user IDs which need fixing
-$res = $dbr->select( 'user', 'user_id', 'user_registration IS NULL', $fname );
+	public function execute() {
+		$dbr = wfGetDB( DB_SLAVE );
+		$dbw = wfGetDB( DB_MASTER );
 
-while ( $row = $dbr->fetchObject( $res ) ) {
-	$id = $row->user_id;
-	// Get first edit time
-	$timestamp = $dbr->selectField( 'revision', 'MIN(rev_timestamp)', array( 'rev_user' => $id ), $fname );
-	// Update
-	if ( !empty( $timestamp ) ) {
-		$dbw->update( 'user', array( 'user_registration' => $timestamp ), array( 'user_id' => $id ), $fname );
-		print "$id $timestamp\n";
-	} else {
-		print "$id NULL\n";
+		// Get user IDs which need fixing
+		$res = $dbr->select( 'user', 'user_id', 'user_registration IS NULL', __METHOD__ );
+		foreach ( $res as $row ) {
+			$id = $row->user_id;
+			// Get first edit time
+			$timestamp = $dbr->selectField( 'revision', 'MIN(rev_timestamp)', array( 'rev_user' => $id ), __METHOD__ );
+			// Update
+			if ( !empty( $timestamp ) ) {
+				$dbw->update( 'user', array( 'user_registration' => $timestamp ), array( 'user_id' => $id ), __METHOD__ );
+				$this->output( "$id $timestamp\n" );
+			} else {
+				$this->output( "$id NULL\n" );
+			}
+		}
+		$this->output( "\n" );
 	}
 }
-print "\n";
-
 
+$maintClass = "FixUserRegistration";
+require_once( DO_MAINTENANCE );