Mode switching fix & pep8

Author: ProgrammingLanguageLeader

telegram_bot/handlers/commands/detailed_mode.py

@@ -1,12 +1,15 @@
 from system.db import db
+from telegram_bot.handlers.utils.decorators import remember_new_user, \
+    send_typing, write_logs
+from telegram_bot.handlers.utils.menu_entries import MenuEntry
+from telegram_bot.handlers.utils.reply_markup import create_main_reply_markup
 from telegram_bot.models import User
-from telegram_bot.handlers.utils.decorators import remember_new_user, send_typing, write_logs
 
 
 @write_logs
 @send_typing
 @remember_new_user
-def handle_detailed_mode_cmd(bot, update) -> None:
+def handle_detailed_mode_cmd(bot, update) -> int:
     db.session.query(User).filter_by(
         telegram_id=update.message.from_user.id
     ).update({
@@ -15,5 +18,7 @@ def handle_detailed_mode_cmd(bot, update) -> None:
     db.session.commit()
     bot.send_message(
         chat_id=update.message.chat_id,
-        text='Switched to detailed mode'
+        text='Switched to detailed mode',
+        reply_markup=create_main_reply_markup()
     )
+    return MenuEntry.START_MENU.value

telegram_bot/handlers/commands/examples.py

@@ -1,6 +1,7 @@
 from telegram_bot.handlers.utils.menu_entries import MenuEntry
 from telegram_bot.handlers.utils.reply_markup import create_main_reply_markup
-from telegram_bot.handlers.utils.decorators import write_logs, remember_new_user, send_typing
+from telegram_bot.handlers.utils.decorators import write_logs, \
+    remember_new_user, send_typing
 
 
 @write_logs

telegram_bot/handlers/commands/help.py

@@ -1,6 +1,7 @@
 from telegram_bot.handlers.utils.menu_entries import MenuEntry
 from telegram_bot.handlers.utils.reply_markup import create_main_reply_markup
-from telegram_bot.handlers.utils.decorators import write_logs, send_typing, remember_new_user
+from telegram_bot.handlers.utils.decorators import write_logs, \
+    send_typing, remember_new_user
 
 
 @write_logs

telegram_bot/handlers/commands/simple_mode.py

@@ -1,12 +1,15 @@
 from system.db import db
+from telegram_bot.handlers.utils.decorators import write_logs, send_typing, \
+    remember_new_user
+from telegram_bot.handlers.utils.menu_entries import MenuEntry
+from telegram_bot.handlers.utils.reply_markup import create_main_reply_markup
 from telegram_bot.models import User
-from telegram_bot.handlers.utils.decorators import write_logs, send_typing, remember_new_user
 
 
 @write_logs
 @send_typing
 @remember_new_user
-def handle_simple_mode_cmd(bot, update) -> None:
+def handle_simple_mode_cmd(bot, update) -> int:
     db.session.query(User).filter_by(
         telegram_id=update.message.from_user.id
     ).update({
@@ -15,5 +18,7 @@ def handle_simple_mode_cmd(bot, update) -> None:
     db.session.commit()
     bot.send_message(
         chat_id=update.message.chat_id,
-        text='Switched to simple mode'
+        text='Switched to simple mode',
+        reply_markup=create_main_reply_markup()
     )
+    return MenuEntry.START_MENU.value

telegram_bot/handlers/commands/start.py

@@ -1,6 +1,7 @@
 from telegram_bot.handlers.utils.menu_entries import MenuEntry
 from telegram_bot.handlers.utils.reply_markup import create_main_reply_markup
-from telegram_bot.handlers.utils.decorators import write_logs, send_typing, remember_new_user
+from telegram_bot.handlers.utils.decorators import write_logs, \
+    send_typing, remember_new_user
 
 
 @write_logs

telegram_bot/handlers/messages/mode_toggling.py

@@ -1,13 +1,14 @@
 from system.db import db
 from telegram_bot.handlers.utils.decorators import write_logs, send_typing, \
     remember_new_user
+from telegram_bot.handlers.utils.menu_entries import MenuEntry
 from telegram_bot.models import User
 
 
 @write_logs
 @send_typing
 @remember_new_user
-def handle_mode_toggling(bot, update) -> None:
+def handle_mode_toggling(bot, update) -> int:
     current_user = db.session.query(User).filter_by(
         telegram_id=update.message.from_user.id
     ).first()
@@ -20,3 +21,4 @@ def handle_mode_toggling(bot, update) -> None:
             'simple' if simple_mode else 'detailed'
         )
     )
+    return MenuEntry.START_MENU.value