abf9f03e88
Added string generator function to Task Added addTask compatibility with optional args Added some exception handling to addTask
112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
|
|
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
|
|
from telegram.error import *
|
|
from chat import *
|
|
import argparse
|
|
import logging
|
|
|
|
chats = {}
|
|
|
|
# Enable logging
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def get_chatname(chat):
|
|
if chat.title is not None:
|
|
return chat.title
|
|
elif chat.first_name is not None:
|
|
if chat.last_name is not None:
|
|
return chat.first_name + " " + chat.last_name
|
|
else:
|
|
return chat.first_name
|
|
else:
|
|
return ""
|
|
def error(bot, update, error):
|
|
logger.warn('Update "%s" caused error "%s"' % (update, error))
|
|
|
|
def start(bot, update):
|
|
update.message.reply_text('Hello, World!!')
|
|
|
|
def addTask(bot, update,args):
|
|
global chats
|
|
tchat = update.message.chat
|
|
ident = str(tchat.id)
|
|
if not ident in chats:
|
|
title = get_chatname(tchat)
|
|
chat = Chat(tchat.id, title)
|
|
else:
|
|
chat = chats[ident]
|
|
try:
|
|
if len(args) > 5:
|
|
t = chat.addTask(args[0],args[1],int(args[2]),int(args[3]),int(args[4]),int(args[5]))
|
|
elif len(args) > 4:
|
|
t = chat.addTask(args[0],args[1],int(args[2]),int(args[3]),int(args[4]))
|
|
elif len(args) > 3:
|
|
t = chat.addTask(args[0],args[1],int(args[2]),int(args[3]))
|
|
elif len(args) > 2:
|
|
t = chat.addTask(args[0],args[1],int(args[2]))
|
|
elif len(args) > 1:
|
|
t = chat.addTask(args[0],args[1])
|
|
else:
|
|
t = chat.addTask(args[0])
|
|
update.message.reply_text('Task ' + t.name + ' added')
|
|
chats[ident] = chat
|
|
except:
|
|
update.message.reply_text('Wrong format')
|
|
|
|
def getTasksToday(bot,update):
|
|
global chats
|
|
tchat = update.message.chat
|
|
ident = str(tchat.id)
|
|
if not ident in chats:
|
|
title = get_chatname(tchat)
|
|
chat = Chat(tchat.id, title)
|
|
else:
|
|
chat = chats[ident]
|
|
update.message.reply_text('showing today tasks:')
|
|
dateNow = datetime.now()
|
|
tasks = chat.getTasks()
|
|
for task in tasks:
|
|
update.message.reply_text('leiendo task')
|
|
dateTask = task.getDate()
|
|
update.message.reply_text(task.toString())
|
|
update.message.reply_text('thats all for today')
|
|
chats[ident] = chat
|
|
def stop(bot, update):
|
|
pass
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='A Telegram task manager bot.')
|
|
parser.add_argument('token', metavar='TOKEN', help='The Bot Token to work with the Telegram Bot API')
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Create the EventHandler and pass it your bot's token.
|
|
updater = Updater(args.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("tasksToday", getTasksToday))
|
|
dp.add_handler(CommandHandler("new", addTask, pass_args=True))
|
|
#dp.add_handler(CommandHandler("about", about))
|
|
#dp.add_handler(CommandHandler("help", help))
|
|
dp.add_handler(CommandHandler("stop", stop))
|
|
|
|
# log all errors
|
|
dp.add_error_handler(error)
|
|
|
|
# Start the Bot
|
|
updater.start_polling()
|
|
|
|
# Run the bot until you press 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()
|