main.py

# coding:utf-8

import werobot
from werobot import client
import config
import brain

robot = werobot.WeRoBot(token=config.token, enable_session=config.session)
client = client.Client(config.wx_config)


@robot.handler
def error(message, session):
    return u'''您给的内容我们暂时无法识别。。。。。
点击下面的 帮助 可以查看使用方法'''

@robot.key_click('main')
def main(message, session):
    return 'main'

@robot.subscribe
@robot.key_click('help')
def help(message, session):
    return 'help'

@robot.key_click('ls')
def ls(message, session):
    count = session.get('ls', 0) + 1
    return u'''历史推送 第%d页''' % (count,)

@robot.unsubscribe
def unsubscribe(message, session):
    return 'subscribe'

@robot.click
def click(message, session):
    return str(message.key)

@robot.text
def text(message, session):
    print session, message.source, message.time
    res = brain.run(message.content)
    return u'''%s''' % (res,)


if __name__ == '__main__':
    try:
        client.create_menu(config.menu)
    except:
        print "client create menu error"
    robot.run(server=config.server, host=config.host, port=config.port)