MediaWiki 1.16.0

[autoinstalls/mediawiki.git] / maintenance / cleanupCaps.php

diff --git a/maintenance/cleanupCaps.php b/maintenance/cleanupCaps.php
index 77a95ffa1a1a93ba74a15492a93d1276d02d7581..6a48ea832b57f0db376f84318a1ebe088b9e38bc 100644 (file)
--- a/maintenance/cleanupCaps.php
+++ b/maintenance/cleanupCaps.php
@@ -25,87 +25,34 @@
  * http://www.gnu.org/copyleft/gpl.html
  *
  * @author Brion Vibber <brion at pobox.com>
- * @addtogroup maintenance
+ * @ingroup maintenance
  */
 
-$options = array( 'dry-run' );
+require_once( dirname(__FILE__) . '/cleanupTable.inc' );
 
-require_once( 'commandLine.inc' );
-require_once( 'FiveUpgrade.inc' );
-
-class CapsCleanup extends FiveUpgrade {
-       function CapsCleanup( $dryrun = false, $namespace=0 ) {
-               parent::FiveUpgrade();
-
-               $this->maxLag = 10; # if slaves are lagged more than 10 secs, wait
-               $this->dryrun = $dryrun;
-               $this->namespace = intval( $namespace );
-       }
-
-       function cleanup() {
-               global $wgCapitalLinks;
-               if( $wgCapitalLinks ) {
-                       echo "\$wgCapitalLinks is on -- no need for caps links cleanup.\n";
-                       return false;
-               }
-
-               $this->runTable( 'page', 'WHERE page_namespace=' . $this->namespace,
-                       array( &$this, 'processPage' ) );
+class CapsCleanup extends TableCleanup {
+       public function __construct() {
+               parent::__construct();
+               $this->mDescription = "Script to cleanup capitalization";
+               $this->addOption( 'namespace', 'Namespace number to run caps cleanup on', false, true );
        }
 
-       function init( $count, $table ) {
-               $this->processed = 0;
-               $this->updated = 0;
-               $this->count = $count;
-               $this->startTime = wfTime();
-               $this->table = $table;
-       }
-
-       function progress( $updated ) {
-               $this->updated += $updated;
-               $this->processed++;
-               if( $this->processed % 100 != 0 ) {
-                       return;
-               }
-               $portion = $this->processed / $this->count;
-               $updateRate = $this->updated / $this->processed;
-
-               $now = wfTime();
-               $delta = $now - $this->startTime;
-               $estimatedTotalTime = $delta / $portion;
-               $eta = $this->startTime + $estimatedTotalTime;
-
-               printf( "%s: %6.2f%% done on %s; ETA %s [%d/%d] %.2f/sec <%.2f%% updated>\n",
-                       wfTimestamp( TS_DB, intval( $now ) ),
-                       $portion * 100.0,
-                       $this->table,
-                       wfTimestamp( TS_DB, intval( $eta ) ),
-                       $this->processed,
-                       $this->count,
-                       $this->processed / $delta,
-                       $updateRate * 100.0 );
-               flush();
+       public function execute() {
+               global $wgCapitalLinks, $wgUser;
+               $this->namespace = intval( $this->getOption( 'namespace', 0 ) );
+               $this->dryrun = $this->hasOption( 'dry-run' );
+               $wgUser->setName( 'Conversion script' );
+               if( $wgCapitalLinks )
+                       $this->error( "\$wgCapitalLinks is on -- no need for caps links cleanup.", true );
+
+               $this->runTable( array(
+                       'table' => 'page',
+                       'conds' => array( 'page_namespace' => $this->namespace ),
+                       'index' => 'page_id',
+                       'callback' => 'processRow' ) );
        }
 
-       function runTable( $table, $where, $callback ) {
-               $fname = 'CapsCleanup::buildTable';
-
-               $count = $this->dbw->selectField( $table, 'count(*)', '', $fname );
-               $this->init( $count, 'page' );
-               $this->log( "Processing $table..." );
-
-               $tableName = $this->dbr->tableName( $table );
-               $sql = "SELECT * FROM $tableName $where";
-               $result = $this->dbr->query( $sql, $fname );
-
-               while( $row = $this->dbr->fetchObject( $result ) ) {
-                       call_user_func( $callback, $row );
-               }
-               $this->log( "Finished $table... $this->updated of $this->processed rows updated" );
-               $this->dbr->freeResult( $result );
-       }
-
-       function processPage( $row ) {
+       protected function processRow( $row ) {
                global $wgContLang;
 
                $current = Title::makeTitle( $row->page_namespace, $row->page_title );
@@ -113,44 +60,38 @@ class CapsCleanup extends FiveUpgrade {
                $upper = $row->page_title;
                $lower = $wgContLang->lcfirst( $row->page_title );
                if( $upper == $lower ) {
-                       $this->log( "\"$display\" already lowercase." );
+                       $this->output( "\"$display\" already lowercase.\n" );
                        return $this->progress( 0 );
                }
 
                $target = Title::makeTitle( $row->page_namespace, $lower );
                $targetDisplay = $target->getPrefixedText();
                if( $target->exists() ) {
-                       $this->log( "\"$display\" skipped; \"$targetDisplay\" already exists" );
+                       $this->output( "\"$display\" skipped; \"$targetDisplay\" already exists\n" );
                        return $this->progress( 0 );
                }
 
                if( $this->dryrun ) {
-                       $this->log( "\"$display\" -> \"$targetDisplay\": DRY RUN, NOT MOVED" );
+                       $this->output( "\"$display\" -> \"$targetDisplay\": DRY RUN, NOT MOVED\n" );
                        $ok = true;
                } else {
                        $ok = $current->moveTo( $target, false, 'Converting page titles to lowercase' );
-                       $this->log( "\"$display\" -> \"$targetDisplay\": $ok" );
+                       $this->output( "\"$display\" -> \"$targetDisplay\": $ok\n" );
                }
                if( $ok === true ) {
                        $this->progress( 1 );
-
                        if( $row->page_namespace == $this->namespace ) {
                                $talk = $target->getTalkPage();
                                $row->page_namespace = $talk->getNamespace();
                                if( $talk->exists() ) {
-                                       return $this->processPage( $row );
+                                       return $this->processRow( $row );
                                }
                        }
                } else {
                        $this->progress( 0 );
                }
        }
-
 }
 
-$wgUser->setName( 'Conversion script' );
-$ns = isset( $options['namespace'] ) ? $options['namespace'] : 0;
-$caps = new CapsCleanup( isset( $options['dry-run'] ), $ns );
-$caps->cleanup();
-
-
+$maintClass = "CapsCleanup";
+require_once( DO_MAINTENANCE );