-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
108 lines (80 loc) · 3.35 KB
/
main.py
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
99
100
101
102
103
104
105
106
107
108
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Simple Bot to reply to Telegram messages
# This program is dedicated to the public domain under the CC0 license.
"""
This Bot uses the Updater class to handle the bot.
First, a few handler functions are defined. Then, those functions are passed to
the Dispatcher and registered at their respective places.
Then, the bot is started and runs until we press Ctrl-C on the command line.
Usage:
Basic Echobot example, repeats messages.
Press Ctrl-C on the command line or send a signal to the process to stop the
bot.
"""
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
import logging
from Database import *
# Enable logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
logger = logging.getLogger(__name__)
DATABASE_FILE_NAME = 'likedmsg.db'
print("Loading database...")
msg_db = load_database(DATABASE_FILE_NAME)
print("Done loading database.")
# Define a few command handlers. These usually take the two arguments bot and
# update. Error handlers also receive the raised TelegramError object in error.
def start(bot, update):
bot.sendMessage(update.message.chat_id, text='Hi!')
def help(bot, update):
bot.sendMessage(update.message.chat_id, text='Help!')
def error(bot, update, error):
logger.warn('Update "%s" caused error "%s"' % (update, error))
def like(bot, update):
msg_db.insert(update.message)
bot.sendMessage(update.message.chat_id, text=msg_db.getStr(update.message))
save_database(msg_db, DATABASE_FILE_NAME)
def getTopLiked(bot, update):
chat_id = update.message.chat_id
bot.sendMessage(chat_id, text=msg_db.getTopNStr(chat_id, 10))
def getTopN(bot, update):
# Should make this more robust
num_top = update.message.text.split()[1]
chat_id = update.message.chat_id
bot.sendMessage(chat_id, text=msg_db.getTopNStr(chat_id, num_top))
def checkArgs(args, type_args):
if(len(args) == len(type_args)):
return "Use of topN: \"/topn [number]\""
def clear_db(bot, update):
chat_id = update.message.chat_id
msg_db.clearForChat(chat_id)
bot.sendMessage(update.message.chat_id, text='The database of liked messages has been deleted')
def main():
with open("botCode.txt", "r") as f:
lines = f.readlines()
api_token = lines[0].rstrip('\n')
# Create the EventHandler and pass it your bot's token.
updater = Updater(api_token)
# Get the dispatcher to register handlers
dp = updater.dispatcher
# on different commands - answer in Telegram
dp.add_handler(CommandHandler("start", start))
dp.add_handler(CommandHandler("help", help))
dp.add_handler(CommandHandler("l", like))
dp.add_handler(CommandHandler("top", getTopLiked))
dp.add_handler(CommandHandler("topn", getTopN))
dp.add_handler(CommandHandler("cleardb", clear_db))
# on noncommand i.e message - echo the message on Telegram
# dp.add_handler(MessageHandler([Filters.text], echo))
# log all errors
dp.add_error_handler(error)
# Start the Bot
updater.start_polling()
# Run the bot until the you presses Ctrl-C or the process receives SIGINT,
# SIGTERM or SIGABRT. This should be used most of the time, since
# start_polling() is non-blocking and will stop the bot gracefully.
updater.idle()
if __name__ == '__main__':
main()