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
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#!/usr/bin/env python3
import argparse
import configparser
import logging
import signal
import sys
from project import botmanager
def sigint_handler(signal, frame):
"""Exit gracefully when receiving an interrupt signal"""
print("Exiting src...")
sys.exit(0)
signal.signal(signal.SIGINT, sigint_handler)
try: # Python 2.7+
from logging import NullHandler
except ImportError:
class NullHandler(logging.Handler):
def emit(self, record):
pass
logging.getLogger(__name__).addHandler(NullHandler())
# Read configs, unsuccessful reads are
# silently ignored
CONFIG_FILE = "bot.config"
config = configparser.ConfigParser()
config.read(CONFIG_FILE)
# 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("-t", "--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)
args = parser.parse_args()
# Show verbose output with the -v option
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("src needs the telepot module to communicate with Telegram.")
print("Please install telepot before using src.")
sys.exit(1)
# Entry point
def main():
print("=== Radiodiodibot ===")
print("Press CTRL-C to exit.")
# Get default parameter values from the config file
telegram_bot_token = config["GENERAL"]["TelegramBotToken"]
shoutbox_api_url = config["GENERAL"]["ShoutboxApiUrl"]
telegram_chat_id = config["GENERAL"]["TelegramChatID"]
api_call_interval = int(config["GENERAL"]["ApiCallInterval"])
api_auth_token = config["GENERAL"]["ApiAuthToken"]
# If the user has specified the parameters as command line
# arguments, use them to override the config values
if args.token is not None:
telegram_bot_token = args.token
if args.shoutbox_api_url is not None:
shoutbox_api_url = args.shoutbox_api_url
if args.telegram_chat_id is not None:
telegram_chat_id = args.telegram_chat_id
if args.interval is not None:
api_call_interval = args.interval
# Store the final values in a manager instance
manager = botmanager.BotManager(telegram_bot_token)
manager.set_parameters(shoutbox_api_url, telegram_chat_id, api_call_interval, api_auth_token)
logging.info("Using Shoutbox API URL: {}".format(shoutbox_api_url))
logging.info("Using Telegram Chat ID: {}".format(telegram_chat_id))
logging.info("Using API call interval of {} seconds.".format(api_call_interval))
# Start listening
manager.start()
if __name__ == "__main__":
main()
|
