cyonjan/PyWxDump
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: specify message type

Browse Source
This commit is contained in:
Zheng Huang 2024-05-22 20:21:45 +08:00
parent 10e62bb88c
commit 18540f4c24
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pywxdump/dbpreprocess/parsingMSG.py
View File

@ -247,17 +247,23 @@ class ParsingMSG(DatabaseBase):
"room_name": StrTalker, "content": content, "CreateTime": CreateTime, "id": id} "room_name": StrTalker, "content": content, "CreateTime": CreateTime, "id": id}
return row_data return row_data
def msg_list(self, wxid="", start_index=0, page_size=500): def msg_list(self, wxid="", start_index=0, page_size=500, msg_type: str = ""):
if wxid: if wxid:
sql = ( sql = (
"SELECT localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType,CreateTime,MsgSvrID,DisplayContent,CompressContent,BytesExtra,ROW_NUMBER() OVER (ORDER BY CreateTime ASC) AS id " "SELECT localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType,CreateTime,MsgSvrID,DisplayContent,CompressContent,BytesExtra,ROW_NUMBER() OVER (ORDER BY CreateTime ASC) AS id "
"FROM MSG WHERE StrTalker=? " "FROM MSG WHERE StrTalker=? "
"ORDER BY CreateTime ASC LIMIT ?,?") "ORDER BY CreateTime ASC LIMIT ?,?")
if msg_type:
sql = sql.replace("ORDER BY CreateTime ASC LIMIT ?,?",
f"AND Type={msg_type} ORDER BY CreateTime ASC LIMIT ?,?")
result1 = self.execute_sql(sql, (wxid, start_index, page_size)) result1 = self.execute_sql(sql, (wxid, start_index, page_size))
else: else:
sql = ( sql = (
"SELECT localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType,CreateTime,MsgSvrID,DisplayContent,CompressContent,BytesExtra,ROW_NUMBER() OVER (ORDER BY CreateTime ASC) AS id " "SELECT localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType,CreateTime,MsgSvrID,DisplayContent,CompressContent,BytesExtra,ROW_NUMBER() OVER (ORDER BY CreateTime ASC) AS id "
"FROM MSG ORDER BY CreateTime ASC LIMIT ?,?") "FROM MSG ORDER BY CreateTime ASC LIMIT ?,?")
if msg_type:
sql = sql.replace("ORDER BY CreateTime ASC LIMIT ?,?",
f"AND Type={msg_type} ORDER BY CreateTime ASC LIMIT ?,?")
result1 = self.execute_sql(sql, (start_index, page_size)) result1 = self.execute_sql(sql, (start_index, page_size))
if not result1: if not result1:
return [], [] return [], []