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