blob: 6fe394e067ef91ca31929fe9eb8691b8d7e8d963 (
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
|
#!/usr/bin/env python
import sys, 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())
logging.basicConfig(level=logging.INFO, 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():
# 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")
args = parser.parse_args()
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 != None):
manager.shoutbox_api_url = shoutbox_api_url
if telegram_chat_id != None:
manager.telegram_chat_id = telegram_chat_id
logging.info("Using Shoutbox API URL: {}".format(manager.shoutbox_api_url))
logging.info("Using Telegram Chat ID: {}".format(manager.telegram_chat_id))
manager.start()
if __name__ == "__main__":
main()
|
