1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
from telegram.ext import Updater, CommandHandler, ConversationHandler, MessageHandler, Filters
from telegram import Update, User, Bot # Typing
from config import TG_TOKEN
from enums import UtilStates, MainStates
from views import UtilViews, MainStatesView
from keyboards import MenuKeyboard, BACK
from logger_config import logger
def error(bot, update, error):
logger.warning('Update "%s" caused error "%s"', update, error)
def main():
conversation = ConversationHandler(
entry_points=[
CommandHandler('start', UtilViews.start),
],
states={
UtilStates.WAIT_FOR_USERNAME: [MessageHandler(Filters.text, UtilViews.wait_for_username)],
MainStates.MAIN_MENU: [
MessageHandler(Filters.regex(MenuKeyboard.NEW_CLIENT), MainStatesView.new_client_name),
MessageHandler(Filters.regex(MenuKeyboard.LIST_CLIENTS), MainStatesView.list_clients),
],
MainStates.ENTERING_NAME: [
MessageHandler(Filters.regex(BACK), MainStatesView.main_menu),
MessageHandler(Filters.text, MainStatesView.new_client),
],
},
fallbacks=[
MessageHandler(Filters.text, UtilViews.fallback),
]
)
updater = Updater(TG_TOKEN)
dp = updater.dispatcher
dp.add_error_handler(error)
dp.add_handler(conversation)
updater.start_polling()
updater.idle()
if __name__ == "__main__":
main()
|
