mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-11-21 22:56:38 +01:00
improve echobot with cleaner code and basic exception handling
This commit is contained in:
parent
c64e577049
commit
ff0d724f32
1 changed files with 29 additions and 23 deletions
|
@ -1,6 +1,7 @@
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
#
|
#
|
||||||
# Simple Bot to reply Telegram messages
|
# Simple Bot to reply to Telegram messages
|
||||||
# Copyright (C) 2015 Leandro Toledo de Souza <leandrotoeldodesouza@gmail.com>
|
# Copyright (C) 2015 Leandro Toledo de Souza <leandrotoeldodesouza@gmail.com>
|
||||||
#
|
#
|
||||||
# This program is free software: you can redistribute it and/or modify
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
@ -19,13 +20,16 @@
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import telegram
|
import telegram
|
||||||
|
from time import sleep
|
||||||
|
|
||||||
|
try:
|
||||||
LAST_UPDATE_ID = None
|
from urllib.error import URLError
|
||||||
|
except ImportError:
|
||||||
|
from urllib2 import URLError # python 2
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
global LAST_UPDATE_ID
|
update_id = None
|
||||||
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||||||
|
@ -33,33 +37,35 @@ def main():
|
||||||
# Telegram Bot Authorization Token
|
# Telegram Bot Authorization Token
|
||||||
bot = telegram.Bot('TOKEN')
|
bot = telegram.Bot('TOKEN')
|
||||||
|
|
||||||
# This will be our global variable to keep the latest update_id when requesting
|
|
||||||
# for updates. It starts with the latest update_id if available.
|
|
||||||
try:
|
|
||||||
LAST_UPDATE_ID = bot.getUpdates()[-1].update_id
|
|
||||||
except IndexError:
|
|
||||||
LAST_UPDATE_ID = None
|
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
echo(bot)
|
try:
|
||||||
|
update_id = echo(bot, update_id)
|
||||||
|
except telegram.TelegramError as e:
|
||||||
|
# These are network problems with Telegram.
|
||||||
|
if e.message in ("Bad Gateway", "Timed out"):
|
||||||
|
sleep(1)
|
||||||
|
else:
|
||||||
|
raise e
|
||||||
|
except URLError as e:
|
||||||
|
# These are network problems on our end.
|
||||||
|
sleep(1)
|
||||||
|
|
||||||
|
|
||||||
def echo(bot):
|
def echo(bot, update_id):
|
||||||
global LAST_UPDATE_ID
|
|
||||||
|
|
||||||
# Request updates after the last updated_id
|
# Request updates after the last update_id
|
||||||
for update in bot.getUpdates(offset=LAST_UPDATE_ID, timeout=10):
|
for update in bot.getUpdates(offset=update_id, timeout=10):
|
||||||
# chat_id is required to reply any message
|
# chat_id is required to reply to any message
|
||||||
chat_id = update.message.chat_id
|
chat_id = update.message.chat_id
|
||||||
reply_text = update.message.text
|
update_id = update.update_id + 1
|
||||||
|
message = update.message.text
|
||||||
|
|
||||||
if reply_text:
|
if message:
|
||||||
# Reply the message
|
# Reply to the message
|
||||||
bot.sendMessage(chat_id=chat_id,
|
bot.sendMessage(chat_id=chat_id,
|
||||||
text=reply_text)
|
text=message)
|
||||||
|
|
||||||
# Updates global offset to get the new updates
|
return update_id
|
||||||
LAST_UPDATE_ID = update.update_id + 1
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
Loading…
Reference in a new issue