コード例 #1
0
async def _relay(bot, listener, target, contents, source):
    targets = _get_linked_targets(listener, target)
    for target in targets:
        target_listener = get_listener(bot, target['listener'])
        for (author, message) in _get_relay_messages(
            listener, contents, source
        ):
            await _relay_message(target_listener, bot, target, author, message)
コード例 #2
0
ファイル: github_relay.py プロジェクト: azuline/chitanda
async def _handle_request(bot, request, **kwargs):
    try:
        logger.info("Received request from GitHub webhook.")
        if not _check_signature(await request.text(), request.headers):
            logger.info("GitHub request contained invalid signature, ignoring.")
            raise BotError("Invalid signature.")

        payload = await request.json()
        repo_cfgs = _get_repo_cfgs(payload["repository"]["id"])
        logger.info('Event for repository {payload["repository"]["name"]}.')
        event_handler = _get_event_handler(request.headers)

        for cfg in repo_cfgs:
            asyncio.ensure_future(
                event_handler(get_listener(bot, cfg["listener"]), payload, cfg)
            )

        return web.Response(body="Received.")
    except InvalidListener:
        return web.Response(body="Invalid listener configured.", status=500)
    except BotError as e:
        return web.Response(body=e.args[0], status=500)
コード例 #3
0
def test_get_listener_discord():
    bot = Mock(discord_listener=1, irc_listeners={'a': 2, 'b': 3})
    assert 1 == get_listener(bot, 'DiscordListener')
コード例 #4
0
def test_get_listener_nonexistent():
    with pytest.raises(InvalidListener):
        get_listener(Mock(), 'AppleListener')
コード例 #5
0
def test_get_listener_irc_nonexistent():
    bot = Mock(discord_listener=1, irc_listeners={'a': 2, 'b': 3})
    with pytest.raises(InvalidListener):
        get_listener(bot, 'IRCListener@c')
コード例 #6
0
def test_get_listener_irc():
    bot = Mock(discord_listener=1, irc_listeners={'a': 2, 'b': 3})
    assert 2 == get_listener(bot, 'IRCListener@a')
コード例 #7
0
ファイル: test_util.py プロジェクト: azuline/chitanda
def test_get_listener_discord():
    bot = Mock(discord_listener=1, irc_listeners={"a": 2, "b": 3})
    assert 1 == get_listener(bot, "DiscordListener")
コード例 #8
0
ファイル: test_util.py プロジェクト: azuline/chitanda
def test_get_listener_irc_nonexistent():
    bot = Mock(discord_listener=1, irc_listeners={"a": 2, "b": 3})
    with pytest.raises(InvalidListener):
        get_listener(bot, "IRCListener@c")
コード例 #9
0
ファイル: test_util.py プロジェクト: azuline/chitanda
def test_get_listener_irc():
    bot = Mock(discord_listener=1, irc_listeners={"a": 2, "b": 3})
    assert 2 == get_listener(bot, "IRCListener@a")