plugins/ActivitySpam/scripts/trainuser.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:t:';
  23 $longoptions = array('id=', 'nickname=', 'category=');
  24
  25 $helptext = <<<END_OF_TRAINUSER_HELP
  26 trainuser.php [options]
  27 Train 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   -t --category Category; one of "spam" or "ham"
  32
  33 END_OF_TRAINUSER_HELP;
  34
  35 require_once INSTALLDIR.'/scripts/commandline.inc';
  36
  37 function trainUser($filter, $user, $category) {
  38
  39     printfnq("Training user %s\n", $user->nickname);
  40
  41     $profile = Profile::getKV('id', $user->id);
  42
  43     $str = new ProfileNoticeStream($profile, $profile);
  44
  45     $offset = 0;
  46     $limit  = 100;
  47
  48     do {
  49         $notice = $str->getNotices($offset, $limit);
  50         while ($notice->fetch()) {
  51             try {
  52                 printfv("Training notice %d...", $notice->id);
  53                 $filter->trainOnError($notice, $category);
  54                 $result = $filter->test($notice);
  55                 $score = Spam_score::save($notice, $result);
  56                 printfv("%s\n", ($result->isSpam) ? "SPAM" : "HAM");
  57             } catch (Exception $e) {
  58                 printfnq("ERROR training notice %d\n: %s", $notice->id, $e->getMessage());
  59             }
  60         }
  61         $offset += $notice->N;
  62     } while ($notice->N > 0);
  63 }
  64
  65 try {
  66     $filter = null;
  67     Event::handle('GetSpamFilter', array(&$filter));
  68     if (empty($filter)) {
  69         throw new Exception(_("No spam filter."));
  70     }
  71     $user = getUser();
  72     $category = get_option_value('t', 'category');
  73     if ($category !== SpamFilter::HAM &&
  74         $category !== SpamFilter::SPAM) {
  75         throw new Exception(_("No such category."));
  76     }
  77     trainUser($filter, $user, $category);
  78 } catch (Exception $e) {
  79     print $e->getMessage()."\n";
  80     exit(1);
  81 }