144 lines
3.9 KiB
Python
144 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
继续截取微信通讯录 - 从当前位置继续滚动到底
|
|
"""
|
|
import uiautomation as auto
|
|
import time
|
|
import os
|
|
from PIL import Image
|
|
|
|
|
|
def find_wechat_window():
|
|
"""查找微信窗口"""
|
|
wechat_window = auto.WindowControl(searchDepth=1, Name='微信')
|
|
if wechat_window.Exists(3, 1):
|
|
return wechat_window
|
|
return None
|
|
|
|
|
|
def capture_contact_region(wechat_window, index, save_dir):
|
|
"""截取通讯录区域"""
|
|
rect = wechat_window.BoundingRectangle
|
|
|
|
x_offset = 70
|
|
y_offset = 130
|
|
width = 280
|
|
height = rect.height() - 160
|
|
|
|
screenshot_path = os.path.join(save_dir, f"region_{index:03d}.png")
|
|
|
|
try:
|
|
bitmap = wechat_window.ToBitmap(x=x_offset, y=y_offset, width=width, height=height)
|
|
bitmap.ToFile(screenshot_path)
|
|
return screenshot_path
|
|
except Exception as e:
|
|
print(f"截图失败: {e}")
|
|
return None
|
|
|
|
|
|
def get_image_hash(image_path):
|
|
"""计算图片哈希"""
|
|
try:
|
|
img = Image.open(image_path)
|
|
img = img.resize((16, 16), Image.Resampling.LANCZOS)
|
|
img = img.convert('L')
|
|
pixels = list(img.get_flattened_data())
|
|
avg = sum(pixels) / len(pixels)
|
|
return ''.join(['1' if p > avg else '0' for p in pixels])
|
|
except:
|
|
return None
|
|
|
|
|
|
def images_similarity(hash1, hash2):
|
|
"""计算相似度"""
|
|
if not hash1 or not hash2:
|
|
return 0
|
|
diff = sum(c1 != c2 for c1, c2 in zip(hash1, hash2))
|
|
return 1 - diff / len(hash1)
|
|
|
|
|
|
def scroll_down(wechat_window):
|
|
"""向下滚动一页"""
|
|
try:
|
|
rect = wechat_window.BoundingRectangle
|
|
center_x = rect.left + 200
|
|
center_y = rect.top + 400
|
|
|
|
auto.SetCursorPos(center_x, center_y)
|
|
auto.Click(center_x, center_y)
|
|
time.sleep(0.2)
|
|
auto.WheelDown(wheelTimes=3)
|
|
time.sleep(0.4)
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
|
|
def main():
|
|
print("=" * 60)
|
|
print("继续截取微信通讯录")
|
|
print("=" * 60)
|
|
|
|
# 使用已有的截图目录
|
|
save_dir = r"D:\夏骥\微信研究\scroll_full"
|
|
|
|
# 获取已有截图数量
|
|
existing_files = [f for f in os.listdir(save_dir) if f.endswith('.png')]
|
|
start_index = len(existing_files)
|
|
|
|
print(f"已有 {start_index} 张截图")
|
|
print(f"将从第 {start_index + 1} 张开始继续...")
|
|
|
|
# 查找微信窗口
|
|
print("\n查找微信窗口...")
|
|
wechat_window = find_wechat_window()
|
|
if not wechat_window:
|
|
print("未找到微信窗口！")
|
|
return
|
|
|
|
print(f"找到微信窗口: {wechat_window.Name}")
|
|
|
|
# 提示用户
|
|
print("\n请确保微信通讯录界面已打开并滚动到之前的位置...")
|
|
time.sleep(3)
|
|
|
|
# 开始截图
|
|
print("\n开始继续截图...")
|
|
screenshots = []
|
|
last_hash = None
|
|
no_change = 0
|
|
max_screenshots = 100
|
|
|
|
for i in range(max_screenshots):
|
|
actual_index = start_index + i
|
|
path = capture_contact_region(wechat_window, actual_index, save_dir)
|
|
if path:
|
|
screenshots.append(path)
|
|
print(f" 截图 {actual_index + 1}")
|
|
|
|
# 检测是否到底
|
|
current_hash = get_image_hash(path)
|
|
if last_hash:
|
|
sim = images_similarity(last_hash, current_hash)
|
|
if sim > 0.95:
|
|
no_change += 1
|
|
if no_change >= 2:
|
|
print(f"\n检测到到底，共新增截图 {len(screenshots)} 张")
|
|
break
|
|
else:
|
|
no_change = 0
|
|
last_hash = current_hash
|
|
|
|
scroll_down(wechat_window)
|
|
|
|
print(f"\n完成！新增 {len(screenshots)} 张截图")
|
|
print(f"总截图数: {start_index + len(screenshots)}")
|
|
print(f"截图保存目录: {save_dir}")
|
|
|
|
# 提示运行OCR
|
|
print("\n接下来请运行 batch_ocr_full.py 进行OCR识别")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|