fix bug, add avt_db support

This commit is contained in:
Sword York
2015-10-09 10:21:41 +08:00
parent c156a3142b
commit 4eae9cd7c1
2 changed files with 10 additions and 9 deletions
+7 -6
View File
@@ -5,8 +5,8 @@
# Author: Yuxin Wu <[email protected]>
import sys
if len(sys.argv) != 5:
sys.exit("Usage: {0} <path to decoded_database.db> <path to resource> <name> <output html>".format(sys.argv[0]))
if len(sys.argv) != 6:
sys.exit("Usage: {0} <path to decoded_database.db> <avt_db> <path to resource> <name> <output html>".format(sys.argv[0]))
from common.textutil import ensure_unicode
from wechat.parser import WeChatDBParser
@@ -14,12 +14,13 @@ from wechat.res import Resource
from wechat.render import HTMLRender
db_file = sys.argv[1]
resource_dir = sys.argv[2]
name = ensure_unicode(sys.argv[3])
output_file = sys.argv[4]
avt_db = sys.argv[2]
resource_dir = sys.argv[3]
name = ensure_unicode(sys.argv[4])
output_file = sys.argv[5]
parser = WeChatDBParser(db_file)
res = Resource(resource_dir)
res = Resource(resource_dir, avt_db)
try:
msgs = parser.msgs_by_talker[name]