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
|
#!/usr/bin/env python
import sys
import argparse
import botmanager
import logging
try: # Python 2.7+
from logging import NullHandler
except ImportError:
class NullHandler(logging.Handler):
def emit(self, record):
pass
logging.getLogger(__name__).addHandler(NullHandler())
# 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", help="Telegram Bot API token")
parser.add_argument("-U", "--shoutbox-api-url", help="Shoutbox API URL")
parser.add_argument("-C", "--telegram-chat-id", help="Telegram Chat ID")
parser.add_argument("-v", "--verbose", help="Verbose output", action="store_true")
parser.add_argument("-i", "--interval", help="Delay between API update calls in seconds", type=int, default=10)
args = parser.parse_args()
if args.verbose:
logging_level = logging.INFO
else:
logging_level = logging.WARNING
logging.basicConfig(level=logging_level, format="%(asctime)s %(levelname)s %(message)s")
# 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(1)
# Entry point
def main():
token = args.telegram_bot_token
shoutbox_api_url = args.shoutbox_api_url
telegram_chat_id = args.telegram_chat_id
manager = botmanager.BotManager(token)
if shoutbox_api_url is not None:
manager.shoutbox_api_url = shoutbox_api_url
if telegram_chat_id is not None:
manager.telegram_chat_id = telegram_chat_id
manager.api_call_interval = args.interval
logging.info("Using Shoutbox API URL: {}".format(manager.shoutbox_api_url))
logging.info("Using Telegram Chat ID: {}".format(manager.telegram_chat_id))
logging.info("Using API call interval of {} seconds.".format(manager.api_call_interval))
manager.start()
if __name__ == "__main__":
main()
|
