fixup_deletions.php script to look for notices posted by now-deleted profiles and remove them.
This commit is contained in:
Brion Vibber
@@ -421,7 +421,9 @@ class Notice extends Memcached_DataObject
|
||||
}
|
||||
|
||||
$profile = Profile::staticGet($this->profile_id);
|
||||
$profile->blowNoticeCount();
|
||||
if (!empty($profile)) {
|
||||
$profile->blowNoticeCount();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
166
scripts/fixup_deletions.php
Executable file
166
scripts/fixup_deletions.php
Executable file
@@ -0,0 +1,166 @@
|
||||
#!/usr/bin/env php
|
||||
<?php
|
||||
/*
|
||||
* StatusNet - a distributed open-source microblogging tool
|
||||
* Copyright (C) 2010 StatusNet, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
|
||||
|
||||
$longoptions = array('dry-run', 'start=', 'end=');
|
||||
|
||||
$helptext = <<<END_OF_USERROLE_HELP
|
||||
fixup_deletions.php [options]
|
||||
Finds notices posted by deleted users and cleans them up.
|
||||
Stray incompletely deleted items cause various fun problems!
|
||||
|
||||
--dry-run look but don't touch
|
||||
--start=N start looking at profile_id N instead of 1
|
||||
--end=N end looking at profile_id N instead of the max
|
||||
|
||||
END_OF_USERROLE_HELP;
|
||||
|
||||
require_once INSTALLDIR.'/scripts/commandline.inc';
|
||||
|
||||
/**
|
||||
* Find the highest profile_id currently listed in the notice table;
|
||||
* this field is indexed and should return very quickly.
|
||||
*
|
||||
* We check notice.profile_id rather than profile.id because we're
|
||||
* looking for notices left behind after deletion; if the most recent
|
||||
* accounts were deleted, we wouldn't have them from profile.
|
||||
*
|
||||
* @return int
|
||||
* @access private
|
||||
*/
|
||||
function get_max_profile_id()
|
||||
{
|
||||
$query = 'SELECT MAX(profile_id) AS id FROM notice';
|
||||
|
||||
$profile = new Profile();
|
||||
$profile->query($query);
|
||||
|
||||
if ($profile->fetch()) {
|
||||
return intval($profile->id);
|
||||
} else {
|
||||
die("Something went awry; could not look up max used profile_id.");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Check for profiles in the given id range that are missing, presumed deleted.
|
||||
*
|
||||
* @param int $start beginning profile.id, inclusive
|
||||
* @param int $end final profile.id, inclusive
|
||||
* @return array of integer profile.ids
|
||||
* @access private
|
||||
*/
|
||||
function get_missing_profiles($start, $end)
|
||||
{
|
||||
$query = sprintf("SELECT id FROM profile WHERE id BETWEEN %d AND %d",
|
||||
$start, $end);
|
||||
|
||||
$profile = new Profile();
|
||||
$profile->query($query);
|
||||
|
||||
$all = range($start, $end);
|
||||
$known = array();
|
||||
while ($row = $profile->fetch()) {
|
||||
$known[] = intval($profile->id);
|
||||
}
|
||||
unset($profile);
|
||||
|
||||
$missing = array_diff($all, $known);
|
||||
return $missing;
|
||||
}
|
||||
|
||||
/**
|
||||
* Look for stray notices from this profile and, if present, kill them.
|
||||
*
|
||||
* @param int $profile_id
|
||||
* @param bool $dry if true, we won't delete anything
|
||||
*/
|
||||
function cleanup_missing_profile($profile_id, $dry)
|
||||
{
|
||||
$notice = new Notice();
|
||||
$notice->profile_id = $profile_id;
|
||||
$notice->find();
|
||||
if ($notice->N == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
$s = ($notice->N == 1) ? '' : 's';
|
||||
print "Deleted profile $profile_id has $notice->N stray notice$s:\n";
|
||||
|
||||
while ($notice->fetch()) {
|
||||
print " notice $notice->id";
|
||||
if ($dry) {
|
||||
print " (skipped; dry run)\n";
|
||||
} else {
|
||||
$victim = clone($notice);
|
||||
try {
|
||||
$victim->delete();
|
||||
print " (deleted)\n";
|
||||
} catch (Exception $e) {
|
||||
print " FAILED: ";
|
||||
print $e->getMessage();
|
||||
print "\n";
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$dry = have_option('dry-run');
|
||||
|
||||
$max_profile_id = get_max_profile_id();
|
||||
$chunk = 1000;
|
||||
|
||||
if (have_option('start')) {
|
||||
$begin = intval(get_option_value('start'));
|
||||
} else {
|
||||
$begin = 1;
|
||||
}
|
||||
if (have_option('end')) {
|
||||
$final = min($max_profile_id, intval(get_option_value('end')));
|
||||
} else {
|
||||
$final = $max_profile_id;
|
||||
}
|
||||
|
||||
if ($begin < 1) {
|
||||
die("Silly human, you can't begin before profile number 1!\n");
|
||||
}
|
||||
if ($final < $begin) {
|
||||
die("Silly human, you can't end at $final if it's before $begin!\n");
|
||||
}
|
||||
|
||||
// Identify missing profiles...
|
||||
for ($start = $begin; $start <= $final; $start += $chunk) {
|
||||
$end = min($start + $chunk - 1, $final);
|
||||
|
||||
print "Checking for missing profiles between id $start and $end";
|
||||
if ($dry) {
|
||||
print " (dry run)";
|
||||
}
|
||||
print "...\n";
|
||||
$missing = get_missing_profiles($start, $end);
|
||||
|
||||
foreach ($missing as $profile_id) {
|
||||
cleanup_missing_profile($profile_id, $dry);
|
||||
}
|
||||
}
|
||||
|
||||
echo "done.\n";
|
||||
|
||||
Reference in New Issue
Block a user