Add option --verbose

judge_message

@@ -32,7 +32,7 @@ def add_message(msg):
                     msg.get_payload(decode=True)
                         .decode(charset, errors='replace'))
 
-def extract_features(msgid):
+def extract_features(msgid, verbose):
     db = psycopg2.connect("dbname=bayes")
     csr = db.cursor(cursor_factory=ppe.DictCursor)
     csr.execute(
@@ -89,6 +89,7 @@ def extract_features(msgid):
     p1 = 1
     p2 = 1
     for i in interesting_evidence:
+        if verbose:
             print("#", i["spam_prob"], i["type"], i["length"], i["feature"], sep="\t")
         csr.execute(
             "insert into used_evidence(message, spam_prob, type, length, feature) values(%s, %s, %s, %s, %s)",
@@ -102,6 +103,7 @@ def extract_features(msgid):
 
 def main():
     ap = argparse.ArgumentParser()
+    ap.add_argument('--verbose', action='store_true')
     ap.add_argument('file', nargs='?')
     args = ap.parse_args()
     if args.file:
@@ -111,7 +113,7 @@ def main():
     parser = email.parser.BytesParser(policy=email.policy.default)
     msg = parser.parse(fh)
     add_message(msg)
-    p = extract_features(msg["Message-Id"])
+    p = extract_features(msg["Message-Id"], args.verbose)
     print(p, "spam" if p > 0.5 else "ham")
 
 main()