Rudimentary scripts for training spam and ham from stdin

2 files changed, 58 insertions, 0 deletions

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()