wechat-dump/dump-msg.py

#!/usr/bin/env python2
# -*- coding: UTF-8 -*-
# File: dump-msg.py
# Date: Mon May 25 15:23:05 2015 +0800
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>

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 decrypted_database.db> <output_dir>".format(sys.argv[0]))

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

parser = WeChatDBParser(db_file)

for name, msgs in parser.msgs_by_chat.iteritems():
    print u"Writing msgs for {}".format(name)
    safe_name = safe_filename(name)
    with open(os.path.join(output_dir, safe_name + '.txt'), 'w') as f:
        for m in msgs:
            print >> f, m