debug_compare.py
1.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os, shutil, sys
sys.path.insert(0, '/data/data/com.termux/files/home/local-testkit')
import hermes_wechat_daemon as d
d.load_config()
src = d.config.get('db_src_path')
# Test copy_db
work1 = '/data/data/com.termux/files/home/hermes_work1'
os.makedirs(work1, exist_ok=True)
copied = d.copy_db(src, work1)
print('copy_db:', copied)
for f in copied:
print(' ', f, os.path.getsize(f))
# Test shutil.copy
work2 = '/data/data/com.termux/files/home/hermes_work2'
if os.path.exists(work2):
shutil.rmtree(work2)
os.makedirs(work2, exist_ok=True)
dst = os.path.join(work2, 'EnMicroMsg.db')
shutil.copy(src, dst)
print('shutil:', dst, os.path.getsize(dst))
# Test decrypt with copy_db file
plain1 = '/data/data/com.termux/files/home/test1.db'
if os.path.exists(plain1):
os.remove(plain1)
ok1 = d.decrypt_db(os.path.join(work1, 'EnMicroMsg.db'), plain1, d.config.get('sqlcipher_key'), 1024, 1)
print('copy_db decrypt=', ok1, os.path.exists(plain1))
# Test decrypt with shutil file
plain2 = '/data/data/com.termux/files/home/test2.db'
if os.path.exists(plain2):
os.remove(plain2)
ok2 = d.decrypt_db(dst, plain2, d.config.get('sqlcipher_key'), 1024, 1)
print('shutil decrypt=', ok2, os.path.exists(plain2))