def test_happy_path():
app = Kutana()
# Simple echo plugin
pl = Plugin("")
@pl.on_messages()
async def __(message, ctx):
await ctx.reply(message.text)
app.add_plugin(pl)
# Simple debug backend
debug = Debug(
messages=[("message 1", 1), ("message 2", 2)],
on_complete=app.stop,
)
app.add_backend(debug)
# Run application
app.run()
# Check replies
assert debug.answers[1] == [("message 1", (), {})]
assert debug.answers[2] == [("message 2", (), {})]
def make_kutana_no_run(backend_source=None):
app = Kutana()
debug = Debug(messages=[])
if backend_source:
debug.get_identity = lambda: backend_source
app.add_backend(debug)
async def _handle_update(update):
ctx = await Context.create(
app=app,
config={
"prefixes": (".",),
"mention_prefix": ("",),
"ignore_initial_spaces": True
},
update=update,
backend=debug
)
return await app._handle_update(update, ctx)
def handle_update(update):
return app.get_loop().run_until_complete(_handle_update(update))
return app, debug, handle_update
def test_get_backend():
app = Kutana()
app.add_backend(Debug([], name="backend1"))
app.add_backend(Debug([], name="backend2"))
assert app.get_backend("backend1").name == "backend1"
assert app.get_backend("backend2").name == "backend2"
assert app.get_backend("backend3") is None
def test_not_active_backend():
app = Kutana()
app.add_backend(Debug([], name="backend1", active=False))
async def test():
with patch('asyncio.ensure_future') as ensure_future:
await app._on_start(None)
ensure_future.assert_not_called
asyncio.get_event_loop().run_until_complete(test())
def test_incorrect_values():
app = Kutana()
with pytest.raises(ValueError):
app.add_backend(None)
with pytest.raises(ValueError):
app.set_storage('permanent', None)
with pytest.raises(ValueError):
app.add_plugin(None)
def make_kutana(messages):
app = Kutana()
debug = Debug(
messages=messages,
on_complete=app.stop,
)
app.add_backend(debug)
return app, debug
def test_same_plugins_and_backends():
app = Kutana()
plugin = Plugin("")
backend = Debug([])
app.add_plugin(plugin)
with pytest.raises(RuntimeError):
app.add_plugin(plugin)
app.add_backend(backend)
with pytest.raises(RuntimeError):
app.add_backend(backend)
assert app.get_backends() == [backend]
def main():
app = Kutana()
# set_logger_level(logging.DEBUG)
backend = Vkontakte(SHUF_SETTINGS['TOKEN'])
app.add_backend(backend)
app.config['settings'] = SHUF_SETTINGS
app.config['prefixes'] = ('еш ', 'есб ', 'esb ', 'ешаф ', 'eshuf '
) # ('шаф ', 'sb ', 'шб ', 'shuf ', 'shufbot ')
app.config['inform_time'] = time(20, 50) # 12:00
app.config['votekick_time'] = 5 * 60
init_db(app)
app.add_plugins(load_plugins(os.path.join(os.curdir, 'bot', 'plugins')))
return app
def test_happy_path(mock_post):
group_change_settings_update = {
"type": "group_change_settings",
"object": {
"changes": {
"screen_name": {"old_value": "", "new_value": "sdffff23f23"},
"title": {"old_value": "Спасибо", "new_value": "Спасибо 2"}
}
}
}
raw_updates = [
{},
{"type": "present"},
group_change_settings_update,
MESSAGES["not_message"],
MESSAGES["message"],
MESSAGES[".echo"],
MESSAGES[".echo chat"],
MESSAGES[".echo wa"],
]
answers = []
updated_longpoll = []
def acquire_updates(content_type=None):
if not raw_updates:
return {"updates": [], "ts": "100"}
if updated_longpoll == [1]:
return {"failed": 3}
return {"updates": [raw_updates.pop(0)], "ts": "0"}
mock_post.return_value.__aenter__.return_value.json = CoroutineMock(
side_effect=acquire_updates
)
class _VkontakteLongpoll(VkontakteLongpoll):
async def _get_response(self, method, kwargs={}):
if method == "groups.setLongPollSettings":
return {"response": 1}
if method == "groups.getById":
return {
"response": [
{"id": 1, "name": "group", "screen_name": "grp"},
],
}
if method == "groups.getLongPollServer":
updated_longpoll.append(1)
return {
"response": {
"server": "s",
"key": "k",
"ts": "1",
},
}
if method == "execute":
answers.extend(kwargs["code"].split("API.")[1:])
return {
"response": [1] * kwargs["code"].count("API."),
}
print(method, kwargs)
app = Kutana()
vkontakte = _VkontakteLongpoll(token="token")
app.add_backend(vkontakte)
echo_plugin = Plugin("echo")
@echo_plugin.on_commands(["echo", "ec"])
async def __(message, ctx):
assert ctx.resolve_screen_name
assert ctx.reply
await ctx.reply(message.text, attachments=message.attachments)
app.add_plugin(echo_plugin)
app.get_loop().call_later(
delay=vkontakte.api_request_pause * 4,
callback=app.stop,
)
app.run()
assert vkontakte.group_name == "Спасибо 2"
assert vkontakte.group_screen_name == "sdffff23f23"
assert len(updated_longpoll) == 2
answers.sort()
assert len(answers) == 3
assert '{"message": ".echo chat [michaelkrukov|Михаил]",' in answers[0]
assert '{"message": ".echo wa",' in answers[1]
assert 'attachment": ""' not in answers[1]
assert '{"message": ".echo",' in answers[2]
def test_add_backend():
app = Kutana()
with pytest.raises(ValueError):
app.add_backend(None)
def test_happy_path():
updates = [
MESSAGES["not_message"],
MESSAGES["message"],
MESSAGES[".echo"],
MESSAGES[".echo chat"],
MESSAGES["/echo@bot chat"],
MESSAGES["/echo chat"],
MESSAGES[".echo@bot chat"],
MESSAGES["_image"],
]
answers = []
class _Telegram(Telegram):
async def _request(self, method, kwargs={}):
if method == "getMe":
return {
"first_name": "te",
"last_name": "st",
"username": "******"
}
if method == "getUpdates":
if not updates:
return []
return [updates.pop(0)]
if method == "sendMessage":
answers.append(("msg", kwargs["text"]))
return 1
if method == "sendPhoto":
answers.append(("image", kwargs["photo"]))
return 1
print(method, kwargs)
app = Kutana()
telegram = _Telegram(token="token")
app.add_backend(telegram)
echo_plugin = Plugin("echo")
@echo_plugin.on_commands(["echo", "ec"])
async def _(message, ctx):
await ctx.reply(message.text)
@echo_plugin.on_attachments(["image"])
async def _(message, ctx):
await ctx.reply(message.text, attachments=message.attachments[0])
app.add_plugin(echo_plugin)
app.get_loop().call_later(
delay=0.5,
callback=app.stop,
)
app.run()
answers.sort()
assert len(answers) == 5
assert answers[0][0] == "image"
assert answers[1] == ("msg", ".echo")
assert answers[2] == ("msg", ".echo chat")
assert answers[3] == ("msg", "/echo chat")
assert answers[4] == ("msg", "/echo chat")
import json
from kutana import Kutana, load_plugins
from kutana.backends import Vkontakte
# Import configuration
with open("config.json") as fh:
config = json.load(fh)
# Create application
app = Kutana()
# Add manager to application
app.add_backend(Vkontakte(token=config["vk_token"]))
# Load and register plugins
app.add_plugins(load_plugins("plugins/"))
if __name__ == "__main__":
# Run application
app.run()
from ai_dungeon import config
from ai_dungeon.storage.redis import RedisStorage
from kutana import Kutana, load_plugins
from kutana.backends import Vkontakte, Telegram
import logging
logging.basicConfig(level=logging.DEBUG)
# Create application
if config.REDIS_URL is not None:
app = Kutana(storage=RedisStorage(config.REDIS_URL))
else:
app = Kutana()
# Add manager to application
if config.TELEGRAM_TOKEN:
app.add_backend(Telegram(token=config.TELEGRAM_TOKEN))
if config.VK_TOKEN:
app.add_backend(Vkontakte(token=config.VK_TOKEN))
# Load and register plugins
app.add_plugins(load_plugins("plugins/"))
if __name__ == "__main__":
# Run application
app.run()
import os
from pathlib import Path
from kutana import Kutana, load_plugins
from kutana.backends import Telegram
import dotenv
env = dotenv.dotenv_values() if Path(".env").exists() else os.environ
# Create application
app = Kutana()
# Add manager to application
app.add_backend(Telegram(env.get("TELEGRAM_TOKEN"), proxy=env.get("PROXY")))
# Load and register plugins
app.add_plugins(load_plugins("plugins/"))
if __name__ == "__main__":
# Run application
app.run()
import json
from kutana import Kutana, load_plugins
from kutana.backends import Vkontakte, Telegram
# Import configuration
with open("configuration.json") as fh:
config = json.load(fh)
# Create application
app = Kutana()
# Set config
app.config.update({
"vk_chat_id": config["vk_chat_id"],
"tg_chat_id": config["tg_chat_id"]
})
# Add manager to application
app.add_backend(Telegram(token=config["tg_token"]))
app.add_backend(Vkontakte(token=config["vk_token"]))
# Load and register plugins
app.add_plugins(load_plugins("plugins/"))
if __name__ == "__main__":
# Run application
app.run()
import json
from kutana import Kutana, load_plugins
from kutana.backends import Vkontakte, VkontakteCallback, Telegram
# Import configuration
with open("config.json") as fh:
config = json.load(fh)
# Create application
app = Kutana()
# Add backends to application
if "vk" in config:
if "address" in config["vk"]:
app.add_backend(
VkontakteCallback(token=config["vk"]["token"],
address=config["vk"]["address"]))
else:
app.add_backend(Vkontakte(token=config["vk"]["token"]))
if "tg" in config:
app.add_backend(Telegram(token=config["tg"]["token"]))
# Load and register plugins
app.add_plugins(load_plugins("plugins/"))
if __name__ == "__main__":
# Run application
app.run()
