大数据 爬虫Python AI Sql程序员技术栈

如何让自己在斗图中立于不败之地?

2019-03-27  本文已影响5人  AirPython
image

阅读文本大概需要 6 分钟。

目 标 场 景

平常和小伙伴们斗图的过程中,你是否觉得微信的表情收藏夹不够用,或者是查找太麻烦,不够个性化?

那是否可以利用 Python,帮我们挑选到合适的表情,发送到指定的朋友和微信群呢?

本文的目标是根据关键字爬取「斗图啦」网站上的表情文件,然后登录网页微信端,挑选到合适的表情图片下载到本地,最后发送给指定的对象。

准 备 工 作

爬取到表情后,需要人工进行一次筛选;另外模拟登录微信、发送文件,我们需要通过 pip3 安装好「matplotlib」和「itchat」库。

# 图形显示库
pip3 install matplotlib

# 安装itchat
pip3 install itchat

分 析 思 路

我们打开斗图啦的官网,输入关键字「装逼」,点击搜索,发现页面分页后匹配到了很多表情图片。

image

首先,我们利用 xpath 获取到当前页面内所有表情父容器元素的列表。

image
# 获取当前页面的元素
resp = requests.get(current_url, headers=HEADERS)

html_raw = etree.HTML(resp.text)

# 获取到所有的表情图片所在的a标签
a_elements = html_raw.xpath('//div[@class="pic-content text-center"]/div[@class="random_picture"]/a')

接着,遍历容器列表,获取到所有表情的名称、下载地址属性信息。

for a_element in a_elements:
            # 获取img标签【最后一个img】【存储地址】
            img_element = a_element.xpath('./img[last()]')[0]

            # 获取p标签【存储名称】
            name = a_element.xpath('./p/text()')[0]

            # 表情的下载地址
            img_url = img_element.get('data-original')

            # 注意:由于itchat没法发送带中文的文件,这里随机生成一个名称
            name_new = make_random_string(6)

            img_name = name_new + os.path.splitext(img_url)[-1]

            imgs.append({
                'name': img_name,
                'url': img_url
            })

然后,利用「urlretrieve」将表情图片下载到本地;使用「matplotlib」将表情展示出来供我们浏览选择。

image
# 表情保存的目录
local_img = './imgs/%s' % target_emoji.get('name')

# 下载到本地
request.urlretrieve(target_emoji.get('url'), local_img)

# 把表情显示出来供用户选择
lena = mpimg.imread(filename)

plt.imshow(lena)  
plt.axis('off')  
plt.show()

通过控制台输入内容,决定发送这个表情图片,或者是查看下一个表情图片;也可以通过输入控制发送多个表情图片。

ok = input('主人满意吗:')

if ok:
   print('好的,就发送这张表情。')

   # 需要再发一张吗
   go_on_send = input('需要再发一张吗?')
   if go_on_send:
         continue
   else:
         print('结束了')
         break

最后,通过 itchat 查询到指定的人,将选中的表情图片发送给他。

def send_to_person(username, file_names):
    """
    发送给某个人
    :param username: 发送对象的昵称
    :param filename: 文件名
    :return:
    """
    room = itchat.search_friends(name=r'%s' % username)

    userName = room[0]['UserName']

    try:
        if isinstance(file_names, list):
            # 多个图片
            for file_name in file_names:
                itchat.send_image(file_name, toUserName=userName)
        else:
            # 一个图片
            itchat.send_image(file_names, toUserName=userName)
        print('发送完毕!')
    except:
        print('发送出错!')

也可以通过群聊名称获取到群聊对象,把表情图片发送到指定的微信群内。

def send_to_group_chat(target_group_chat_name, file_names):
    """
    发送表情到指定的群聊
    :param target_group_chat_name:
    :param file_name:
    :return:
    """
    rooms = itchat.get_chatrooms(update=True)

    # 目标群聊对象
    target_room = None
    for room in rooms:
        group_chat_name = room.get('NickName')
        if target_group_chat_name == group_chat_name:
            target_room = room
            break

    if target_room:
        if isinstance(file_names, list):
            for file_name in file_names:
                target_room.send_image(file_name)
        else:
            target_room.send_image(file_names)

        print('发送完毕!')
    else:
        print('抱歉,不存在这个群聊')

结 果 结 论

运行程序后,通过微信客户端扫描二维码登录微信,然后输入要发送的对象、表情类型,程序会自动爬取到匹配后的表情,最后通过用户筛选后,发送到指定的人。

本文首发于公众号「AirPython」,后台回复「斗图」即可获取完整代码。

上一篇下一篇

猜你喜欢

热点阅读