From: Andrew Dolgov <noreply@fakecake.org>
Date: Thu, 18 Jun 2015 07:56:42 +0000 (+0300)
Subject: bayes: fix check for already categorized articles
X-Git-Tag: 16.3~283
X-Git-Url: https://git.wh0rd.org/?p=tt-rss.git;a=commitdiff_plain;h=5df13f31ef5642e583d9fb8755bd3ed78e16f594

bayes: fix check for already categorized articles
---

diff --git a/plugins/af_sort_bayes/init.php b/plugins/af_sort_bayes/init.php
index d2e659ca..0bdfafa1 100644
--- a/plugins/af_sort_bayes/init.php
+++ b/plugins/af_sort_bayes/init.php
@@ -263,12 +263,17 @@ class Af_Sort_Bayes extends Plugin {
 	function hook_article_filter($article) {
 		$owner_uid = $article["owner_uid"];
 
-		$nbs = new NaiveBayesianStorage($owner_uid);
-		$nb = new NaiveBayesianNgram($nbs, 3);
+		// guid already includes owner_uid so we don't need to include it
+		$result = $this->dbh->query("SELECT id FROM {$this->sql_prefix}_references WHERE
+			document_id = '" . $this->dbh->escape_string($article['guid_hashed']) . "'");
 
-		$ref = $nbs->getReference($article["guid"], false);
+		if (db_num_rows($result) != 0) {
+			_debug("bayes: article already categorized");
+			return $article;
+		}
 
-		if (isset($ref["category_id"])) return $article; // already categorized
+		$nbs = new NaiveBayesianStorage($owner_uid);
+		$nb = new NaiveBayesianNgram($nbs, 3);
 
 		$categories = $nbs->getCategories();