elmgates/WeChatMsg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

统一管理导出文件路径

Browse Source
This commit is contained in:
SiYuan
2024-02-04 23:11:43 +08:00
parent 1fa8241aba
commit 4f868525dd
12 changed files with 111 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/DataBase/exporter.py
View File

@@ -8,9 +8,10 @@ import filecmp
from PyQt5.QtCore import pyqtSignal, QThread
from ..config import output_dir
from ..person import Me, Contact
os.makedirs('./data/聊天记录', exist_ok=True)
os.makedirs(os.path.join(output_dir, '聊天记录'), exist_ok=True)
def set_global_font(doc, font_name):
@@ -103,8 +104,8 @@ class ExporterBase(QThread):
self.last_timestamp = 0
self.time_range = time_range
self.messages = messages
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
makedirs(origin_docx_path)
origin_path = os.path.join(os.path.abspath('.'),output_dir,'聊天记录',self.contact.remark)
makedirs(origin_path)
def run(self):
self.export()

7
app/DataBase/exporter_csv.py
View File

@@ -3,14 +3,15 @@ import os
from app.DataBase import msg_db
from app.DataBase.exporter import ExporterBase
from app.config import output_dir
class CSVExporter(ExporterBase):
def to_csv(self):
print(f"【开始导出 CSV {self.contact.remark}】")
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
os.makedirs(origin_docx_path, exist_ok=True)
filename = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}/{self.contact.remark}_utf8.csv"
origin_path = os.path.join(os.path.abspath('.'),output_dir,'聊天记录',self.contact.remark)
os.makedirs(origin_path, exist_ok=True)
filename = os.path.join(origin_path,f"{self.contact.remark}_utf8.csv")
columns = ['localId', 'TalkerId', 'Type', 'SubType',
'IsSender', 'CreateTime', 'Status', 'StrContent',
'StrTime', 'Remark', 'NickName', 'Sender']

50
app/DataBase/exporter_docx.py
View File

@@ -13,6 +13,7 @@ from docxcompose.composer import Composer
from app.DataBase import msg_db, hard_link_db
from app.DataBase.exporter import ExporterBase, escape_js_and_html
from app.config import output_dir
from app.log import logger
from app.person import Me
from app.util.compress_content import parser_reply, share_card, music_share
@@ -66,28 +67,23 @@ class DocxExporter(ExporterBase):
p = content_cell.paragraphs[0]
p.paragraph_format.alignment = WD_PARAGRAPH_ALIGNMENT.RIGHT
doc.add_paragraph()
def image(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
type_ = message[2]
str_content = message[7]
str_time = message[8]
is_send = message[4]
BytesExtra = message[10]
timestamp = message[5]
is_chatroom = 1 if self.contact.is_chatroom else 0
avatar = self.get_avatar_path(is_send, message)
display_name = self.get_display_name(is_send, message)
avatar = self.get_avatar_path(is_send, message, True)
content = self.create_table(doc, is_send, avatar)
run = content.paragraphs[0].add_run()
str_content = escape_js_and_html(str_content)
image_path = hard_link_db.get_image(str_content, BytesExtra, thumb=True)
base_path = os.path.join(output_dir, '聊天记录', self.contact.remark, 'image')
if not os.path.exists(os.path.join(Me().wx_dir, image_path)):
image_thumb_path = hard_link_db.get_image(str_content, BytesExtra, thumb=False)
if not os.path.exists(os.path.join(Me().wx_dir, image_thumb_path)):
return
image_path = image_thumb_path
image_path = get_image_abs_path(image_path, base_path=f'/data/聊天记录/{self.contact.remark}/image')
image_path = get_image_abs_path(image_path, base_path=base_path)
try:
run.add_picture(image_path, height=shared.Inches(2))
doc.add_paragraph()
@@ -95,7 +91,6 @@ class DocxExporter(ExporterBase):
print("Error!image")
def audio(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
str_content = message[7]
str_time = message[8]
is_send = message[4]
@@ -131,7 +126,6 @@ class DocxExporter(ExporterBase):
doc.add_paragraph()
def file(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
bytesExtra = message[10]
str_time = message[8]
is_send = message[4]
@@ -196,7 +190,6 @@ class DocxExporter(ExporterBase):
doc.add_paragraph(str_content).alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
def video(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
type_ = message[2]
str_content = message[7]
str_time = message[8]
@@ -247,14 +240,14 @@ class DocxExporter(ExporterBase):
return content_cell
def music_share(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
is_send = message[4]
timestamp = message[5]
content = music_share(message[11])
music_path = ''
if content.get('audio_url') != '':
music_path = get_music_path(content.get('audio_url'), content.get('title'),
output_path=origin_docx_path + '/music')
output_path=origin_path + '/music')
if music_path != '':
music_path = f'./music/{os.path.basename(music_path)}'
music_path = music_path.replace('\\', '/')
@@ -263,7 +256,7 @@ class DocxExporter(ExporterBase):
display_name = self.get_display_name(is_send, message)
def share_card(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
is_send = message[4]
timestamp = message[5]
bytesExtra = message[10]
@@ -276,7 +269,7 @@ class DocxExporter(ExporterBase):
if card_data.get('thumbnail'):
thumbnail = os.path.join(Me().wx_dir, card_data.get('thumbnail'))
if os.path.exists(thumbnail):
shutil.copy(thumbnail, os.path.join(origin_docx_path, 'image', os.path.basename(thumbnail)))
shutil.copy(thumbnail, os.path.join(origin_path, 'image', os.path.basename(thumbnail)))
thumbnail = './image/' + os.path.basename(thumbnail)
else:
thumbnail = ''
@@ -284,22 +277,22 @@ class DocxExporter(ExporterBase):
if card_data.get('app_logo'):
app_logo = os.path.join(Me().wx_dir, card_data.get('app_logo'))
if os.path.exists(app_logo):
shutil.copy(app_logo, os.path.join(origin_docx_path, 'image', os.path.basename(app_logo)))
shutil.copy(app_logo, os.path.join(origin_path, 'image', os.path.basename(app_logo)))
app_logo = './image/' + os.path.basename(app_logo)
else:
app_logo = ''
def merge_docx(self, conRemark, n):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{conRemark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录')
all_file_path = []
for i in range(n):
file_name = f"{conRemark}{i}.docx"
all_file_path.append(origin_docx_path + '/' + file_name)
all_file_path.append(origin_path + '/' + file_name)
filename = f"{conRemark}.docx"
# print(all_file_path)
doc = docx.Document()
doc.save(origin_docx_path + '/' + filename)
master = docx.Document(origin_docx_path + '/' + filename)
doc.save(origin_path + '/' + filename)
master = docx.Document(origin_path + '/' + filename)
middle_new_docx = Composer(master)
num = 0
for word in all_file_path:
@@ -309,40 +302,41 @@ class DocxExporter(ExporterBase):
middle_new_docx.append(word_document)
num = num + 1
os.remove(word)
middle_new_docx.save(origin_docx_path + '/' + filename)
middle_new_docx.save(origin_path + '/' + filename)
def export(self):
print(f"【开始导出 DOCX {self.contact.remark}】")
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
# messages = self.messages
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
messages = msg_db.get_messages(self.contact.wxid, time_range=self.time_range)
Me().save_avatar(os.path.join(f"{origin_docx_path}/avatar/{Me().wxid}.png"))
Me().save_avatar(os.path.join(origin_path, 'avatar', f'{Me().wxid}.png'))
if self.contact.is_chatroom:
for message in messages:
if message[4]: # is_send
continue
try:
chatroom_avatar_path = f"{origin_docx_path}/avatar/{message[13].wxid}.png"
chatroom_avatar_path =os.path.join(origin_path, 'avatar', f'{message[13].wxid}.png')
message[13].save_avatar(chatroom_avatar_path)
except:
print(message)
pass
else:
self.contact.save_avatar(os.path.join(f"{origin_docx_path}/avatar/{self.contact.wxid}.png"))
self.contact.save_avatar(os.path.join(origin_path, 'avatar', f'{self.contact.wxid}.png'))
self.rangeSignal.emit(len(messages))
def newdoc():
nonlocal n, doc
doc = docx.Document()
doc.styles["Normal"].font.name = "Cambria"
doc.styles["Normal"]._element.rPr.rFonts.set(qn("w:eastAsia"), "宋体")
n += 1
doc = None
n = 0
index = 0
newdoc()
for index, message in enumerate(messages):
if index % 200 == 0 and index:
filename = os.path.join(origin_docx_path, f"{self.contact.remark}_{n}.docx")
filename = os.path.join(origin_path, f"{self.contact.remark}_{n}.docx")
doc.save(filename)
self.okSignal.emit(n)
newdoc()
@@ -374,7 +368,7 @@ class DocxExporter(ExporterBase):
print(f"【导出 DOCX {self.contact.remark}】{index}/{len(messages)}")
if index % 25:
print(f"【导出 DOCX {self.contact.remark}】{index + 1}/{len(messages)}")
filename = os.path.join(origin_docx_path, f"{self.contact.remark}_{n}.docx")
filename = os.path.join(origin_path, f"{self.contact.remark}_{n}.docx")
try:
# document.save(filename)
doc.save(filename)

65
app/DataBase/exporter_html.py
View File

@@ -8,6 +8,7 @@ from PyQt5.QtCore import pyqtSignal, QThread
from app.DataBase import msg_db, hard_link_db, media_msg_db
from app.DataBase.exporter import ExporterBase, escape_js_and_html
from app.config import output_dir
from app.log import logger
from app.person import Me
from app.util import path
@@ -44,7 +45,7 @@ class HtmlExporter(ExporterBase):
)
def image(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
base_path = os.path.join(output_dir, '聊天记录', self.contact.remark, 'image')
type_ = message[2]
str_content = message[7]
str_time = message[8]
@@ -56,13 +57,13 @@ class HtmlExporter(ExporterBase):
display_name = self.get_display_name(is_send, message)
str_content = escape_js_and_html(str_content)
image_path = hard_link_db.get_image(str_content, BytesExtra, up_dir=Me().wx_dir, thumb=False)
image_path = get_image_path(image_path, base_path=f'/data/聊天记录/{self.contact.remark}/image')
image_path = get_image_path(image_path, base_path=base_path)
doc.write(
f'''{{ type:{type_}, text: '{image_path}',is_send:{is_send},avatar_path:'{avatar}',timestamp:{timestamp},is_chatroom:{is_chatroom},displayname:'{display_name}'}},'''
)
def audio(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
str_content = message[7]
str_time = message[8]
is_send = message[4]
@@ -72,7 +73,7 @@ class HtmlExporter(ExporterBase):
avatar = self.get_avatar_path(is_send, message)
display_name = self.get_display_name(is_send, message)
try:
audio_path = media_msg_db.get_audio_path(msgSvrId, output_path=origin_docx_path + "/voice")
audio_path = media_msg_db.get_audio_path(msgSvrId, output_path=origin_path + "/voice")
audio_path = "./voice/" + os.path.basename(audio_path)
except:
logger.error(traceback.format_exc())
@@ -98,7 +99,7 @@ class HtmlExporter(ExporterBase):
)
def file(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
bytesExtra = message[10]
compress_content = message[11]
str_time = message[8]
@@ -107,7 +108,7 @@ class HtmlExporter(ExporterBase):
is_chatroom = 1 if self.contact.is_chatroom else 0
avatar = self.get_avatar_path(is_send, message)
display_name = self.get_display_name(is_send, message)
file_info = file(bytesExtra, compress_content, output_path=origin_docx_path + '/file')
file_info = file(bytesExtra, compress_content, output_path=origin_path + '/file')
if file_info.get('is_error') == False:
icon_path = None
for icon, extensions in icon_files.items():
@@ -169,7 +170,7 @@ class HtmlExporter(ExporterBase):
)
def video(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
type_ = message[2]
str_content = message[7]
str_time = message[8]
@@ -185,8 +186,8 @@ class HtmlExporter(ExporterBase):
image_path = path.get_relative_path(image_path, base_path=f'/data/聊天记录/{self.contact.remark}/image')
try:
# todo 网络图片问题
print(origin_docx_path + image_path[1:])
os.utime(origin_docx_path + image_path[1:], (timestamp, timestamp))
print(origin_path + image_path[1:])
os.utime(origin_path + image_path[1:], (timestamp, timestamp))
doc.write(
f'''{{ type:3, text: '{image_path}',is_send:{is_send},avatar_path:'{avatar}',timestamp:{timestamp},is_chatroom:{is_chatroom},displayname:'{display_name}'}},'''
)
@@ -200,9 +201,9 @@ class HtmlExporter(ExporterBase):
video_path = f'{Me().wx_dir}/{video_path}'
video_path = video_path.replace('\\', '/')
if os.path.exists(video_path):
new_path = origin_docx_path + '/video/' + os.path.basename(video_path)
new_path = origin_path + '/video/' + os.path.basename(video_path)
if not os.path.exists(new_path):
shutil.copy(video_path, os.path.join(origin_docx_path, 'video'))
shutil.copy(video_path, os.path.join(origin_path, 'video'))
os.utime(new_path, (timestamp, timestamp))
video_path = f'./video/{os.path.basename(video_path)}'
doc.write(
@@ -210,7 +211,7 @@ class HtmlExporter(ExporterBase):
)
def music_share(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
is_send = message[4]
timestamp = message[5]
content = music_share(message[11])
@@ -218,7 +219,7 @@ class HtmlExporter(ExporterBase):
if content.get('is_error') == False:
if content.get('audio_url') != '':
music_path = get_music_path(content.get('audio_url'), content.get('title'),
output_path=origin_docx_path + '/music')
output_path=origin_path + '/music')
if music_path != '':
music_path = f'./music/{os.path.basename(music_path)}'
music_path = music_path.replace('\\', '/')
@@ -231,7 +232,7 @@ class HtmlExporter(ExporterBase):
)
def share_card(self, doc, message):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
is_send = message[4]
timestamp = message[5]
bytesExtra = message[10]
@@ -244,7 +245,7 @@ class HtmlExporter(ExporterBase):
if card_data.get('thumbnail'):
thumbnail = os.path.join(Me().wx_dir, card_data.get('thumbnail'))
if os.path.exists(thumbnail):
shutil.copy(thumbnail, os.path.join(origin_docx_path, 'image', os.path.basename(thumbnail)))
shutil.copy(thumbnail, os.path.join(origin_path, 'image', os.path.basename(thumbnail)))
thumbnail = './image/' + os.path.basename(thumbnail)
else:
thumbnail = ''
@@ -252,7 +253,7 @@ class HtmlExporter(ExporterBase):
if card_data.get('app_logo'):
app_logo = os.path.join(Me().wx_dir, card_data.get('app_logo'))
if os.path.exists(app_logo):
shutil.copy(app_logo, os.path.join(origin_docx_path, 'image', os.path.basename(app_logo)))
shutil.copy(app_logo, os.path.join(origin_path, 'image', os.path.basename(app_logo)))
app_logo = './image/' + os.path.basename(app_logo)
else:
app_logo = card_data.get('app_logo')
@@ -296,7 +297,8 @@ class HtmlExporter(ExporterBase):
def export(self):
print(f"【开始导出 HTML {self.contact.remark}】")
messages = msg_db.get_messages(self.contact.wxid, time_range=self.time_range)
filename = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}/{self.contact.remark}.html"
filename = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark,
f'{self.contact.remark}.html')
file_path = './app/resources/data/template.html'
if not os.path.exists(file_path):
resource_dir = getattr(sys, '_MEIPASS', os.path.abspath(os.path.dirname(__file__)))
@@ -377,14 +379,14 @@ class OutputMedia(QThread):
self.contact = contact
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
messages = msg_db.get_messages_by_type(self.contact.wxid, 34)
for message in messages:
is_send = message[4]
msgSvrId = message[9]
try:
audio_path = media_msg_db.get_audio(
msgSvrId, output_path=origin_docx_path + "/voice"
msgSvrId, output_path=origin_path + "/voice"
)
except:
logger.error(traceback.format_exc())
@@ -406,13 +408,13 @@ class OutputEmoji(QThread):
self.contact = contact
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
messages = msg_db.get_messages_by_type(self.contact.wxid, 47)
for message in messages:
str_content = message[7]
try:
pass
# emoji_path = get_emoji(str_content, thumb=True, output_path=origin_docx_path + '/emoji')
# emoji_path = get_emoji(str_content, thumb=True, output_path=origin_path + '/emoji')
except:
logger.error(traceback.format_exc())
finally:
@@ -443,8 +445,9 @@ class OutputImage(QThread):
print("图片导出完成")
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
messages = msg_db.get_messages_by_type(self.contact.wxid, 3)
base_path = os.path.join(output_dir, '聊天记录', self.contact.remark, 'image')
for message in messages:
str_content = message[7]
BytesExtra = message[10]
@@ -461,10 +464,10 @@ class OutputImage(QThread):
continue
image_path = image_thumb_path
image_path = get_image(
image_path, base_path=f"/data/聊天记录/{self.contact.remark}/image"
image_path, base_path=base_path
)
try:
os.utime(origin_docx_path + image_path[1:], (timestamp, timestamp))
os.utime(origin_path + image_path[1:], (timestamp, timestamp))
except:
pass
except:
@@ -472,16 +475,6 @@ class OutputImage(QThread):
finally:
self.progressSignal.emit(1)
self.okSingal.emit(47)
# sublist_length = len(messages) // self.child_thread_num
# index = 0
# for i in range(0, len(messages), sublist_length):
# child_messages = messages[i:i + sublist_length]
# self.child_threads[index] = OutputImageChild(self.contact, child_messages)
# self.child_threads[index].okSingal.connect(self.count1)
# self.child_threads[index].progressSignal.connect(self.progressSignal)
# self.child_threads[index].start()
# print('开启一个新线程')
# index += 1
class OutputImageChild(QThread):
@@ -494,7 +487,7 @@ class OutputImageChild(QThread):
self.messages = messages
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
for message in self.messages:
str_content = message[7]
BytesExtra = message[10]
@@ -514,7 +507,7 @@ class OutputImageChild(QThread):
image_path, base_path=f"/data/聊天记录/{self.contact.remark}/image"
)
try:
os.utime(origin_docx_path + image_path[1:], (timestamp, timestamp))
os.utime(origin_path + image_path[1:], (timestamp, timestamp))
except:
pass
except:

7
app/DataBase/exporter_txt.py
View File

@@ -2,6 +2,7 @@ import os
from app.DataBase import msg_db
from app.DataBase.exporter import ExporterBase
from app.config import output_dir
from app.util.compress_content import parser_reply, share_card
@@ -111,9 +112,9 @@ class TxtExporter(ExporterBase):
def export(self):
# 实现导出为txt的逻辑
print(f"【开始导出 TXT {self.contact.remark}】")
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
os.makedirs(origin_docx_path, exist_ok=True)
filename = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}/{self.contact.remark}.txt"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录', self.contact.remark)
os.makedirs(origin_path, exist_ok=True)
filename = os.path.join(origin_path, self.contact.remark+'.txt')
messages = msg_db.get_messages(self.contact.wxid, time_range=self.time_range)
total_steps = len(messages)
with open(filename, mode='w', newline='', encoding='utf-8') as f:

37
app/DataBase/output.py
View File

@@ -15,13 +15,14 @@ from app.DataBase.exporter_docx import DocxExporter
from app.DataBase.exporter_html import HtmlExporter
from app.DataBase.exporter_txt import TxtExporter
from app.DataBase.hard_link import decodeExtraBuf
from app.config import output_dir
from .package_msg import PackageMsg
from ..DataBase import media_msg_db, hard_link_db, micro_msg_db, msg_db
from ..log import logger
from ..person import Me
from ..util.image import get_image
os.makedirs('./data/聊天记录', exist_ok=True)
os.makedirs(os.path.join(output_dir, '聊天记录'), exist_ok=True)
class Output(QThread):
@@ -79,8 +80,9 @@ class Output(QThread):
导出全部聊天记录到CSV
@return:
"""
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/"
os.makedirs(origin_docx_path, exist_ok=True)
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录')
os.makedirs(origin_path, exist_ok=True)
filename = QFileDialog.getSaveFileName(None, "save file", os.path.join(os.getcwd(), 'messages.csv'),
"csv files (*.csv);;all files(*.*)")
if not filename[0]:
@@ -167,11 +169,11 @@ class Output(QThread):
def merge_docx(self, n):
conRemark = self.contact.remark
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{conRemark}"
filename = f"{origin_docx_path}/{conRemark}_{n}.docx"
origin_path = os.path.join(os.path.abspath('.'), output_dir, '聊天记录',conRemark)
filename = f"{origin_path}/{conRemark}_{n}.docx"
if n == 10086:
# self.document.append(self.document)
file = os.path.join(origin_docx_path, f'{conRemark}.docx')
file = os.path.join(origin_path, f'{conRemark}.docx')
try:
self.document.save(file)
except PermissionError:
@@ -184,7 +186,7 @@ class Output(QThread):
os.remove(filename)
if n % 50 == 0:
# self.document.append(self.document)
file = os.path.join(origin_docx_path, f'{conRemark}-{n//50}.docx')
file = os.path.join(origin_path, f'{conRemark}-{n // 50}.docx')
try:
self.document.save(file)
except PermissionError:
@@ -308,13 +310,13 @@ class OutputMedia(QThread):
self.time_range = time_range
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'),output_dir,'聊天记录',self.contact.remark)
messages = msg_db.get_messages_by_type(self.contact.wxid, 34, time_range=self.time_range)
for message in messages:
is_send = message[4]
msgSvrId = message[9]
try:
audio_path = media_msg_db.get_audio(msgSvrId, output_path=origin_docx_path + "/voice")
audio_path = media_msg_db.get_audio(msgSvrId, output_path=origin_path + "/voice")
except:
logger.error(traceback.format_exc())
finally:
@@ -335,13 +337,13 @@ class OutputEmoji(QThread):
self.time_range = time_range
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'),output_dir,'聊天记录',self.contact.remark)
messages = msg_db.get_messages_by_type(self.contact.wxid, 47, time_range=self.time_range)
for message in messages:
str_content = message[7]
try:
pass
# emoji_path = get_emoji(str_content, thumb=True, output_path=origin_docx_path + '/emoji')
# emoji_path = get_emoji(str_content, thumb=True, output_path=origin_path + '/emoji')
except:
logger.error(traceback.format_exc())
finally:
@@ -372,17 +374,18 @@ class OutputImage(QThread):
print('图片导出完成')
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'),output_dir,'聊天记录',self.contact.remark)
messages = msg_db.get_messages_by_type(self.contact.wxid, 3, time_range=self.time_range)
base_path = os.path.join(output_dir,'聊天记录',self.contact.remark,'image')
for message in messages:
str_content = message[7]
BytesExtra = message[10]
timestamp = message[5]
try:
image_path = hard_link_db.get_image(str_content, BytesExtra, up_dir=Me().wx_dir,thumb=False)
image_path = get_image(image_path, base_path=f'/data/聊天记录/{self.contact.remark}/image')
image_path = hard_link_db.get_image(str_content, BytesExtra, up_dir=Me().wx_dir, thumb=False)
image_path = get_image(image_path, base_path=base_path)
try:
os.utime(origin_docx_path + image_path[1:], (timestamp, timestamp))
os.utime(origin_path + image_path[1:], (timestamp, timestamp))
except:
pass
except:
@@ -403,7 +406,7 @@ class OutputImageChild(QThread):
self.time_range = time_range
def run(self):
origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
origin_path = os.path.join(os.path.abspath('.'),output_dir,'聊天记录',self.contact.remark)
for message in self.messages:
str_content = message[7]
BytesExtra = message[10]
@@ -417,7 +420,7 @@ class OutputImageChild(QThread):
image_path = image_thumb_path
image_path = get_image(image_path, base_path=f'/data/聊天记录/{self.contact.remark}/image')
try:
os.utime(origin_docx_path + image_path[1:], (timestamp, timestamp))
os.utime(origin_path + image_path[1:], (timestamp, timestamp))
except:
pass
except: