mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 14:31:18 +08:00
97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
import os.path
|
|
import re
|
|
import sqlite3
|
|
import threading
|
|
|
|
DB = []
|
|
cursor = []
|
|
msg_root_path = "./app/Database/Msg/"
|
|
lock = threading.Lock()
|
|
# misc_path = './Msg/Misc.db'
|
|
if os.path.exists(msg_root_path):
|
|
for root, dirs, files in os.walk(msg_root_path):
|
|
for file in files:
|
|
if re.match('^MSG[0-9]+\.db$', file):
|
|
# print('ok', file)
|
|
msg_path = os.path.join(msg_root_path, file)
|
|
DB0 = sqlite3.connect(msg_path, check_same_thread=False)
|
|
# '''创建游标'''
|
|
cursor0 = DB0.cursor()
|
|
DB.append(DB0)
|
|
cursor.append(cursor0)
|
|
|
|
|
|
def init_database():
|
|
global DB
|
|
global cursor
|
|
if not DB:
|
|
if os.path.exists(msg_root_path):
|
|
for root, dirs, files in os.walk(msg_root_path):
|
|
for file in files:
|
|
# print(file)
|
|
if re.match('^MSG[0-9]+\.db$', file):
|
|
print('ok', file)
|
|
msg_path = os.path.join(msg_root_path, file)
|
|
DB0 = sqlite3.connect(msg_path, check_same_thread=False)
|
|
# '''创建游标'''
|
|
cursor0 = DB0.cursor()
|
|
DB.append(DB0)
|
|
cursor.append(cursor0)
|
|
|
|
|
|
def get_messages(username_):
|
|
sql = '''
|
|
select localId,TalkerId,Type,SubType,IsSender,CreateTime,Status,StrContent,strftime('%Y-%m-%d %H:%M:%S',CreateTime,'unixepoch','localtime') as StrTime
|
|
from MSG
|
|
where StrTalker=?
|
|
order by CreateTime
|
|
'''
|
|
result = []
|
|
for cur in cursor:
|
|
cur.execute(sql, [username_])
|
|
result_ = cur.fetchall()
|
|
# print(len(result))
|
|
result += result_
|
|
result.sort(key=lambda x: x[5])
|
|
return result
|
|
|
|
|
|
def get_message_by_num(username_, n):
|
|
sql = '''
|
|
select localId,TalkerId,Type,SubType,IsSender,CreateTime,Status,StrContent,strftime('%Y-%m-%d %H:%M:%S',CreateTime,'unixepoch','localtime') as StrTime
|
|
from MSG
|
|
where StrTalker=?
|
|
order by CreateTime
|
|
limit 10
|
|
'''
|
|
result = []
|
|
try:
|
|
lock.acquire(True)
|
|
for cur in cursor:
|
|
cur = cursor[-1]
|
|
cur.execute(sql, [username_])
|
|
result_ = cur.fetchall()
|
|
result += result_
|
|
return result_
|
|
finally:
|
|
lock.release()
|
|
result.sort(key=lambda x: x[5])
|
|
return result
|
|
|
|
|
|
def close():
|
|
for db in DB:
|
|
db.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
msg_root_path = './Msg/'
|
|
init_database()
|
|
|
|
# username = 'wxid_0o18ef858vnu22'
|
|
# result = get_messages(username)
|
|
# pprint(result)
|
|
# pprint(len(result))
|
|
result = get_message_by_num('wxid_0o18ef858vnu22', 0)
|
|
print(result)
|