update docs about sfs/avatar.index

This commit is contained in:
Yuxin Wu
2021-02-02 11:00:48 -08:00
parent 541f6aa5ee
commit 5b7d293b06
2 changed files with 6 additions and 1 deletions
+1 -1
View File
@@ -17,7 +17,7 @@ def get_args():
parser.add_argument('name', help='name of contact')
parser.add_argument('--output', help='output html file', default='output.html')
parser.add_argument('--db', default='decoded.db', help='path to decoded database')
parser.add_argument('--avt', default='avatar.index', help='path to avatar.index file')
parser.add_argument('--avt', default='avatar.index', help='path to avatar.index file that only exists in old version of wechat')
parser.add_argument('--res', default='resource', help='reseource directory')
args = parser.parse_args()
return args
+5
View File
@@ -30,6 +30,11 @@ JPEG_QUALITY = 50
class Resource(object):
""" multimedia resources in chat"""
def __init__(self, parser, res_dir, avt_db):
"""
Args:
res_dir: path to the resource directory
avt_db: "avatar.index" file that only exists in old versions of wechat
"""
def check(subdir):
dir_to_check = os.path.join(res_dir, subdir)
assert os.path.isdir(dir_to_check), f"No such directory: {dir_to_check}"