diff --git a/.gitignore b/.gitignore index 8cdc60e..9ff10d7 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,6 @@ -/resource/ .idea /myenv/ +/build/ +/dist/ +/resource/ + diff --git a/libiconv.dll b/libiconv.dll new file mode 100644 index 0000000..a31664d Binary files /dev/null and b/libiconv.dll differ diff --git a/libzbar-64.dll b/libzbar-64.dll new file mode 100644 index 0000000..5353d43 Binary files /dev/null and b/libzbar-64.dll differ diff --git a/main.py b/main.py index c92506e..59a6b6f 100644 --- a/main.py +++ b/main.py @@ -78,9 +78,51 @@ def signal_handler(signal, frame): def save_data(): - df = pd.DataFrame(texts, columns=['时间', '内容']) - df.to_excel(result_path + uin + '.xlsx', index=False) - print('导出成功，请查看 ' + result_path + uin + '.xlsx') + user_save_path = result_path + uin + '/' + pic_save_path = user_save_path + 'pic/' + if not os.path.exists(user_save_path): + os.makedirs(user_save_path) + print(f"Created directory: {user_save_path}") + if not os.path.exists(pic_save_path): + os.makedirs(pic_save_path) + print(f"Created directory: {pic_save_path}") + pd.DataFrame(texts, columns=['时间', '内容', '图片链接']).to_excel(user_save_path + uin + '_全部列表.xlsx', index=False) + pd.DataFrame(all_friends, columns=['昵称', 'QQ', '空间主页']).to_excel(user_save_path + uin + '_好友列表.xlsx', index=False) + for item in tqdm(texts, desc="处理消息列表", unit="item"): + item_text = item[1] + item_pic_link = item[2] + if item_pic_link is not None and len(item_pic_link) > 0 and 'http' in item_pic_link: + # 保存图片 + pic_name = re.sub(r'[\\/:*?"<>|]', '_', item_text) + '.jpg' + # pic_name = pic_name.split('：')[1] + '.jpg' + response = requests.get(item_pic_link) + if response.status_code == 200: + with open(pic_save_path + pic_name, 'wb') as f: + f.write(response.content) + if user_nickname in item_text: + if '留言' in item_text: + leave_message.append(item) + elif '转发' in item_text: + forward_message.append(item) + else: + user_message.append(item) + else: + other_message.append(item) + pd.DataFrame(user_message, columns=['时间', '内容', '图片链接']).to_excel(user_save_path + uin + '_说说列表.xlsx', index=False) + pd.DataFrame(forward_message, columns=['时间', '内容', '图片链接']).to_excel(user_save_path + uin + '_转发列表.xlsx', index=False) + pd.DataFrame(leave_message, columns=['时间', '内容', '图片链接']).to_excel(user_save_path + uin + '_留言列表.xlsx', index=False) + pd.DataFrame(other_message, columns=['时间', '内容', '图片链接']).to_excel(user_save_path + uin + '_其他列表.xlsx', index=False) + print('\033[36m' + '导出成功，请查看 ' + user_save_path + uin + ' 文件夹内容' + '\033[0m') + print('\033[32m' + '共有 ' + str(len(texts)) + ' 条消息' + '\033[0m') + print('\033[36m' + '最早的一条说说发布在' + texts[texts.__len__() - 1][0] + '\033[0m') + print('\033[32m' + '好友列表共有 ' + str(len(all_friends)) + ' 个好友' + '\033[0m') + print('\033[36m' + '说说列表共有 ' + str(len(user_message)) + ' 条说说' + '\033[0m') + print('\033[32m' + '转发列表共有 ' + str(len(forward_message)) + ' 条转发' + '\033[0m') + print('\033[36m' + '留言列表共有 ' + str(len(leave_message)) + ' 条留言' + '\033[0m') + print('\033[32m' + '其他列表共有 ' + str(len(other_message)) + ' 条内容' + '\033[0m') + print('\033[36m' + '图片列表共有 ' + str(len(os.listdir(pic_save_path))) + ' 张图片' + '\033[0m') + current_directory = os.getcwd() + os.startfile(current_directory + user_save_path[1:]) def bkn(pSkey): @@ -268,9 +310,9 @@ def get_login_user_info(): def get_message_count(): # 初始的总量范围 lower_bound = 0 - upper_bound = 100000 # 假设最大总量为1000000 + upper_bound = 10000000 # 假设最大总量为1000000 total = upper_bound // 2 # 初始的总量为上下界的中间值 - with tqdm(desc="正在获取消息列表数量...", total=17, unit='页') as pbar: + with tqdm(desc="正在获取消息列表数量...", total=23, unit='页') as pbar: while lower_bound <= upper_bound: response = get_message(total, 100) if "li" in response.text: @@ -323,6 +365,11 @@ if __name__ == '__main__': print(f"登录失败:请重新登录,错误信息:{str(e)}") exit(0) texts = [] + all_friends = [] + other_message = [] + user_message = [] + leave_message = [] + forward_message = [] count = get_message_count() try: # 注册信号处理函数 @@ -339,13 +386,25 @@ if __name__ == '__main__': for element in soup.find_all('li', class_='f-single f-s-s'): put_time = None text = None + img = None + friend_element = element.find('a', class_='f-name q_namecard') + # 获取好友昵称和QQ + if friend_element is not None: + friend_name = friend_element.get_text() + friend_qq = friend_element.get('link')[9:] + friend_link = friend_element.get('href') + if friend_qq not in [sublist[1] for sublist in all_friends]: + all_friends.append([friend_name, friend_qq, friend_link]) time_element = element.find('div', class_='info-detail') text_element = element.find('p', class_='txt-box-title ellipsis-one') + img_element = element.find('a', class_='img-item') if time_element is not None and text_element is not None: put_time = time_element.get_text().replace('\xa0', ' ') text = text_element.get_text().replace('\xa0', ' ') + if img_element is not None: + img = img_element.find('img').get('src') if text not in [sublist[1] for sublist in texts]: - texts.append([put_time, text]) + texts.append([put_time, text, img]) if len(texts) > 0: save_data() diff --git a/main.spec b/main.spec new file mode 100644 index 0000000..59e3777 --- /dev/null +++ b/main.spec @@ -0,0 +1,38 @@ +# -*- mode: python ; coding: utf-8 -*- + + +a = Analysis( + ['main.py'], + pathex=[], + binaries=[('libiconv.dll', '.'), ('libzbar-64.dll', '.')], + datas=[('config.ini', '.')], + hiddenimports=[], + hookspath=[], + hooksconfig={}, + runtime_hooks=[], + excludes=[], + noarchive=False, + optimize=0, +) +pyz = PYZ(a.pure) + +exe = EXE( + pyz, + a.scripts, + a.binaries, + a.datas, + [], + name='main', + debug=False, + bootloader_ignore_signals=False, + strip=False, + upx=True, + upx_exclude=[], + runtime_tmpdir=None, + console=True, + disable_windowed_traceback=False, + argv_emulation=False, + target_arch=None, + codesign_identity=None, + entitlements_file=None, +)