import itchat
import os
import math
from PIL import Image

# 獲取數(shù)據(jù)
def download_image():
    # 掃描二維碼登陸微信，即通過網(wǎng)頁版微信登陸
    itchat.auto_login()
    # 返回一個包含用戶信息字典的列表
    friends = itchat.get_friends(update=True)
    #  在當前位置創(chuàng)建一個用于存儲頭像的目錄wechatImages
    base_path = 'wechatImages'
    if not os.path.exists(base_path):
        os.mkdir(base_path)

    # 獲取所有好友頭像
    for friend in friends:
        # 獲取頭像數(shù)據(jù)
        img_data = itchat.get_head_img(userName = friend['UserName'])
        #判斷備注名是否為空
        if friend['RemarkName'] != '':
            img_name = friend['RemarkName']
        else :
            img_name = friend['NickName']
         #   在實際操作中如果文件名中含有*標志，會報錯。則直接可以將其替換掉
        if img_name is "*":
            img_name = ""
        #通過os.path.join()函數(shù)來拼接文件名
        img_file = os.path.join(base_path, img_name + '.jpg')
        print(img_file)
        with open(img_file, 'wb') as file:
            file.write(img_data)


# 拼接頭像
def join_image():
    base_path = 'headImages'
    files = os.listdir(base_path) #返回指定的文件或文件夾的名字列表
    print(len(files))
    each_size = int(math.sqrt(float(6400 * 6400) / len(files)))#計算每個粘貼圖片的邊長
    lines = int(6400 / each_size)#計算總共有多少行
    print(lines)
    image = Image.new('RGB', (6400, 6400))# new(mode, size, color=0) 定義一張大小為640*640大小的圖片，不給出第三個參數(shù)默認為黑色
    x = 0 #定義橫坐標
    y = 0 #定義縱坐標
    for file_name in files:
        img = Image.open(os.path.join(base_path, file_name)) #找到/打開圖片
        img = img.resize((each_size, each_size), Image.ANTIALIAS)#實現(xiàn)圖片同比例縮放，Image.ANTIALIAS添加濾鏡效果
        image.paste(img, (x * each_size, y * each_size))#將縮放后的照片放到對應的坐標下
        x += 1
        if x == lines:#如果每行的粘貼內(nèi)容夠了，則換行
            x = 0
            y += 1
    image.save('jointPic.jpg')#最后將全部的照片保存下來

if __name__ == '__main__':
    download_image()
    join_image()

import psutil# 用于獲取微信電腦版的進程信息；
import pywinauto# 用于自動化控制微信電腦版
from pywinauto.application import Application

	PID = 0#用來保存微信的進程號

    for proc in psutil.process_iter():#循環(huán)電腦上的進程，獲取進程號和名稱
        try:
            pinfo = proc.as_dict(attrs=['pid', 'name'])
        except psutil.NoSuchProcess:#沒有運行微信程序
            pass
        else:
            if 'WeChat.exe' == pinfo['name']:#當進程名為WeChat.exe的時候，把進程號記下來
                PID = pinfo['pid']

#進程ID用來提供給 PyWinAuto.application 以連接微信電腦版，connect是要已經(jīng)運行微信才行
    app = Application(backend='uia').connect(process=PID)
    #獲得微信窗口實例
    Win_wechat = app['微信']

Win_wechat.print_control_identifiers()#以樹形方式打印窗口上所有控件

#獲取微信窗口上朋友圈按鈕實例
    Button_pyq = Win_wechat['Pane6'].child_window(title="朋友圈", control_type="Button")

	#獲取按鈕坐標
    cords = Button_pyq.rectangle()
    # 接著控制微信電腦版，模擬鼠標點擊，把朋友圈窗口打開
    pywinauto.mouse.click(button='left', coords=(cords.left + 10, cords.top + 10))

	Win_pyq = app['朋友圈']#獲取朋友圈窗口實例

    Win_pyq.draw_outline(colour = 'red',thickness = 2)# 在當前定位到的窗口圍畫出一條邊界線，方便我們看出定位到了哪個控件
    Win_pyq.dump_tree()

import psutil# 用于獲取微信電腦版的進程信息；
import pywinauto# 用于自動化控制微信電腦版
from pywinauto.application import Application

if __name__ == '__main__':
    PID = 0#用來保存微信的進程號

    for proc in psutil.process_iter():#循環(huán)電腦上的進程，獲取進程號和名稱
        try:
            pinfo = proc.as_dict(attrs=['pid', 'name'])
        except psutil.NoSuchProcess:#沒有運行微信程序
            pass
        else:
            if 'WeChat.exe' == pinfo['name']:#當進程名為WeChat.exe的時候，把進程號記下來
                PID = pinfo['pid']

    #進程ID用來提供給 PyWinAuto.application 以連接微信電腦版，connect是要已經(jīng)運行微信才行
    app = Application(backend='uia').connect(process=PID)
    #獲得微信窗口實例
    Win_wechat = app['微信']
    #win.print_control_identifiers()#以樹形方式打印窗口上所有控件

    #獲取微信窗口上朋友圈按鈕實例
    Button_pyq = Win_wechat['Pane6'].child_window(title="朋友圈", control_type="Button")
    #獲取按鈕坐標
    cords = Button_pyq.rectangle()
    # 接著控制微信電腦版，把朋友圈窗口打開
    pywinauto.mouse.click(button='left', coords=(cords.left + 10, cords.top + 10))

    Win_pyq = app['朋友圈']#獲取朋友圈窗口實例

    Win_pyq.draw_outline(colour = 'red',thickness = 2)# 在當前定位到的窗口圍畫出一條邊界線，方便我們看出定位到了哪個控件
    Win_pyq.dump_tree()#樹形打印