nikdoof/smsbot
1
0
Fork 0
mirror of https://github.com/nikdoof/smsbot.git synced 2025-12-23 06:39:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Prometheus metrics

Browse Source
This commit is contained in:
Andrew Williams
2022-06-09 12:11:48 +01:00
parent db04de242a
commit 7f481c47ed
4 changed files with 38 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
smsbot/telegram.py
View File

@@ -4,24 +4,27 @@ from setuptools import Command
from telegram.ext import CommandHandler, Updater
from smsbot.utils import get_smsbot_version
from prometheus_client import make_wsgi_app, Counter, Summary
REQUEST_TIME = Summary('telegram_request_processing_seconds', 'Time spent processing request')
COMMAND_COUNT = Counter('telegram_command_count', 'Total number of commands processed')
class TelegramSmsBot(object):
def __init__(self, token, allow_subscribing=False, owner=None, subscribers=None):
def __init__(self, telegram_token, allow_subscribing=False, owner=None, subscribers=None):
self.logger = logging.getLogger(self.__class__.__name__)
self.bot_token = token
self.bot_token = telegram_token
self.subscriber_ids = subscribers or []
self.set_owner(owner)
self.updater = Updater(self.bot_token, use_context=True)
self.updater.dispatcher.add_handler(CommandHandler('help', self.help_handler))
self.updater.dispatcher.add_handler(CommandHandler('start', self.help_handler))
if allow_subscribing:
self.updater.dispatcher.add_handler(CommandHandler('subscribe', self.subscribe_handler))
self.updater.dispatcher.add_handler(CommandHandler('unsubscribe', self.unsubscribe_handler))
self.updater.dispatcher.add_error_handler(self.error_handler)
def start(self):
@@ -33,6 +36,7 @@ class TelegramSmsBot(object):
def stop(self):
self.updater.stop()
@REQUEST_TIME.time()
def help_handler(self, update, context):
"""Send a message when the command /help is issued."""
self.logger.info('/help command received in chat: %s', update.message.chat)
@@ -42,7 +46,9 @@ class TelegramSmsBot(object):
commands.extend(['/{0}'.format(x) for x in command.command])
update.message.reply_markdown('Smsbot v{0}\n\n{1}'.format(get_smsbot_version(), '\n'.join(commands)))
COMMAND_COUNT.inc()
@REQUEST_TIME.time()
def subscribe_handler(self, update, context):
self.logger.info('/subscribe command received')
if update.message.chat['id'] not in self.subscriber_ids:
@@ -52,7 +58,9 @@ class TelegramSmsBot(object):
update.message.reply_markdown('You have been subscribed to SMS notifications')
else:
update.message.reply_markdown('You are already subscribed to SMS notifications')
COMMAND_COUNT.inc()
@REQUEST_TIME.time()
def unsubscribe_handler(self, update, context):
self.logger.info('/unsubscribe command received')
if update.message.chat['id'] in self.subscriber_ids:
@@ -62,6 +70,7 @@ class TelegramSmsBot(object):
update.message.reply_markdown('You have been unsubscribed to SMS notifications')
else:
update.message.reply_markdown('You are not subscribed to SMS notifications')
COMMAND_COUNT.inc()
def error_handler(self, update, context):
"""Log Errors caused by Updates."""