valvin
/
ttrss
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add button to clear bayes database

This commit is contained in:
Andrew Dolgov 2015-06-17 16:50:21 +03:00
parent b02e8bc8f2
commit 3c43def90f
2 changed files with 60 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
plugins/af_sort_bayes/init.js
View File

@ -15,3 +15,23 @@ function bayesTrain(id, train_up) {
}
}
function bayesClearDatabase() {
try {
if (confirm(__("Clear classifier database?"))) {
var query = "backend.php?op=pluginhandler&plugin=af_sort_bayes&method=clearDatabase";
new Ajax.Request("backend.php", {
parameters: query,
onComplete: function (transport) {
notify(transport.responseText);
}
});
}
} catch (e) {
exception_error("showTrgmRelated", e);
}
}

42
plugins/af_sort_bayes/init.php
View File

@ -6,6 +6,7 @@ class Af_Sort_Bayes extends Plugin {
private $filters = array();
private $dbh;
private $score_modifier = 50;
private $sql_prefix = "ttrss_plugin_af_sort_bayes";
function about() {
return array(1.0,
@ -71,6 +72,10 @@ class Af_Sort_Bayes extends Plugin {
return file_get_contents(__DIR__ . "/init.js");
}
function get_prefs_js() {
return file_get_contents(__DIR__ . "/init.js");
}
function hook_article_button($line) {
return "<img src=\"plugins/af_sort_bayes/thumb_up.png\"
style=\"cursor : pointer\" style=\"cursor : pointer\"
@ -84,7 +89,7 @@ class Af_Sort_Bayes extends Plugin {
}
function init_database() {
$prefix = "ttrss_plugin_af_sort_bayes";
$prefix = $this->sql_prefix;
// TODO there probably should be a way for plugins to determine their schema version to upgrade tables
@ -163,7 +168,27 @@ class Af_Sort_Bayes extends Plugin {
function hook_prefs_tab($args) {
if ($args != "prefPrefs") return;
print "<div dojoType=\"dijit.layout.AccordionPane\" title=\"".__('af_sort_bayes')."\">";
print "<div dojoType=\"dijit.layout.AccordionPane\" title=\"".__('Bayesian classifier (af_sort_bayes)')."\">";
$result = $this->dbh->query("SELECT category, probability, word_count FROM {$this->sql_prefix}_categories WHERE owner_uid = " . $_SESSION["uid"]);
print "<table>";
print "<tr><th>Category</th><th>Probability</th><th>Word count</th></tr>";
while ($line = $this->dbh->fetch_assoc($result)) {
print "<tr>";
foreach ($line as $k => $v) {
if ($k == "probability") $v = sprintf("%.3f", $v);
print "<td>$v</td>";
}
print "</tr>";
}
print "</table>";
print "<button dojoType=\"dijit.form.Button\" onclick=\"return bayesClearDatabase()\">".
__('Clear database')."</button> ";
//
@ -224,6 +249,19 @@ class Af_Sort_Bayes extends Plugin {
}
function clearDatabase() {
$prefix = $this->sql_prefix;
$this->dbh->query("BEGIN");
$this->dbh->query("DELETE FROM ${prefix}_references WHERE owner_uid = " . $_SESSION["uid"]);
$this->dbh->query("DELETE FROM ${prefix}_wordfreqs WHERE owner_uid = " . $_SESSION["uid"]);
$this->dbh->query("COMMIT");
$nbs = new NaiveBayesianStorage($_SESSION["uid"]);
$nb = new NaiveBayesian($nbs);
$nb->updateProbabilities();
}
function api_version() {
return 2;
}