X-Git-Url: http://git.heureux-cyclage.org/?a=blobdiff_plain;f=maintenance%2FupdateRestrictions.php;h=8699dc26b7252995e856b319a77f234f003175fc;hb=43a4dcc2d25f5aee2f88b92f72703f55d1c63381;hp=59879eaa5e5bdbf61d3d28b795bf1d11711ce74c;hpb=c56fb130e73ca5116dac49af04c214f317d65482;p=lhc%2Fweb%2Fwiklou.git diff --git a/maintenance/updateRestrictions.php b/maintenance/updateRestrictions.php index 59879eaa5e..8699dc26b7 100644 --- a/maintenance/updateRestrictions.php +++ b/maintenance/updateRestrictions.php @@ -5,88 +5,113 @@ * schema change. All remaining page_restriction column values are moved * to the new table. * + * 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 + * * @file * @ingroup Maintenance */ -define( 'BATCH_SIZE', 100 ); +require_once( __DIR__ . '/Maintenance.php' ); -require_once 'commandLine.inc'; - -$db =& wfGetDB( DB_MASTER ); -if ( !$db->tableExists( 'page_restrictions' ) ) { - echo "page_restrictions does not exist\n"; - exit( 1 ); -} +/** + * Maintenance script that updates page_restrictions table from + * old page_restriction column. + * + * @ingroup Maintenance + */ +class UpdateRestrictions extends Maintenance { + public function __construct() { + parent::__construct(); + $this->mDescription = "Updates page_restrictions table from old page_restriction column"; + $this->setBatchSize( 100 ); + } + + public function execute() { + $db = wfGetDB( DB_MASTER ); + if ( !$db->tableExists( 'page_restrictions' ) ) { + $this->error( "page_restrictions table does not exist", true ); + } -migrate_page_restrictions( $db ); + $start = $db->selectField( 'page', 'MIN(page_id)', false, __METHOD__ ); + if ( !$start ) { + $this->error( "Nothing to do.", true ); + } + $end = $db->selectField( 'page', 'MAX(page_id)', false, __METHOD__ ); -function migrate_page_restrictions( $db ) { - $start = $db->selectField( 'page', 'MIN(page_id)', false, __FUNCTION__ ); - if( !$start ) { - die("Nothing to do.\n"); - } - $end = $db->selectField( 'page', 'MAX(page_id)', false, __FUNCTION__ ); - - # Do remaining chunk - $end += BATCH_SIZE - 1; - $blockStart = $start; - $blockEnd = $start + BATCH_SIZE - 1; - $encodedExpiry = 'infinity'; - while( $blockEnd <= $end ) { - echo "...doing page_id from $blockStart to $blockEnd\n"; - $cond = "page_id BETWEEN $blockStart AND $blockEnd AND page_restrictions !=''"; - $res = $db->select( 'page', array('page_id','page_namespace','page_restrictions'), $cond, __FUNCTION__ ); - $batch = array(); - while( $row = $db->fetchObject( $res ) ) { - $oldRestrictions = array(); - foreach( explode( ':', trim( $row->page_restrictions ) ) as $restrict ) { - $temp = explode( '=', trim( $restrict ) ); - // Make sure we are not settings restrictions to "" - if( count($temp) == 1 && $temp[0] ) { - // old old format should be treated as edit/move restriction - $oldRestrictions["edit"] = trim( $temp[0] ); - $oldRestrictions["move"] = trim( $temp[0] ); - } else if( $temp[1] ) { - $oldRestrictions[$temp[0]] = trim( $temp[1] ); + # Do remaining chunk + $end += $this->mBatchSize - 1; + $blockStart = $start; + $blockEnd = $start + $this->mBatchSize - 1; + $encodedExpiry = 'infinity'; + while ( $blockEnd <= $end ) { + $this->output( "...doing page_id from $blockStart to $blockEnd\n" ); + $cond = "page_id BETWEEN $blockStart AND $blockEnd AND page_restrictions !=''"; + $res = $db->select( 'page', array( 'page_id', 'page_namespace', 'page_restrictions' ), $cond, __METHOD__ ); + $batch = array(); + foreach ( $res as $row ) { + $oldRestrictions = array(); + foreach ( explode( ':', trim( $row->page_restrictions ) ) as $restrict ) { + $temp = explode( '=', trim( $restrict ) ); + // Make sure we are not settings restrictions to "" + if ( count( $temp ) == 1 && $temp[0] ) { + // old old format should be treated as edit/move restriction + $oldRestrictions["edit"] = trim( $temp[0] ); + $oldRestrictions["move"] = trim( $temp[0] ); + } elseif ( $temp[1] ) { + $oldRestrictions[$temp[0]] = trim( $temp[1] ); + } + } + # Clear invalid columns + if ( $row->page_namespace == NS_MEDIAWIKI ) { + $db->update( 'page', array( 'page_restrictions' => '' ), + array( 'page_id' => $row->page_id ), __FUNCTION__ ); + $this->output( "...removed dead page_restrictions column for page {$row->page_id}\n" ); + } + # Update restrictions table + foreach ( $oldRestrictions as $action => $restrictions ) { + $batch[] = array( + 'pr_page' => $row->page_id, + 'pr_type' => $action, + 'pr_level' => $restrictions, + 'pr_cascade' => 0, + 'pr_expiry' => $encodedExpiry + ); } } - # Clear invalid columns - if( $row->page_namespace == NS_MEDIAWIKI ) { - $db->update( 'page', array( 'page_restrictions' => '' ), - array( 'page_id' => $row->page_id ), __FUNCTION__ ); - echo "...removed dead page_restrictions column for page {$row->page_id}\n"; - } - # Update restrictions table - foreach( $oldRestrictions as $action => $restrictions ) { - $batch[] = array( - 'pr_page' => $row->page_id, - 'pr_type' => $action, - 'pr_level' => $restrictions, - 'pr_cascade' => 0, - 'pr_expiry' => $encodedExpiry - ); - } - } - # We use insert() and not replace() as Article.php replaces - # page_restrictions with '' when protected in the restrictions table - if ( count( $batch ) ) { - $ok = $db->deadlockLoop( array( $db, 'insert' ), 'page_restrictions', - $batch, __FUNCTION__, array( 'IGNORE' ) ); - if( !$ok ) { - throw new MWException( "Deadlock loop failed wtf :(" ); + # We use insert() and not replace() as Article.php replaces + # page_restrictions with '' when protected in the restrictions table + if ( count( $batch ) ) { + $ok = $db->deadlockLoop( array( $db, 'insert' ), 'page_restrictions', + $batch, __FUNCTION__, array( 'IGNORE' ) ); + if ( !$ok ) { + throw new MWException( "Deadlock loop failed wtf :(" ); + } } + $blockStart += $this->mBatchSize - 1; + $blockEnd += $this->mBatchSize - 1; + wfWaitForSlaves(); } - $blockStart += BATCH_SIZE - 1; - $blockEnd += BATCH_SIZE - 1; - wfWaitForSlaves( 5 ); + $this->output( "...removing dead rows from page_restrictions\n" ); + // Kill any broken rows from previous imports + $db->delete( 'page_restrictions', array( 'pr_level' => '' ) ); + // Kill other invalid rows + $db->deleteJoin( 'page_restrictions', 'page', 'pr_page', 'page_id', array( 'page_namespace' => NS_MEDIAWIKI ) ); + $this->output( "...Done!\n" ); } - echo "...removing dead rows from page_restrictions\n"; - // Kill any broken rows from previous imports - $db->delete( 'page_restrictions', array( 'pr_level' => '' ) ); - // Kill other invalid rows - $db->deleteJoin( 'page_restrictions', 'page', 'pr_page', 'page_id', array('page_namespace' => NS_MEDIAWIKI) ); - echo "...Done!\n"; } - +$maintClass = "UpdateRestrictions"; +require_once( RUN_MAINTENANCE_IF_MAIN );