Huge cleanup and refactoring :>

2017-08-22 17:43:48 +02:00
parent 7f5571fc09
commit f33a17038a
21 changed files with 127 additions and 185 deletions
--- a/bot/rape.py
+++ b/bot/rape.py
@ -0,0 +1,76 @@
+# -*- 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'],
+            ))