AddCommand

Complexity

Total Complexity

8

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

4 Methods

Rating	Name	Duplication	Size	Complexity
A	default()	0	8	3
A	add()	0	3	1
A	process_howmuch()	0	9	2
A	get_amount()	0	8	2

import re
from functools import partial

import telegram

from . import BotCommand, CommandError


class AddCommand(BotCommand):
    command = '/add'

    def get_amount(self, content):
        match = re.match('((/add )|(/remove ))?(?P<amount>\d+(\.\d+)?)( (?P<reason>.*))?', content)
        if not match:
            raise CommandError()

        amount = float(match.groupdict()['amount'])
        reason = match.groupdict()['reason']
        return amount, reason

    def add(self, tab_id, user_id, message_id, date, amount, reason=''):
        tab = self.get_tab(tab_id)
        tab.add(message_id, user_id, date, amount, reason)

    def default(self, message):
        content = message.text.split(' ', 1)[1] if len(message.text.split(' ', 1)) == 2 else ''
        if content:
            self.process_howmuch(message)
        else:
            msg = self.bot.say(message, 'How much?',
                               reply_markup=telegram.ForceReply(selective=True))
            self.queue(msg, partial(self.process_howmuch))

    def process_howmuch(self, message):
        try:
            amount, reason = self.get_amount(message.text)
        except CommandError:
            self.bot.say(message, "Nope, I don't get ya")
            return
        self.add(message.chat.id, message.from_user.id, message.message_id,
                 message.date, amount, reason)
        self.bot.say(message, telegram.Emoji.THUMBS_UP_SIGN)