rewrite with python3

dump-msg.py

@@ -1,38 +1,40 @@
-#!/usr/bin/env python2
+#!/usr/bin/env python3
 # -*- coding: UTF-8 -*-
-# File: dump-msg.py
-# Date: Mon May 25 15:23:05 2015 +0800
-# Author: Yuxin Wu
 
+import logging
 from wechat.parser import WeChatDBParser
 from common.textutil import safe_filename
 import sys, os
 
-if len(sys.argv) != 3:
-    sys.exit("Usage: {0} <path to decoded_database.db> <output_dir>".format(sys.argv[0]))
+logger = logging.getLogger("wechat")
 
-db_file = sys.argv[1]
-output_dir = sys.argv[2]
-try:
-    os.mkdir(output_dir)
-except:
-    pass
-if not os.path.isdir(output_dir):
-    sys.exit("Error creating directory {}".format(output_dir))
+if __name__ == '__main__':
+    if len(sys.argv) != 3:
+        sys.exit("Usage: {0} <path to decoded_database.db> <output_dir>".format(sys.argv[0]))
 
-parser = WeChatDBParser(db_file)
+    db_file = sys.argv[1]
+    output_dir = sys.argv[2]
+    try:
+        os.mkdir(output_dir)
+    except:
+        pass
+    if not os.path.isdir(output_dir):
+        sys.exit("Error creating directory {}".format(output_dir))
 
-for chatid, msgs in parser.msgs_by_chat.iteritems():
-    name = parser.contacts[chatid]
-    if len(name) == 0:
-        print u"Chat {} doesn't have a valid display name".format(chatid)
-        name = str(id(chatid))
-    print u"Writing msgs for {}".format(name)
-    safe_name = safe_filename(name)
-    outf = os.path.join(output_dir, safe_name + '.txt')
-    if os.path.isfile(outf):
-        print(u"File {} exists! Skip contact {}".format(outf, name))
-        continue
-    with open(outf, 'w') as f:
-        for m in msgs:
-            print >> f, m
+    parser = WeChatDBParser(db_file)
+
+    for chatid, msgs in parser.msgs_by_chat.items():
+        name = parser.contacts[chatid]
+        if len(name) == 0:
+            logger.info(f"Chat {chatid} doesn't have a valid display name.")
+            name = str(id(chatid))
+        logger.info(f"Writing msgs for {name}")
+        safe_name = safe_filename(name)
+        outf = os.path.join(output_dir, safe_name + '.txt')
+        if os.path.isfile(outf):
+            logger.info(f"File {outf} exists! Skip contact {name}")
+            continue
+        with open(outf, 'w') as f:
+            for m in msgs:
+                f.write(str(m))
+                f.write("\n")