123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106 |
- <?php
- /*
- * StatusNet - the distributed open-source microblogging tool
- * Copyright (C) 2012 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__) . '/../../..'));
- $shortoptions = 'i:n:a';
- $longoptions = array('id=', 'nickname=', 'all');
- $helptext = <<<END_OF_TESTUSER_HELP
- testuser.php [options]
- Test user activities against the spam filter
- -i --id ID of user to export
- -n --nickname nickname of the user to export
- -a --all All users
- END_OF_TESTUSER_HELP;
- require_once INSTALLDIR.'/scripts/commandline.inc';
- function testAllUsers($filter) {
- $found = false;
- $offset = 0;
- $limit = 1000;
- do {
- $user = new User();
- $user->orderBy('created');
- $user->limit($offset, $limit);
- $found = $user->find();
- if ($found) {
- while ($user->fetch()) {
- try {
- testUser($filter, $user);
- } catch (Exception $e) {
- printfnq("ERROR testing user %s\n: %s", $user->nickname, $e->getMessage());
- }
- }
- $offset += $found;
- }
- } while ($found > 0);
- }
- function testUser($filter, $user) {
- printfnq("Testing user %s\n", $user->nickname);
- $profile = Profile::getKV('id', $user->id);
- $str = new ProfileNoticeStream($profile, $profile);
- $offset = 0;
- $limit = 100;
- do {
- $notice = $str->getNotices($offset, $limit);
- while ($notice->fetch()) {
- try {
- printfv("Testing notice %d...", $notice->id);
- $result = $filter->test($notice);
- Spam_score::save($notice, $result);
- printfv("%s\n", ($result->isSpam) ? "SPAM" : "HAM");
- } catch (Exception $e) {
- printfnq("ERROR testing notice %d: %s\n", $notice->id, $e->getMessage());
- }
- }
- $offset += $notice->N;
- } while ($notice->N > 0);
- }
- try {
- $filter = null;
- Event::handle('GetSpamFilter', array(&$filter));
- if (empty($filter)) {
- throw new Exception(_("No spam filter."));
- }
- if (have_option('a', 'all')) {
- testAllUsers($filter);
- } else {
- $user = getUser();
- testUser($filter, $user);
- }
- } catch (Exception $e) {
- print $e->getMessage()."\n";
- exit(1);
- }
|