叮当 是一款可以工作在 Raspberry Pi 上的开源中文语音对话机器人 /智能音箱项目,目的是让中国的 Hacker 们也能快速打造类似 Amazon Echo 的个性化的智能音箱。
由于叮当是完全开源的,编写插件来满足自己的个性化需求就成了使用叮当的最大乐趣。您可以自由地接入自己需要的服务,让叮当更好的为您服务。
这篇文章将一步步教你如何编写一个天气插件,并最终发布到 dingdang-contrib 中。
交互示例:
首先需要确保您已安装了叮当的第三方插件库 dingdang-contrib :
cd /home/pi/.dingdang
git clone http://github.com/wzpan/dingdang-contrib contrib
pip install -r contrib/requirements.txt
sudo apt-get install sox # 用于播放音乐
sudo apt-get install libsox-fmt-mp3 # 添加 sox 的 mp3 格式支持
接下来可以在这个插件库中开发您的插件。
要实现天气预报功能,少不了要了解一下天气 API。通过搜索,我找到了一款免费的天气 API —— 心知天气 。心知天气提供了天气、空气质量、生活指数等多种数据信息。其中逐日天气预报是免费的,可以利用来实现天气预报查询插件。
选择心知天气的另一个理由是他们的 API 文档非常详细,还提供了多种语言的 demo。下面是官方提供的一个 Python 版的示例:
import requests
from utils.const_value import API, KEY, UNIT, LANGUAGE
from utils.helper import getLocation
def fetchWeather(location):
result = requests.get(API, params={
'key': KEY,
'location': location,
'language': LANGUAGE,
'unit': UNIT
}, timeout=1)
return result.text
if __name__ == '__main__':
location = getLocation()
result = fetchWeather(location)
print(result)
其中,API
是 API 的地址,逐日天气预报的 API 地址是 https://api.seniverse.com/v3/weather/daily.json ;KEY
则是心知天气的 API 密钥,每个注册账户都可以得到一个密钥;location
是城市名,例如深圳就是 深圳
或者 shenzhen
;而 language
和 unit
分别表示语言和单位,由于是可选参数,这里不做详细介绍。有兴趣的朋友请阅读官方文档。
整段代码也没有什么特别好说的:先是定义了一个 fetchWeather
函数,该函数使用 requests 模块发起 API 请求,请求超时设置为 1 秒。之后调用这个函数并打印返回的结果。
了解了心知天气的 API 后,编写插件就是一件水到渠成的事情了。
编写插件主要要考虑以下几个问题:
下面我们将在编写这个插件的过程中解决这些问题。
首先在 ~/.dingdang/contrib
中创建一个新的文件 Weather.py:
# -*- coding: utf-8-*- # 天气插件
import sys
reload(sys)
sys.setdefaultencoding('utf8')
# Standard module stuff WORDS = []
def handle(text, mic, profile, wxbot=None):
"""
Responds to user-input, typically speech text
Arguments:
text -- user-input, typically transcribed speech
mic -- used to interact with the user (for both input and output)
profile -- contains information related to the user (e.g., phone
number)
wxbot -- wechat bot instance
"""
pass
def isValid(text):
"""
Returns True if the input is related to weather.
Arguments:
text -- user-input, typically transcribed speech
"""
pass
这个是插件的模板文件。一个标准的插件至少包含 WORDS
,isValid
函数和 handle
函数三个部分。下面将一一介绍。
WORDS
是一个关键词列表,用于存储这个插件的指令关键词(的拼音)。当 Active Listening 的 SST 引擎设置成离线引擎时,这里设置的关键词会被自动添加到 PocketSphinx 的语音指令集中,从而使得 PocketSphinx 能识别出这个指令。
WORDS
的另一个作用是作为是否插件的判断标准。如果 contrib 目录中的某个 .py
文件没有 WORDS
变量,那么这个文件就会被当成无效插件,而不会响应它。
与关键词有关的还有 isValid
函数,该函数用于判断用户输入的指令是否要用这个插件来处理。如果 isValid
返回结果为 true
,handle
函数就会被调用,以处理指令。对于天气插件,关键词可以设置为天气,即只要包含“天气”的输入都使用本插件做处理。因此,可以将 WORDS
和 isValid
函数改写成:
...
WORDS = ["TIANQI"]
...
def isValid(text):
"""
Returns True if the input is related to weather.
Arguments:
text -- user-input, typically transcribed speech
"""
return u"天气" in text
接下来需要实现 handle()
函数。
def handle(text, mic, profile, wxbot=None):
"""
Responds to user-input, typically speech text
Arguments:
text -- user-input, typically transcribed speech
mic -- used to interact with the user (for both input and output)
profile -- contains information related to the user (e.g., phone
number)
wxbot -- wechat bot instance
"""
pass
这个函数接收四个参数:
text
是 STT 识别到的用户指令;mic
是麦克风和喇叭模块,最常用的是通过调用 mic.say()
函数来让喇叭说话;profile
是用户配置信息,它是一个字典,记录了 ~/.dingdang/profile.yml
的全部内容;wxbot
是一个微信机器人实例,可以利用它给用户发送微信消息。Camera 插件提供了通过微信发送照片给用户的例子。我们可以把心知的 Demo 给出的 fetchWeather
函数做一点调整,放进代码中方便复用:
def fetch_weather(api, key, location):
result = requests.get(api, params={
'key': key,
'location': location
}, timeout=3)
res = json.loads(result.text, encoding='utf-8')
return res
显然,KEY
和 location
应该作为用户的配置项,允许用户在配置文件中进行设置。因此我们可以在 profile.yml 配置文件中添加如下配置:
# 天气
# 使用心知天气的接口
# https://www.seniverse.com/
weather:
key: 'etxzx9abupxplhic' # 心知天气 API Key
location: '深圳'
接下来在 handle
函数中调用 fetch_weather
函数,得到天气信息:
def handle(text, mic, profile, wxbot=None):
"""
Responds to user-input, typically speech text
Arguments:
text -- user-input, typically transcribed speech
mic -- used to interact with the user (for both input and output)
profile -- contains information related to the user (e.g., phone
number)
wxbot -- wechat bot instance
"""
logger = logging.getLogger(__name__)
# get config if 'weather' not in profile or \
not profile['weather'].has_key('key') or \
not profile['weather'].has_key('location'):
mic.say('天气插件配置有误,插件使用失败')
return
key = profile['weather']['key']
location = profile['weather']['location']
WEATHER_API = 'https://api.seniverse.com/v3/weather/daily.json'
try:
weather = fetch_weather(WEATHER_API, key, location)
logger.debug("Weather report: ", weather)
if weather.has_key('results'):
daily = weather['results'][0]['daily']
day_text = [u'今天', u'明天', u'后天']
responds = u'%s 天气:' % location
for day in range(len(day_text)):
responds += u'%s:%s,%s 到%s 摄氏度。' % (day_text[day], daily[day]['text_day'], daily[day]['low'], daily[day]['high'])
mic.say(responds)
else:
mic.say('抱歉,我获取不到天气数据,请稍后再试')
except Exception, e:
logger.error(e)
mic.say('抱歉,我获取不到天气数据,请稍后再试')
完成后,可以重启下叮当,看看插件是否能正常工作。
插件正常工作后,可以将该插件发布到 dingdang-contrib ,让更多人用上您的插件。
首先先访问 dingdang-contrib 的 Github 主页 ,点击右上角的 [ fork ] 按钮,将仓库 fork 到自己的账户。如果之前已经 fork 过,这一步可以跳过。
fork 完仓库后,在您的账户下也会有一个 dingdang-contrib 项目,点击绿色的 [ Clone or download ] 按钮,记下新的仓库的地址。
之后在树莓派中执行如下命令,添加新的仓库地址:
cd ~/.dingdang/contrib
git remote add mine 新的仓库地址
将新建的插件提交推送到您的 dingdang-contrib 仓库中:
git add Weather.py
git commit -m "新增天气插件"
git push -u mine master
完成后访问您的 dingdang-contrib 仓库主页,可以看到一个创建 pull request 的提示:
点击 [ compare and pull request ] 按钮,进入 pull request 创建页面,申请将您的改动合并到 dingdang-contrib 项目中:
在里头认真填写插件的用途、使用示例和配置项。完成后点击 [ Create pull requset ] ,完成创建,等待 dingdang-robot 组织的审核。
一旦审核通过,您的插件就发布成功了。
为了让更多人了解您的插件的用途,还应该到 dingdang-contrib 的 Wiki 中添加该插件的用途。先在首页中增加一条插件记录:
完成后首页就增加了您创建的插件的记录:
点击该插件的链接,进入插件详情页创建页面,填入和刚刚创建 pull request 时相同的内容,即插件的用途、使用示例和配置项即可(也可以根据需求增加更多信息)。