53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import functools
|
|
|
|
from sqlalchemy.future import Engine
|
|
import telebot
|
|
|
|
from . import handlers, utils
|
|
|
|
__all__ = ["handlers", "utils"]
|
|
|
|
|
|
def create_bot(token: str, engine: Engine) -> telebot.TeleBot:
|
|
bot = telebot.TeleBot(token)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.set_master_password, bot, engine),
|
|
commands=["set_master_pass"],
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.get_account, bot, engine), commands=["get_account"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.get_accounts, bot, engine), commands=["get_accounts"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.add_account, bot, engine), commands=["add_account"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.delete_all, bot, engine), commands=["delete_all"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.reset_master_pass, bot, engine),
|
|
commands=["reset_master_pass"],
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.delete_account, bot, engine),
|
|
commands=["delete_account"],
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.help, bot), commands=["help", "start"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.cancel, bot), commands=["cancel"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.export, bot, engine), commands=["export"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.import_accounts, bot, engine), commands=["import"]
|
|
)
|
|
bot.register_message_handler(
|
|
functools.partial(handlers.gen_password, bot), commands=["gen_password"]
|
|
)
|
|
return bot
|