From 82bdf837d993fdf8db5f6e8fa6169be964adb687 Mon Sep 17 00:00:00 2001
From: Mike Pirnat <mpirnat@gmail.com>
Date: Wed, 8 Aug 2012 23:10:18 -0400
Subject: Rudimentary scripts for training spam and ham from stdin

---
 train_ham.py  | 29 +++++++++++++++++++++++++++++
 train_spam.py | 29 +++++++++++++++++++++++++++++
 2 files changed, 58 insertions(+)
 create mode 100644 train_ham.py
 create mode 100644 train_spam.py

diff --git a/train_ham.py b/train_ham.py
new file mode 100644
index 0000000..4989c7c
--- /dev/null
+++ b/train_ham.py
@@ -0,0 +1,29 @@
+"""
+A rudimentary way to train additional ham into our pickle file.
+
+Example usage:
+
+$ python train_spam.py /path/to/spam.pkl
+blah blah blah^D
+"""
+import sys
+from ConfigParser import SafeConfigParser as ConfigParser, \
+        NoSectionError, NoOptionError
+from spambayes.storage import PickledClassifier
+
+
+config = ConfigParser()
+with open("/etc/email_gateway.cfg") as fp:
+    config.readfp(fp)
+
+
+def main():
+    pickle_filename = sys.argv[-1]
+    bayes = PickledClassifier(pickle_filename)
+    message = sys.stdin.readlines()
+    bayes.learn(message, False)
+    bayes.store()
+
+
+if __name__ == '__main__':
+    main()
diff --git a/train_spam.py b/train_spam.py
new file mode 100644
index 0000000..c1f2065
--- /dev/null
+++ b/train_spam.py
@@ -0,0 +1,29 @@
+"""
+A rudimentary way to train additional spam into our pickle file.
+
+Example usage:
+
+$ python train_spam.py /path/to/spam.pkl
+blah blah blah^D
+"""
+import sys
+from ConfigParser import SafeConfigParser as ConfigParser, \
+        NoSectionError, NoOptionError
+from spambayes.storage import PickledClassifier
+
+
+config = ConfigParser()
+with open("/etc/email_gateway.cfg") as fp:
+    config.readfp(fp)
+
+
+def main():
+    pickle_filename = sys.argv[-1]
+    bayes = PickledClassifier(pickle_filename)
+    message = sys.stdin.readlines()
+    bayes.learn(message, True)
+    bayes.store()
+
+
+if __name__ == '__main__':
+    main()
-- 
cgit v1.2.3