OpenSourceJoker/GetQzonehistory
3
0
Fork 0
mirror of https://github.com/LibraHp/GetQzonehistory.git synced 2025-02-23 14:09:07 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

添加错误处理

Browse Source
This commit is contained in:
LibraHp_0928 2024-02-13 21:15:37 +08:00
parent 9f5a6cc5d1
commit 87d663e121
1 changed files with 38 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
main.py
View File

@ -4,12 +4,34 @@ import util.RequestUtil as Request
import util.ToolsUtil as Tools import util.ToolsUtil as Tools
import util.ConfigUtil as Config import util.ConfigUtil as Config
import pandas as pd import pandas as pd
import signal
# 信号处理函数
def signal_handler(signal, frame):
# 在手动结束程序时保存已有的数据
if len(texts) > 0:
save_data()
exit(0)
def save_data():
df = pd.DataFrame(texts, columns=['内容'])
df.to_excel(Config.result_path + Request.uin + '.xlsx', index=False)
print('导出成功，请查看 ' + Config.result_path + Request.uin + '.xlsx')
if __name__ == '__main__': if __name__ == '__main__':
user_info = Request.get_login_user_info() user_info = Request.get_login_user_info()
user_nickname = user_info[Request.uin][6] user_nickname = user_info[Request.uin][6]
print(f"用户<{Request.uin}>,<{user_nickname}>登录成功") print(f"用户<{Request.uin}>,<{user_nickname}>登录成功")
texts = [] texts = []
try:
# 注册信号处理函数
signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)
for i in trange(1000, desc='Progress', unit='iteration'): for i in trange(1000, desc='Progress', unit='iteration'):
message = Request.get_message(i * 100, 100).content.decode('utf-8') message = Request.get_message(i * 100, 100).content.decode('utf-8')
html = Tools.process_old_html(message) html = Tools.process_old_html(message)
@ -20,9 +42,10 @@ if __name__ == '__main__':
text = element.get_text().replace('\xa0', ' ') text = element.get_text().replace('\xa0', ' ')
if text not in texts: if text not in texts:
texts.append(text) texts.append(text)
# 创建一个DataFrame对象
df = pd.DataFrame(texts, columns=['内容'])
# 导出为Excel if len(texts) > 0:
df.to_excel(Config.result_path + Request.uin + '.xlsx', index=False) save_data()
print('导出成功,请查看 ' + Config.result_path + Request.uin + '.xlsx') except Exception as e:
print(f"发生异常: {str(e)}")
if len(texts) > 0:
save_data()