1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
<?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;
|
GitWeb