从零开始打造聊天机器人(1)之微信机器人实现自动推送打卡提醒，天气预报

引言

上一篇中以钉钉作为载体，主动推送功能确实非常好用，推送的格式丰富多彩。

但是，考虑到后面需要进行交互。而钉钉官方文档中表示当前机器人尚不支持应答机制 (该机制指的是群里成员在聊天@机器人的时候，钉钉回调指定的服务地址，即Outgoing机器人)。

应答是聊天机器人必备的技能，所以钉钉是果断要被舍弃了。私下评估一番，考虑到平台成熟度、用户量、实用程度，最后还是选择了个人微信作为机器人的基础载体。

需求

把上一篇中利用钉钉平台进行推送的《权游》更新提示移植到微信平台
新增每天早上8:30和下午6:05的上下班打卡提醒
新增每天早上8:00进行当天的天气推送

设计

《权游》更新推送的实现逻辑同上一篇
上下班打卡提醒只需要编辑对应的文本即可使用send接口发送
天气预报推送这里使用第三方极速的API获取到天气信息后，再进行筛选处理最后推送到微信
定时功能使用python的schedule
微信的接口代码主要使用wxpy，致敬

主要代码

核心逻辑见代码中注释，不懂之处可留言

#!export PYTHONIOENCODING=UTF-8; python3
# coding=UTF-8
'''
@Author: 公小虽
@LastEditors: 公小虽
@Description: -
@Date: 2019-04-26 14:24:13
@LastEditTime: 2019-04-27 22:01:42
'''
import time
from datetime import datetime
import json
import requests
import wxpy
import logging
import schedule
from notification import teleplay_update  # 上一篇中自定义的权游更新函数

DEFAULT_MSG = 'hello, every one ~ This is xxx\'s robot\n发送时间：' + \
                              datetime.now().strftime('%Y-%m-%d %H:%M:%S')
DEFAULT_TIME = '8:00'

class TimedPush():
    '''
    @description: 定时主动推送消息
    '''
    def __init__(self):
        self.bot = wxpy.Bot(cache_path=True)
        self.group_list = ['haha']  # 微信群名

    def push_to_user(self, user_name, msg):
        '''
        @description: 
        @param {user_name:昵称, msg:消息} 
        @return: 
        '''
        user = self.bot.search(user_name)[0]
        user.send(msg)

    def push_to_group(self, group_name, msg):
        '''
        @description: 
        @param {group_name:群昵称, msg:消息} 
        @return: 
        '''
        group = self.bot.groups().search(group_name)[0]
        group.send(msg)
    
    def push_to_groups(self, msg):
        for group in self.group_list:
            self.push_to_group(group, msg)
            logging.info('send msg to group %s: %s' % (group, msg))

# 权游更新提示
def is_update(tp):
    num = teleplay_update.teleplay_update()  #teleplay_update即上一篇文中的更新函数
    if num != '':
        msg = '<权利的游戏>已更新至'+num
        tp.push_to_groups(msg)

# 天气查询
def get_weather(tp):
    WEATHER_KEY = '**************'  # 这里填写自己申请的KEY
    API = 'https://api.jisuapi.com/weather/query'  # 这里使用极速的API
    CITY = '成都'
    payload = {'appkey': WEATHER_KEY, 'city': CITY}
    msg = ''
    try:
        result = requests.post(API, data=payload)
        res = json.loads(result.text)
        if res['msg'] == 'ok':  # 如果成功返回，则拼接自己需要的信息
            res = res['result']
            msg = res['date'] + '\t' + res['week'] + '\n' +\
                    res['city'] + '\n' + \
                    res['weather'] + '\n' + \
                    '温度：' + res['templow'] + '~' + res['temphigh'] + '\n' + \
                    res['winddirect'] + res['windpower']
            tp.push_to_groups(msg)
    except Exception as e:
        logging.error('查询天气失败\n %s' % str(e))

## TEST
def main():
    logging.basicConfig(level=logging.DEBUG)
    tp = TimedPush()

    # 每天12点进行剧集更新扫描
    t1 = '12:00:00'
    schedule.every().day.at(t1).do(is_update, tp)

    # # 每天8点进行天气预报
    t2 = '08:00:00'
    schedule.every().day.at(t2).do(get_weather, tp)
    
    # 每天8:30进行上班打卡提醒
    msg3 = '生活需要仪式感，嘀嘀一声打卡了冒？'
    t3 = '08:30'
    schedule.every().day.at(t3).do(tp.push_to_groups, msg3)

    # 每天18:05进行下班打卡提醒
    msg4 = '又下班了，打！卡！'
    t4 = '18:05'
    schedule.every().day.at(t4).do(tp.push_to_groups, msg4)

    while True:
        schedule.run_pending()
        time.sleep(1)
    tp.bot.join()

main()

效果

微信机器人定时推送

最后编辑于：2019.04.27 22:36:02