3 * Makes the required database updates for Special:ProtectedPages
4 * to show all protected pages, even ones before the page restrictions
5 * schema change. All remaining page_restriction column values are moved
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
21 * http://www.gnu.org/copyleft/gpl.html
24 * @ingroup Maintenance
27 require_once __DIR__
. '/Maintenance.php';
30 * Maintenance script that updates page_restrictions table from
31 * old page_restriction column.
33 * @ingroup Maintenance
35 class UpdateRestrictions
extends Maintenance
{
36 public function __construct() {
37 parent
::__construct();
38 $this->addDescription( 'Updates page_restrictions table from old page_restriction column' );
39 $this->setBatchSize( 1000 );
42 public function execute() {
43 $db = $this->getDB( DB_MASTER
);
44 $batchSize = $this->getBatchSize();
45 if ( !$db->tableExists( 'page_restrictions' ) ) {
46 $this->fatalError( "page_restrictions table does not exist" );
49 $start = $db->selectField( 'page', 'MIN(page_id)', false, __METHOD__
);
51 $this->fatalError( "Nothing to do." );
53 $end = $db->selectField( 'page', 'MAX(page_id)', false, __METHOD__
);
56 $end +
= $batchSize - 1;
58 $blockEnd = $start +
$batchSize - 1;
59 $encodedExpiry = 'infinity';
60 while ( $blockEnd <= $end ) {
61 $this->output( "...doing page_id from $blockStart to $blockEnd out of $end\n" );
62 $cond = "page_id BETWEEN " . (int)$blockStart . " AND " . (int)$blockEnd .
63 " AND page_restrictions !=''";
66 [ 'page_id', 'page_namespace', 'page_restrictions' ],
71 foreach ( $res as $row ) {
72 $oldRestrictions = [];
73 foreach ( explode( ':', trim( $row->page_restrictions
) ) as $restrict ) {
74 $temp = explode( '=', trim( $restrict ) );
75 // Make sure we are not settings restrictions to ""
76 if ( count( $temp ) == 1 && $temp[0] ) {
77 // old old format should be treated as edit/move restriction
78 $oldRestrictions["edit"] = trim( $temp[0] );
79 $oldRestrictions["move"] = trim( $temp[0] );
80 } elseif ( $temp[1] ) {
81 $oldRestrictions[$temp[0]] = trim( $temp[1] );
84 # Clear invalid columns
85 if ( $row->page_namespace
== NS_MEDIAWIKI
) {
86 $db->update( 'page', [ 'page_restrictions' => '' ],
87 [ 'page_id' => $row->page_id
], __FUNCTION__
);
88 $this->output( "...removed dead page_restrictions column for page {$row->page_id}\n" );
90 # Update restrictions table
91 foreach ( $oldRestrictions as $action => $restrictions ) {
93 'pr_page' => $row->page_id
,
95 'pr_level' => $restrictions,
97 'pr_expiry' => $encodedExpiry
101 # We use insert() and not replace() as Article.php replaces
102 # page_restrictions with '' when protected in the restrictions table
103 if ( count( $batch ) ) {
104 $ok = $db->deadlockLoop( [ $db, 'insert' ], 'page_restrictions',
105 $batch, __FUNCTION__
, [ 'IGNORE' ] );
107 throw new MWException( "Deadlock loop failed wtf :(" );
110 $blockStart +
= $batchSize - 1;
111 $blockEnd +
= $batchSize - 1;
114 $this->output( "...removing dead rows from page_restrictions\n" );
115 // Kill any broken rows from previous imports
116 $db->delete( 'page_restrictions', [ 'pr_level' => '' ] );
117 // Kill other invalid rows
123 [ 'page_namespace' => NS_MEDIAWIKI
]
125 $this->output( "...Done!\n" );
129 $maintClass = "UpdateRestrictions";
130 require_once RUN_MAINTENANCE_IF_MAIN
;