12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- <?php
- define('INSTALLDIR', realpath(dirname(__FILE__) . '/../../..'));
- $shortoptions = 'i:n:t:';
- $longoptions = array('id=', 'nickname=', 'category=');
- $helptext = <<<END_OF_TRAINUSER_HELP
- trainuser.php [options]
- Train user activities against the spam filter
- -i --id ID of user to export
- -n --nickname nickname of the user to export
- -t --category Category; one of "spam" or "ham"
- END_OF_TRAINUSER_HELP;
- require_once INSTALLDIR.'/scripts/commandline.inc';
- function trainUser($filter, $user, $category) {
- printfnq("Training 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("Training notice %d...", $notice->id);
- $filter->trainOnError($notice, $category);
- $result = $filter->test($notice);
- $score = Spam_score::save($notice, $result);
- printfv("%s\n", ($result->isSpam) ? "SPAM" : "HAM");
- } catch (Exception $e) {
- printfnq("ERROR training notice %d\n: %s", $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."));
- }
- $user = getUser();
- $category = get_option_value('t', 'category');
- if ($category !== SpamFilter::HAM &&
- $category !== SpamFilter::SPAM) {
- throw new Exception(_("No such category."));
- }
- trainUser($filter, $user, $category);
- } catch (Exception $e) {
- print $e->getMessage()."\n";
- exit(1);
- }
|