remindmoi_bot_handler.py

import json
import requests

from typing import Any, Dict

from bot_helpers import (ADD_ENDPOINT,
                         REMOVE_ENDPOINT,
                         LIST_ENDPOINT,
                         REPEAT_ENDPOINT,
                         MULTI_REMIND_ENDPOINT,
                         is_add_command,
                         is_remove_command,
                         is_list_command,
                         is_repeat_reminder_command,
                         is_multi_remind_command,
                         parse_add_command_content,
                         parse_remove_command_content,
                         generate_reminders_list,
                         parse_repeat_command_content,
                         parse_multi_remind_command_content)


USAGE = '''
A bot that schedules reminders for users.

To store a reminder, mention or send a message to the bot in the following format:

`add int <UNIT> <title_of_reminder>`

`add 1 day clean the dishes`
`add 10 hours eat`

Avaliable time units: minutes, hours, days, weeks

To remove a reminder:
`remove <reminder_id>`

To list reminders:
`list`

To repeat a reminder: 
repeat <reminder_id> every <int> <time_unit>

`repeat 23 every 2 weeks`

Avaliable units: days, weeks, months

'''


class RemindMoiHandler(object):
    '''
    A docstring documenting this bot.
    the reminder bot reminds people of its reminders
    '''

    def usage(self) -> str:
        return USAGE

    def handle_message(self, message: Dict[str, Any], bot_handler: Any) -> None:
        bot_response = get_bot_response(message, bot_handler)
        bot_handler.send_reply(message, bot_response)


def get_bot_response(message: Dict[str, Any], bot_handler: Any) -> str:
    if message['content'].startswith(('help', '?', 'halp')):
        return USAGE

    try:
        if is_add_command(message['content']):
            reminder_object = parse_add_command_content(message)
            response = requests.post(url=ADD_ENDPOINT, json=reminder_object)
            response = response.json()
            assert response['success']
            return f"Reminder stored. Your reminder id is: {response['reminder_id']}"
        if is_remove_command(message['content']):
            reminder_id = parse_remove_command_content(message['content'])
            response = requests.post(url=REMOVE_ENDPOINT, json=reminder_id)
            response = response.json()
            assert response['success']
            return "Reminder deleted."
        if is_list_command(message['content']):
            zulip_user_email = {'zulip_user_email': message["sender_email"]}
            response = requests.post(url=LIST_ENDPOINT, json=zulip_user_email)
            response = response.json()
            assert response['success']
            return generate_reminders_list(response)
        if is_repeat_reminder_command(message['content']):
            repeat_request = parse_repeat_command_content(message['content'])
            response = requests.post(url=REPEAT_ENDPOINT, json=repeat_request)
            response = response.json()
            assert response['success']
            return f"Reminder will be repeated every {repeat_request['repeat_value']} {repeat_request['repeat_unit']}."
        if is_multi_remind_command(message['content']):
            multi_remind_request = parse_multi_remind_command_content(message['content'])
            response = requests.post(url=MULTI_REMIND_ENDPOINT, json=multi_remind_request)
            response = response.json()
            assert response['success']
            return f"Reminder will be sent to the specified recepients."  # Todo: add list of recepients
        return "Invalid input. Please check help."
    except requests.exceptions.ConnectionError:
        return "Server not running, call Karim"
    except (json.JSONDecodeError, AssertionError):
        return "Something went wrong"
    except OverflowError:
        return "What's wrong with you?"


handler_class = RemindMoiHandler