mirror of
https://github.com/LibraHp/GetQzonehistory.git
synced 2024-12-29 07:29:40 +00:00
完成导出功能
This commit is contained in:
LibraHp_0928
parent
28f88e2c3d
commit
75d91f20e2
99
main.py
99
main.py
@ -8,6 +8,8 @@ import json
|
||||
import threading
|
||||
from bs4 import BeautifulSoup
|
||||
from collections import Counter
|
||||
import os
|
||||
import pandas as pd
|
||||
|
||||
# 初始化所有消息列表
|
||||
all_messages = []
|
||||
@ -171,7 +173,7 @@ class PaginatedContainer(ft.UserControl):
|
||||
items=[
|
||||
ft.PopupMenuItem(text="导出为JSON", on_click=self.export_json),
|
||||
ft.PopupMenuItem(text="导出为Excel", on_click=self.export_excel),
|
||||
ft.PopupMenuItem(text="导出为HTML", on_click=self.export_html),
|
||||
# ft.PopupMenuItem(text="导出为HTML", on_click=self.export_html),
|
||||
ft.PopupMenuItem(text="导出为Markdown", on_click=self.export_markdown),
|
||||
]
|
||||
)
|
||||
@ -224,16 +226,93 @@ class PaginatedContainer(ft.UserControl):
|
||||
print("请输入有效的页码。")
|
||||
|
||||
def export_json(self,e):
|
||||
print("Exporting JSON...")
|
||||
json_data = []
|
||||
for item in self.data:
|
||||
if isinstance(item, User):
|
||||
json_data.append({
|
||||
"username": item.username,
|
||||
"uin": item.uin,
|
||||
"link": item.link,
|
||||
"avatar_url": item.avatar_url
|
||||
})
|
||||
elif isinstance(item, Message):
|
||||
json_data.append({
|
||||
"username": item.user.username,
|
||||
"time": str(item.time),
|
||||
"content": item.content,
|
||||
"images": item.images,
|
||||
"comment": item.comment.content if item.comment else None,
|
||||
"avatar_url": item.user.avatar_url
|
||||
})
|
||||
|
||||
# 将数据转换为 JSON 字符串
|
||||
json_string = json.dumps(json_data, ensure_ascii=False, indent=4)
|
||||
# 写入到文件
|
||||
with open(f"{now_login_user.uin}/{now_login_user.uin}_{self.title}_data.json", "w", encoding="utf-8") as f:
|
||||
f.write(json_string)
|
||||
|
||||
|
||||
def export_excel(self,e):
|
||||
print("Exporting Excel...")
|
||||
export_data = []
|
||||
for item in self.data:
|
||||
if isinstance(item, User):
|
||||
export_data.append({
|
||||
'Type': 'User',
|
||||
'Username': item.username,
|
||||
'QQ': item.uin,
|
||||
'Avatar URL': item.avatar_url
|
||||
})
|
||||
elif isinstance(item, Message):
|
||||
export_data.append({
|
||||
'Type': 'Message',
|
||||
'Username': item.user.username,
|
||||
'Avatar URL': item.user.avatar_url,
|
||||
'Time': str(item.time),
|
||||
'Content': item.content,
|
||||
'Images': item.images if item.images else '',
|
||||
'Comment': item.comment.content if item.comment else '',
|
||||
})
|
||||
|
||||
# 将数据转换为 DataFrame
|
||||
df = pd.DataFrame(export_data)
|
||||
# 保存为 Excel 文件
|
||||
df.to_excel(f"{now_login_user.uin}/{now_login_user.uin}_{self.title}_data.xlsx", index=False)
|
||||
|
||||
|
||||
def export_html(self,e):
|
||||
print("Exporting HTML...")
|
||||
|
||||
def export_markdown(self,e):
|
||||
print("Exporting Markdown...")
|
||||
# 创建 Markdown 内容的列表
|
||||
markdown_lines = []
|
||||
|
||||
# 添加标题
|
||||
markdown_lines.append(f"# {self.title}\n")
|
||||
|
||||
# 填充数据
|
||||
for item in self.data:
|
||||
if isinstance(item, User):
|
||||
markdown_lines.append(f"## 用户: {item.username}\n")
|
||||
markdown_lines.append(f"**QQ**: {item.uin}\n")
|
||||
markdown_lines.append(f"**头像 URL**: ![{item.uin}]({item.avatar_url}\n")
|
||||
markdown_lines.append("\n")
|
||||
elif isinstance(item, Message):
|
||||
# 处理时间格式
|
||||
time_str = item.time.strftime('%Y-%m-%d %H:%M:%S') if isinstance(item.time, datetime) else item.time
|
||||
markdown_lines.append(f"## 消息来自: {item.user.username}\n")
|
||||
markdown_lines.append(f"**时间**: {time_str}\n")
|
||||
markdown_lines.append(f"**内容**: {item.content}\n")
|
||||
markdown_lines.append(f"**图片**: \n")
|
||||
if item.comment:
|
||||
markdown_lines.append(f"**评论**: {item.comment.content}\n")
|
||||
markdown_lines.append(f"**头像 URL**: \n")
|
||||
markdown_lines.append("\n") # 添加空行以分隔消息
|
||||
|
||||
# 生成 Markdown 内容
|
||||
markdown_content = "\n".join(markdown_lines)
|
||||
|
||||
with open(f"{now_login_user.uin}/{now_login_user.uin}_{self.title}_data.md", 'w', encoding='utf-8') as f:
|
||||
f.write(markdown_content)
|
||||
|
||||
def did_mount(self):
|
||||
"""This method is called when the control is added to the page."""
|
||||
@ -405,7 +484,7 @@ def main(page: ft.Page):
|
||||
|
||||
def logout():
|
||||
page.session.clear()
|
||||
user_info.content.controls[0].src = "assets/logo.jpg"
|
||||
user_info.content.controls[0].src = "https://raw.githubusercontent.com/LibraHp/GetQzonehistory/refs/heads/gui/assets/logo.jpg"
|
||||
user_info.content.controls[1].value = "LibraHp"
|
||||
global now_login_user
|
||||
now_login_user = None
|
||||
@ -512,7 +591,10 @@ def main(page: ft.Page):
|
||||
content.visible = True
|
||||
return progress_bar, login_text
|
||||
|
||||
|
||||
def create_user_dir():
|
||||
if not os.path.exists(now_login_user.uin):
|
||||
os.mkdir(now_login_user.uin)
|
||||
|
||||
# 获取内容页面
|
||||
def create_get_content_page():
|
||||
if page.session.contains_key("user_cookies"):
|
||||
@ -553,6 +635,7 @@ def main(page: ft.Page):
|
||||
page.session.set("user_cookies", target_cookies)
|
||||
log(f"登录成功,欢迎您，{page.session.get('user_cookies')['uin']}", "success")
|
||||
get_login_user_info()
|
||||
create_user_dir()
|
||||
progress_bar, login_text = show_login_content()
|
||||
create_card_list_view(progress_bar, login_text)
|
||||
# p_skey = requests.utils.dict_from_cookiejar(r.cookies).get('p_skey')
|
||||
@ -599,7 +682,7 @@ def main(page: ft.Page):
|
||||
alignment=ft.MainAxisAlignment.CENTER,
|
||||
data='not_login'
|
||||
),
|
||||
ft.Image(src="assets/loading.gif", expand=True, data='login_pic', visible=False),
|
||||
ft.Image(src="https://raw.githubusercontent.com/LibraHp/GetQzonehistory/refs/heads/gui/assets/loading.gif", expand=True, data='login_pic', visible=False),
|
||||
ft.Text("获取空间消息中...", size=24, weight="bold", data='login_text',visible=False),
|
||||
ft.ProgressBar(data='login_progress', visible=False,bar_height=10,border_radius=10),
|
||||
],
|
||||
@ -877,7 +960,7 @@ def main(page: ft.Page):
|
||||
user_info = ft.Container(
|
||||
content=ft.Column(
|
||||
controls=[
|
||||
ft.Image(src="assets/logo.jpg", width=80, height=80, border_radius=100), # Replace with actual avatar URL
|
||||
ft.Image(src="https://raw.githubusercontent.com/LibraHp/GetQzonehistory/refs/heads/gui/assets/logo.jpg", width=80, height=80, border_radius=100), # Replace with actual avatar URL
|
||||
ft.Text("LibraHp", size=20, weight="bold")
|
||||
],
|
||||
alignment="center",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user