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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
# -*- coding: utf-8 -*-
import random
import sys
import telepot
import traceback
import time
import logging
from jsonfactory import JSONFactory
from shoutboxapicommunicator import Communicator
class BotManager(object):
shoutbox_api_url = "http://localhost:8000"
telegram_chat_id = "default_id"
def __init__(self, token):
self.token = token
# Attempt to create a bot with telepot
try:
logging.info("Creating bot listener with token {}...".format(token))
self.bot = telepot.Bot(token)
logging.info("Bot succesfully created.")
except:
traceback.print_exc()
logging.error("Error creating bot listener. radiodiodibot will now exit...")
sys.exit(1)
def start(self):
# Try fetching bot information from Telegram to check connection
try:
logging.info("Bot info: {}".format(self.bot.getMe()))
except:
logging.error("Could not fetch bot info. Check your token and connectivity to Telegram!")
sys.exit(1)
self.bot.message_loop(self.handle)
logging.info("Listening for messages...")
while True:
# TODO send to telegram
messages = Communicator.fetch(self.shoutbox_api_url)
for message in messages:
self.bot.sendMessage(self.telegram_chat_id,
"{}: {}".format(message["user"], message["text"]))
time.sleep(10)
def default_action(self, chat_id):
self.bot.sendMessage(chat_id, "Radio palaa keväällä 2017!")
# Placeholder action for testing commands
def now_playing(self, chat_id):
songs = ["Ace of Spades", "Mökkitie", "Alpha Russian XXL Night Mixtape", "teekkarihymni"]
self.bot.sendMessage(chat_id, "Radiossa soi {}!".format(random.choice(songs)))
def not_supported(self, chat_id):
self.bot.sendMessage(chat_id, "Tätä toimintoa ei ole tuettu.")
# Determine which action to take
def parse_message(self, msg):
t = msg["text"]
content_type, chat_type, chat_id = telepot.glance(msg)
if "/nowplaying" in t:
self.now_playing(chat_id)
elif "/start" in t:
self.not_supported(chat_id)
elif "/stop" in t:
self.not_supported(chat_id)
elif str(chat_id) == self.telegram_chat_id.strip():
user_name = msg["from"]["first_name"]
data = JSONFactory.make_object(t, user_name, msg["date"], "null")
logging.info("Created JSON packet:\n{}".format(data))
Communicator.send(self.shoutbox_api_url, data)
else:
logging.info("Got non-forwarded message from chat_id: {}".format(chat_id))
# Start parsing the incoming message if it is a text message
def handle(self, msg):
content_type, chat_type, chat_id = telepot.glance(msg)
if content_type == 'text':
self.parse_message(msg)
|
