nxy/bot/rape.py

# -*- coding: utf-8 -*-
import random

from docopt import Dict
from irc3.plugins.command import command
from irc3.utils import IrcString

from . import DatabasePlugin


class Rape(DatabasePlugin):
    @command
    def owe(self, mask: IrcString, target: IrcString, args: Dict):
        """Shows how much a nick owes

        %%owe [<nick>]
        """
        nick = args.get('<nick>', mask.nick)

        # Fetch result from database
        self.cur.execute('''
        SELECT
          fines
        FROM
          users
        WHERE
          lower(nick) = lower(%s)
        ''', [nick])
        owes = self.cur.fetchone()

        # Colorize owe amount and return string
        if owes:
            fines = '4${fines}'.format(fines=owes['fines'])
        else:
            fines = '3$0'

        # Return total owes
        return '{nick} owes: \x03{fines}\x03'.format(nick=nick, fines=fines)

    @command
    def rape(self, mask: IrcString, target: IrcString, args: Dict):
        """Rapes a nick and eventually charge for it

        %%rape <nick>
        """
        nick = args.get('<nick>', mask.nick)
        rand = random.randint(0, 3)

        if rand not in (0, 1):
            self.bot.action(target, 'rapes {nick}'.format(nick=nick))
        else:
            fine = random.randint(1, 500)

            # Insert or add fine to database and return total owe
            self.cur.execute('''
            INSERT INTO
              users (nick, fines)
            VALUES
              (lower(%s), %s)
            ON CONFLICT (nick) DO UPDATE SET
              fines = users.fines + excluded.fines
            RETURNING
              fines
            ''', [mask.nick, fine])
            self.con.commit()

            # Get reason based on rand value
            reason = ('raping', 'being too lewd and getting raped')[rand]

            # Print fine and total owe
            self.bot.action(target, 'fines {nick} \x02${fine}\x02 for {reason}. You owe: \x0304${total}\x03'.format(
                nick=mask.nick,
                fine=fine,
                reason=reason,
                total=self.cur.fetchone()['fines'],
            ))