用Python製作表白神器——愛心牆

阿新 • • 發佈：2021-01-19

前言
本文的文字及圖片來源於網路,僅供學習、交流使用,不具有任何商業用途,如有問題請及時聯絡我們以作處理。

PS：如有需要Python學習資料的小夥伴可以加下Python快樂交流群：1136201545

一、❤️愛心牆❤️

通過爬蟲蒐集到粉絲的頭像，然後利用PIL庫拼接出愛心牆的形狀：

二、程式碼分析

1.頭像爬取

在個人中心點選我的粉絲便可以看到自己的粉絲，通過抓包可知對應的介面為：

url = 'https://me.csdn.net/api/relation/index?pageno=1&pagesize=20&relation_type=fans' # 介面地址

那麼，可以定義一個函式來獲取粉絲的資訊：

def get_fansInfo():
    '''
    獲取粉絲相關資訊
    '''
    url = 'https://me.csdn.net/api/relation/index?pageno=%d&pagesize=%d&relation_type=fans' # 介面地址
    cookies = {} # 使用者登陸cookies
    headers = {  # 請求頭
        'User-Agent': 'Mozilla/5.0 (Windows NT 6.3; Win64; x64; rv:81.0) Gecko/20100101 Firefox/81.0',
        'Accept': 'application/json, text/plain, */*',
        'Accept-Language': 'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2',
        'Referer': 'https://i.csdn.net/',
        'Origin': 'https://i.csdn.net',
        'Connection': 'keep-alive',
        'TE': 'Trailers',
    }
    # 獲取粉絲總數
    res = requests.get(url%(1,10),headers=headers,cookies=cookies)
    res_json = res.json()
    N_fans = res_json['data']['data_all']
    print('一共有%d個粉絲'%N_fans)
    # 獲取全部粉絲資料
    res = requests.get(url%(1,N_fans),headers=headers,cookies=cookies)
    res_json = res.json()
    return res_json

在返回的資料中，包括一個avatar欄位，這個就是使用者的頭像地址，拿到頭像地址之後便可以定義個函式來下載相應的頭像：

def download_avatar(username,url):
    '''
    下載使用者頭像
    '''
    savePath = './avatars' # 頭像儲存目錄
    res = requests.get(url)
    with open('%s/%s.jpg'%(savePath,username),'wb') as f:
        f.write(res.content)

定義主函式，執行程式碼：

if __name__ == '__main__':
    fans = get_fansInfo()
    for f in fans['data']['list']:
        username = f['fans'] # 使用者名稱
        url = f['avatar']    # 頭像地址
        download_avatar(username,url)
        print('使用者"%s"頭像下載完成!'%username)

最後我成功將所有頭像下載到本地資料夾中：

2.頭像去重

聰明的你應該已經發現，在爬取到的頭像中有兩個頭像重複出現（想必這應該是官方預設頭像）:

於是乎，為了更好地展示，我們得對頭像進行去重。這裡我們利用每個頭像的MD5值來進行去重，首先定義函式來計算頭像的MD5值：

def get_md5(filename):
    '''
    獲取檔案的md5值cls
    '''
    m = hashlib.md5()
    with open(filename,'rb') as f:
        for line in f:
            m.update(line)
    md5 = m.hexdigest()
    return md5

說明：每個檔案通過MD5計算出摘要，理論來說只有檔案完全一致MD5值才會相同。因此，可以利用它來進行影象的去重。
對頭像進行去重，並把去重後的頭像儲存到另外的目錄中：

# 照片去重
md5_already = [] # 用於儲存已經記錄過的圖片，便於去重
for filename in os.listdir('./avatars'):
    md5 = get_md5('./avatars/'+filename)  
    if md5 not in md5_already:
        md5_already.append(md5)
        shutil.copyfile('./avatars/'+filename,'./avatars(dr)/'+filename)

3.繪製愛心牆

這一步，主要是利用python中的PIL庫來把頭像按照設定的框架拼接成一個更大的圖片。首先匯入相關庫：

import os
import random
import numpy as np
import PIL.Image as Image

定義繪製圖形的框架（用二維陣列表示）：

FRAME = [[0,1,1,0,0,0,0,1,1,0],
         [1,1,1,1,0,0,1,1,1,1],
         [1,1,1,1,1,1,1,1,1,1],
         [1,1,1,1,1,1,1,1,1,1],
         [0,1,1,1,1,1,1,1,1,0],
         [0,0,1,1,1,1,1,1,0,0],
         [0,0,0,1,1,1,1,0,0,0],
         [0,0,0,0,1,1,0,0,0,0]]

這裡大家完全可以發揮自己的想象，畫你心中所想

其中，0表示不進行填充，1表示用頭像進行填充。

定義相關引數，包括每張用於填充的頭像的大小、每個點位填充的次數等：

# 定義相關引數
SIZE = 50 # 每張圖片的尺寸為50*50
N = 2     # 每個點位上放置2*2張圖片

# 計算相關引數
width = np.shape(FRAME)[1]*N*SIZE  # 照片牆寬度
height = np.shape(FRAME)[0]*N*SIZE # 照片牆高度
n_img = np.sum(FRAME)*(N**2)       # 照片牆需要的照片數
filenames = random.sample(os.listdir('./avatars(dr)'),n_img) # 隨機選取n_img張照片
filenames = ['./avatars(dr)/'+f for f in filenames]

遍歷FRAME，用頭像對背景圖片進行填充：

# 繪製愛心牆
img_bg = Image.new('RGB',(width,height)) # 設定照片牆背景
i = 0
for y in range(np.shape(FRAME)[0]):
    for x in range(np.shape(FRAME)[1]):
         if FRAME[y][x] == 1: # 如果需要填充
             pos_x = x*N*SIZE # 填充起始X座標位置
             pos_y = y*N*SIZE # 填充起始Y座標位置
             for yy in range(N):
                 for xx in range(N):
                     img = Image.open(filenames[i])
                     img = img.resize((SIZE,SIZE),Image.ANTIALIAS)
                     img_bg.paste(img,(pos_x+xx*SIZE,pos_y+yy*SIZE))
                     i += 1
                
# 儲存圖片
img_bg.save('love.jpg')