Python自動採集微信聯繫人

疫情終於有所好轉了,感謝所有的爲之奮鬥的白衣天使們,你們是最棒的!贊!

白衣天使們在前線奮戰,我們也總不能總在家裏躺着做貢獻,也要加強學習,爭取開春有個好工作:)

 

Python採集微信聯繫人

 

這周花了幾天的時間,把Python的pywinauto相關的使用及一堆的庫函數看了一遍,感覺這個庫還是非常強大的。

  • pywinauto是什麼?

pywinauto是一組用於自動化Windows GUI的python模塊。簡單地說,它允許您將鼠標和鍵盤操作發送到windows對話框和控件。這是官方的解釋,簡單說就是可以用python代碼來操縱桌面程序。

  • 安裝
pip install pywinauto
  • 使用

pywinauto的使用方式原理很簡單,但是它提供了大量的對GUI界面元素操作的函數,需要慢慢去理解。

官方文檔資料很齊全:https://pywinauto.readthedocs.io/

入門級的細節就不說了,看不懂官方英文,也可以去網上搜索,一大堆的介紹文檔。

我們直接進入今天的主題:用Python採集微信聯繫人,先上程序執行效果視頻:

 

程序原理

(1)首先微信要打開,並登錄成功。獲取當前微信的進程ID:

    PID=0
    for proc in psutil.process_iter():
        try:
            pinfo = proc.as_dict(attrs=['pid','name'])
        except psutil.NoSuchProcess:
            pass
        else:
            if 'WeChat.exe' == pinfo['name']:
                PID = pinfo['pid']

(2)使用pywinauto 連接到這個進程:

    app = Application(backend='uia').connect(process=PID)
    win = app[u'微信']

(3)找到左邊的“通訊錄”按鈕所在位置,點擊鼠標:

    addresslist = win.child_window(title=u"通訊錄", control_type="Button")
    addresslist.draw_outline()
    cords =  addresslist.rectangle()
    pywinauto.mouse.click(button='left', coords=(cords.left + 10, cords.top+10))
    win.draw_outline()

(4)核心代碼,通過調用向下的鍵盤指令,逐個打開聯繫人詳情頁,根據詳情頁的頁面結構,頁面內的所有數據:呢稱、備註、地區、微信號、來源。

    #聯繫人
    con_list = win.child_window(title=u"聯繫人", control_type="List")
    pywinauto.mouse.click(button='left', coords=(cords.left + 100, cords.top+10))
    data = []
    while True:
        con_list.type_keys("{DOWN}")
        rect_edit = win.Edit2.rectangle()
        #找到聯繫人詳情頁
        if rect_edit.left > cords.left + 300:
            #整個面板 
            page = {u'呢稱':'',u'備   注':'',u'地   區':'',u'微信號':'',u'來   源':''}
            #呢稱
            win.Edit2.draw_outline()
            page[u'呢稱'] = win.Edit2.get_value()
            #由於詳情頁面要素不固定,不能用控件序號來檢索
            static_wxh = win.child_window(title=u"微信號", control_type="Text")
            Pane46 = static_wxh.parent().parent()
            for item in Pane46.children():
                item.draw_outline()
                lines = item.children()
                key = lines[0].window_text()
                value = lines[1].window_text()
                page[key] = value
            data.append(page)
            #最後一個聯繫人時停止
            if len(data) >2 and data[-2][u'微信號'] == data[-1][u'微信號']:
                data.remove(data[-1])
                saveTxt(data)
                break

第四步的方法前後優化過多次,這是最後優化的版本,這個版本是多種方案中速度最快,同時兼容性最好的方法,如果哪位朋友還有更好的方案,歡迎留言討論!

下面附上完整代碼:

#coding=gbk
'''
Created on 
@author: Administrator
'''
import psutil
import pywinauto
from pywinauto.application import Application
import os
import sys
reload(sys)
sys.setdefaultencoding('utf-8')

def getWinxin():
    PID=0
    for proc in psutil.process_iter():
        try:
            pinfo = proc.as_dict(attrs=['pid','name'])
        except psutil.NoSuchProcess:
            pass
        else:
            if 'WeChat.exe' == pinfo['name']:
                PID = pinfo['pid']
    app = Application(backend='uia').connect(process=PID)
    win = app[u'微信']
    addresslist = win.child_window(title=u"通訊錄", control_type="Button")
    addresslist.draw_outline()
    cords =  addresslist.rectangle()
    pywinauto.mouse.click(button='left', coords=(cords.left + 10, cords.top+10))
    win.draw_outline()
    
    #聯繫人
    con_list = win.child_window(title=u"聯繫人", control_type="List")
    pywinauto.mouse.click(button='left', coords=(cords.left + 100, cords.top+10))
    data = []
    while True:
        con_list.type_keys("{DOWN}")
        rect_edit = win.Edit2.rectangle()
        #找到聯繫人詳情頁
        if rect_edit.left > cords.left + 300:
            #整個面板 
            page = {u'呢稱':'',u'備   注':'',u'地   區':'',u'微信號':'',u'來   源':''}
            #呢稱
            win.Edit2.draw_outline()
            page[u'呢稱'] = win.Edit2.get_value()
            #由於詳情頁面要素不固定,不能用控件序號來檢索
            static_wxh = win.child_window(title=u"微信號", control_type="Text")
            Pane46 = static_wxh.parent().parent()
            for item in Pane46.children():
                item.draw_outline()
                lines = item.children()
                key = lines[0].window_text()
                value = lines[1].window_text()
                page[key] = value
            data.append(page)
            #最後一個聯繫人時停止
            if len(data) >2 and data[-2][u'微信號'] == data[-1][u'微信號']:
                data.remove(data[-1])
                saveTxt(data)
                break
def saveTxt(data):
    with open("data.txt",'ab') as f:
        f.write(u'微信號,備註,地區,來源,呢稱\r\n')
        for item in data:
            f.write('%s,%s,%s,%s,%s\r\n' % (item[u'微信號'],item[u'備   注'],item[u'地   區'],item[u'來   源'],item[u'呢稱']))
    os.system('notepad.exe data.txt')
if __name__ == '__main__':
    getWinxin()

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章