scripts/testuser.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2012 StatusNet, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 define('INSTALLDIR', realpath(dirname(__FILE__) . '/../../../..'));
  21
  22 $shortoptions = 'i:n:a';
  23 $longoptions = array('id=', 'nickname=', 'all');
  24
  25 $helptext = <<<END_OF_TESTUSER_HELP
  26 testuser.php [options]
  27 Test user activities against the spam filter
  28
  29   -i --id       ID of user to export
  30   -n --nickname nickname of the user to export
  31   -a --all      All users
  32 END_OF_TESTUSER_HELP;
  33
  34 require_once INSTALLDIR.'/scripts/commandline.inc';
  35
  36 function testAllUsers() {
  37     $found = false;
  38     $offset = 0;
  39     $limit  = 1000;
  40
  41     do {
  42
  43         $user = new User();
  44         $user->orderBy('created');
  45         $user->limit($offset, $limit);
  46
  47         $found = $user->find();
  48
  49         if ($found) {
  50             while ($user->fetch()) {
  51                 testUser($filter, $user);
  52             }
  53             $offset += $found;
  54         }
  55
  56     } while ($found > 0);
  57 }
  58
  59 function testUser($filter, $user) {
  60
  61     $profile = $user->getProfile();
  62
  63     $str = new ProfileNoticeStream($profile, $profile);
  64
  65     $offset = 0;
  66     $limit  = 100;
  67
  68     do {
  69         $notice = $str->getNotices($offset, $limit);
  70         while ($notice->fetch()) {
  71             print "Testing notice " . $notice->id . "...";
  72             $result = $filter->test($notice);
  73             Spam_score::save($notice, $result);
  74             print (($result->isSpam) ? "SPAM" : "HAM")."\n";
  75         }
  76         $offset += $notice->N;
  77     } while ($notice->N > 0);
  78 }
  79
  80 try {
  81     $filter = null;
  82     Event::handle('GetSpamFilter', array(&$filter));
  83     if (empty($filter)) {
  84         throw new Exception(_("No spam filter."));
  85     }
  86     if (get_option('a', 'all')) {
  87         testAllUsers($filter);
  88     } else {
  89         $user = getUser();
  90         testUser($filter, $user);
  91     }
  92 } catch (Exception $e) {
  93     print $e->getMessage()."\n";
  94     exit(1);
  95 }