blob: 5282ae9eeb627394907dd0a30e5e42a4cad75d0e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
#!/usr/env python
import sys, argparse, traceback
import time
# Make a globally accessible bot instance
bot = None
# Try to import telepot
# If not installed, inform user and fail gracefully
try:
import telepot
except:
print("radiodiodibot needs the telepot module to communicate with Telegram.")
print("Please install telepot before using radiodiodibot.")
sys.exit()
def handle(msg):
content_type, chat_type, chat_id = telepot.glance(msg)
print(content_type, chat_type, chat_id)
if content_type == 'text':
bot.sendMessage(chat_id, "Radio palaa keväällä 2017!")
# Entry point
def main():
global bot
# Get the bot token as an argument from the user
parser = argparse.ArgumentParser(description="Telegram bot for the Radiodiodi Student Radio broadcast.")
parser.add_argument("telegram_bot_token")
args = parser.parse_args()
token = args.telegram_bot_token
# Attempt to create a bot with telepot
try:
print("Creating bot listener with token {}...".format(token))
bot = telepot.Bot(token)
print("Bot succesfully created.")
except:
traceback.print_exc()
print("Error creating bot listener. radiodiodibot will now exit...")
return
# Try fetching bot information from Telegram to check connection
try:
print("Bot info: {}".format(bot.getMe()))
except:
print("Could not fetch bot info. Check your token and connectivity to Telegram!")
return
bot.message_loop(handle)
print("Listening for messages...")
while True:
time.sleep(10)
if __name__ == "__main__":
main()
|
