mirror of
https://github.com/house-of-vanity/conf_bot.git
synced 2025-07-06 14:24:08 +00:00
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
import logging
|
||
|
|
import settings
|
||
|
|
import signal
|
||
|
|
import sys
|
||
|
|
#from webhook import WebHook
|
||
|
|
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
|
||
|
|
|
||
|
|
logging.basicConfig(level=logging.DEBUG,
|
||
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
# catch ctrl+c
|
||
|
|
def signal_handler(signal, frame):
|
||
|
|
print('Exiting...')
|
||
|
|
settings.db.close()
|
||
|
|
sys.exit(0)
|
||
|
|
signal.signal(signal.SIGINT, signal_handler)
|
||
|
|
|
||
|
|
|
||
|
|
# Define a few command handlers. These usually take the two arguments bot and
|
||
|
|
# update. Error handlers also receive the raised TelegramError object in error.
|
||
|
|
def start(update, context):
|
||
|
|
"""Send a message when the command /start is issued."""
|
||
|
|
update.message.reply_text('Hi!')
|
||
|
|
|
||
|
|
|
||
|
|
def help(update, context):
|
||
|
|
"""Send a message when the command /help is issued."""
|
||
|
|
update.message.reply_text('Help!')
|
||
|
|
|
||
|
|
|
||
|
|
def echo(update, context):
|
||
|
|
"""Echo the user message."""
|
||
|
|
update.message.reply_text(update.message.text)
|
||
|
|
|
||
|
|
|
||
|
|
def error(update, context):
|
||
|
|
"""Log Errors caused by Updates."""
|
||
|
|
logger.warning('Update "%s" caused error "%s"', update, context.error)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""Start the bot."""
|
||
|
|
# Create the Updater and pass it your bot's token.
|
||
|
|
# Make sure to set use_context=True to use the new context based callbacks
|
||
|
|
# Post version 12 this will no longer be necessary
|
||
|
|
token = settings.parser.get('bot', 'telegram_key')
|
||
|
|
updater = Updater(token, use_context=True)
|
||
|
|
|
||
|
|
# Get the dispatcher to register handlers
|
||
|
|
dp = updater.dispatcher
|
||
|
|
|
||
|
|
# on different commands - answer in Telegram
|
||
|
|
dp.add_handler(CommandHandler("start", start))
|
||
|
|
dp.add_handler(CommandHandler("help", help))
|
||
|
|
|
||
|
|
# on noncommand i.e message - echo the message on Telegram
|
||
|
|
dp.add_handler(MessageHandler(Filters.text, echo))
|
||
|
|
|
||
|
|
# log all errors
|
||
|
|
dp.add_error_handler(error)
|
||
|
|
|
||
|
|
# Start the Bot
|
||
|
|
updater.start_polling()
|
||
|
|
|
||
|
|
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
||
|
|
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||
|
|
# start_polling() is non-blocking and will stop the bot gracefully.
|
||
|
|
updater.idle()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|