pywechat获取通讯录好友信息实现方法及原理

pywechat实现微信自动化的原理是UI自动化+模拟键鼠来人为操作微信。

        微信中的好友信息全部都位于通讯录中.微信的通讯录布局为：左侧为一个ListView,内置scrollbar，鼠标点击后可以激活使用Up,Down,Pagedown,Pageup,Home,End六个按键控制，右侧为一个Pane内部为好友的所有信息。由此便想到了获取器内部所有内容的一个方法。即，pyautogui模拟按键控制左侧Listview内元素滚动，同时pywinauto定位右侧Pane容器内所有属性为Text的元素。思路很简单，但是实现起来还是有一点点难度。

废话不多说，直接上代码：

import time
import json
from pywinauto import mouse
import pyautogui
from pywechat.WechatTools import Tools
from pywechat.WinSettings import Systemsettings
def get_friends_info(wechat_path:str=None,is_maximize:bool=True,close_wechat:bool=True):
    '''
    该函数用来获取通讯录中所有微信好友的基本信息(昵称,备注,微信号),速率约为1秒7-22个好友,注:不包含企业微信好友,\n
    结果以json格式返回\n
    Args:
        wechat_path:\t微信的WeChat.exe文件地址,当微信未登录时,该函数使用此地址启动微信并点击登录完成打开微信界面。\n
            这里强烈建议将微信路径加入到windows环境变量中,因为该函数默认使用windows环境变量中的Wechat.exe路径启动微信,此时调用该函数无需传入该参数
            若你没有设置微信的Wechat.exe地址为环境变量,那么你需要传入该文件地址作为参数否则会引发错误无法打开微信！
        is_maximize:\t微信界面是否全屏,默认全屏。\n
        close_wechat:\t任务结束后是否关闭微信,默认关闭\n
    '''
    #实现思路：
    #使用while循环模拟用户在通讯录中的好友列表中人为手动按下键盘上的down键,
    #此时微信右侧pane容器内会出现好友的详细信息,定位到pane容器内的Text元素后并将其记录下来，即是每个好友的信息
    #直到记录到最后一个好友的信息后，结束while循环

    #判断最后一个好友的原理
    #首先点击通讯录列表的边界处，将scrollbar激活,此时按下键盘上的end键便会瞬间
    #跳转到通讯录列表的最后一个好友,记录下最后一个好友的微信号,注意必须是微信号!(微信号不可能重复,昵称和备注都有可能重复导致提前结束)
    #然后再按下home键又会顺间跳转到通讯录最顶部,一般是新的朋友选项
    #此时按下down键便到了第一个好友选项卡,这时我们将最后一个好友的微信号append进入wechatnumbers列表中，并再向其中填充一个任意元素后,
    #使用while wechatnumbers[0]!=wechatnumbers[-1]:重复get_info函数操作获取每个好友信息,当记录到最后一个好友的信息时wechatnumbers[0]==wechatnumbers[-1]结束循环
    #最后再删除掉wechatnumbers[0]开始时记录的最后一个好友的微信号,以及wechatnumbers[1]任意填充的元素便大功告成了

    #获取右侧变化的好友信息面板内的信息
    def get_info():
        nickname=None
        wechatnumber=None
        remark=None
        try: #通讯录界面右侧的好友信息面板  
            global base_info_pane
            try:
                base_info_pane=main_window.children(title='',control_type='Pane')[1].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[1].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0]
            except IndexError:
                base_info_pane=main_window.children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[1].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0].children(title='',control_type='Pane')[0]
            base_info=base_info_pane.descendants(control_type='Text')
            base_info=[element.window_text() for element in base_info]
            # #如果有昵称选项,说明好友有备注
            if base_info[1]=='昵称：':
                remark=base_info[0]
                nickname=base_info[2]
                wechatnumber=base_info[4]
            else:#没有昵称选项，好友昵称就是备注,备注就是昵称
                nickname=base_info[0]
                remark=nickname
                wechatnumber=base_info[2]
            return nickname,remark,wechatnumber
        except IndexError:
            return '非联系人'
    Systemsettings.open_listening_mode(volume=False)
    main_window=Tools.open_contacts(wechat_path=wechat_path,is_maximize=is_maximize)
    ContactsLists=main_window.child_window(title='联系人',control_type='List')
    #############################
    #先去通讯录列表最底部把最后一个好友的信息记录下来，通过键盘上的END健实现
    rec=ContactsLists.rectangle()
    mouse.click(coords=(rec.right-5,rec.top))
    pyautogui.press('End')
    last_member_info=get_info()
    while last_member_info=='非联系人':
        pyautogui.press('up',_pause=False)
        time.sleep(0.01)
        last_member_info=get_info()
    last_member_info={'wechatnumber':last_member_info[2]}
    pyautogui.press('Home')
    ######################################################################
    pyautogui.press('down')
    nicknames=[] 
    remarks=[]
    #初始化微信号列表为最后一个好友的微信号与任意字符,填充任意
    wechatnumbers=[last_member_info['wechatnumber'],'nothing']
    #核心思路，一直比较存放所有好友微信号列表的首个元素和最后一个元素是否相同，
    #当记录到最后一个好友时,列表首位元素相同,此时结束while循环,while循环内是一直按下down健，记录右侧变换
    #的好友信息面板内的好友信息
    while wechatnumbers[-1]!=wechatnumbers[0]:
        pyautogui.keyDown('down',_pause=False)
        time.sleep(0.01)
        friends_info=get_info()
        if friends_info!='非联系人':
            nicknames.append(friends_info[0])
            remarks.append(friends_info[1])
            wechatnumbers.append(friends_info[2])
    #删除一开始存放在起始位置的最后一个好友的微信号,不然重复了
    del(wechatnumbers[0])
    #第二个位置上是填充的任意字符,删掉上一个之后它变成了第一个,也删掉
    del(wechatnumbers[0])
    ##########################################
    #转为json格式
    records=zip(nicknames,remarks,wechatnumbers)
    contacts=[{'昵称':name[0],'备注':name[1],'微信号':name[2]} for name in records]
    contacts_json=json.dumps(contacts,ensure_ascii=False,separators=(',', ':'),indent=4)
    ##############################################################
    pyautogui.press('Home')
    if close_wechat:
        main_window.close()
    Systemsettings.close_listening_mode()
    return contacts_json

systemsettings是使用pywin32的一些内置API写的用来设置windows系统的函数所构成的类,其中open_listening_mode()函数运行后,你的电脑屏幕将一直常亮不会息屏，除非重启电脑或运行systemsettings.close_listening_mode()函数。

该函数已内置到pywechat内置模块中,使用时只需：

import pywechat.WechatAuto as wechat
friend_data=wechat.get_friends_info()
print(friend_data)
with open('好友信息.txt','w',encoding='utf-8') as f:
    f.write(friend_data)

 注意,若你需要写入获得的信息时，务必使用'utf-8'编码格式,因为不排除有的微信好友使用表情符号作为名称,此时需要特别使用utf-8格式才可以保证表情符号可以正常写入,否则将出现错误或乱码。

https://github.com/Hello-Mr-Crab/pywechat 项目github链接

若有大佬对源码感兴趣,欢迎前往批评指正,顺便帮我点一个小小的star