mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Echo
synced 2024-11-24 16:04:35 +00:00
019d10b69d
The centralIdsFromNames() includes optimizations and error checking that was not in the original version. Bug: T178313 Change-Id: I3d6334f2e81f3cee66b36690734f70958fdcefdb
94 lines
2.1 KiB
PHP
94 lines
2.1 KiB
PHP
<?php
|
|
/**
|
|
* Update the Per User Blocklist from Usernames to User Ids.
|
|
*
|
|
* @ingroup Maintenance
|
|
*/
|
|
require_once getenv( 'MW_INSTALL_PATH' ) !== false
|
|
? getenv( 'MW_INSTALL_PATH' ) . '/maintenance/Maintenance.php'
|
|
: __DIR__ . '/../../../maintenance/Maintenance.php';
|
|
|
|
/**
|
|
* Maintenance script that changes the usernames to ids.
|
|
*
|
|
* @ingroup Maintenance
|
|
*/
|
|
class EchoUpdatePerUserBlacklist extends LoggedUpdateMaintenance {
|
|
|
|
public function __construct() {
|
|
parent::__construct();
|
|
|
|
$this->addDescription( 'Update echo-notifications-blacklist User Preference from Usernames to Ids' );
|
|
$this->setBatchSize( 100 );
|
|
$this->requireExtension( 'Echo' );
|
|
}
|
|
|
|
public function getUpdateKey() {
|
|
return __CLASS__;
|
|
}
|
|
|
|
public function doDBUpdates() {
|
|
$dbw = wfGetDB( DB_MASTER );
|
|
$dbr = wfGetDB( DB_REPLICA );
|
|
$iterator = new BatchRowIterator(
|
|
$dbr,
|
|
'user_properties',
|
|
[ 'up_user', 'up_property' ],
|
|
$this->mBatchSize
|
|
);
|
|
$iterator->setFetchColumns( [
|
|
'up_user',
|
|
'up_value'
|
|
] );
|
|
$iterator->addConditions( [
|
|
'up_property' => 'echo-notifications-blacklist'
|
|
] );
|
|
|
|
$this->output( "Updating Echo Notification Blacklist...\n" );
|
|
|
|
$lookup = CentralIdLookup::factory();
|
|
$processed = 0;
|
|
foreach ( $iterator as $batch ) {
|
|
foreach ( $batch as $row ) {
|
|
if ( !$row->up_value ) {
|
|
continue;
|
|
}
|
|
|
|
$value = explode( "\n", $row->up_value );
|
|
$names = array_filter( $value, function ( $item ) {
|
|
return !is_numeric( $item );
|
|
} );
|
|
|
|
// If all of the values are numeric then the user has already been
|
|
// converted.
|
|
if ( !$names ) {
|
|
continue;
|
|
}
|
|
|
|
$user = User::newFromId( $row->up_user );
|
|
$ids = $lookup->centralIdsFromNames( $names, $user );
|
|
|
|
$dbw->update(
|
|
'user_properties',
|
|
[
|
|
'up_value' => implode( "\n", $ids ),
|
|
],
|
|
[
|
|
'up_user' => $row->up_user,
|
|
'up_property' => 'echo-notifications-blacklist',
|
|
]
|
|
);
|
|
$processed += $dbw->affectedRows();
|
|
wfWaitForSlaves();
|
|
}
|
|
|
|
$this->output( "Updated $processed Users\n" );
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
$maintClass = 'EchoUpdatePerUserBlacklist';
|
|
require_once RUN_MAINTENANCE_IF_MAIN;
|