telegram bot commands group

Author: ProgrammingLanguageLeader

Procfile

@@ -1,3 +1,3 @@
 web: gunicorn system.wsgi:app
-webhook: flask webhook
+webhook: flask telegram webhook
 upgrade: flask db upgrade

system/commands.py

@@ -1,23 +1,5 @@
-import click
-
-from flask.cli import with_appcontext
-
-import telegram_bot.polling
-import telegram_bot.webhook
-
-
-@click.command('polling')
-@with_appcontext
-def start_polling():
-    telegram_bot.polling.start_polling()
-
-
-@click.command('webhook')
-@with_appcontext
-def start_webhook():
-    telegram_bot.webhook.set_webhook()
+from telegram_bot.commands import telegram_cli
 
 
 def register_commands(app):
-    app.cli.add_command(start_polling)
-    app.cli.add_command(start_webhook)
+    app.cli.add_command(telegram_cli)

telegram_bot/commands.py

@@ -0,0 +1,25 @@
+from flask.cli import AppGroup, with_appcontext
+
+import telegram_bot.polling
+import telegram_bot.webhook
+
+
+telegram_cli = AppGroup('telegram')
+
+
+@telegram_cli.command('polling')
+@with_appcontext
+def start_polling():
+    """
+    Starts telegram bot in the long-polling mode
+    """
+    telegram_bot.polling.start_polling()
+
+
+@telegram_cli.command('webhook')
+@with_appcontext
+def start_webhook():
+    """
+    Starts telegram bot in the webhook mode
+    """
+    telegram_bot.webhook.set_webhook()

telegram_bot/webhook.py

@@ -4,9 +4,9 @@
 
 
 def set_webhook():
-    host_url = current_app.config['HOST_URL']
-    host_port = current_app.config['HOST_PORT']
-    telegram_token = current_app.config['TELEGRAM_TOKEN']
+    host_url = current_app.config.get('HOST_URL')
+    host_port = current_app.config.get('HOST_PORT')
+    telegram_token = current_app.config.get('TELEGRAM_TOKEN')
     bot = TelegramBot(telegram_token)
     bot.set_webhook(
         webhook_url='https://%s:%s/telegram/%s' % (