jkhsjdhjs/nxy
1
0
Fork 0
Code Issues 5 Pull Requests Projects Releases Wiki Activity

Migrated rape to pgsql

Browse Source
This commit is contained in:
mrhanky 2017-06-29 23:57:15 +02:00
parent bfd438e8fa
commit 7ebcd7168d
No known key found for this signature in database
GPG Key ID: 67D772C481CB41B8
1 changed files with 16 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
nxy/plugins/rape.py
View File

@ -4,37 +4,34 @@ import random
from docopt import Dict as DocOptDict from docopt import Dict as DocOptDict
from irc3.plugins.command import command from irc3.plugins.command import command
from irc3.utils import IrcString from irc3.utils import IrcString
import irc3
from . import DatabasePlugin from . import DatabasePlugin
@irc3.plugin
class Rape(DatabasePlugin): class Rape(DatabasePlugin):
requires = ['irc3.plugins.command', requires = ['irc3.plugins.command',
'nxy.plugins.database'] 'nxy.plugins.database']
@command @command
def owe(self, mask: IrcString, channel: IrcString, args: DocOptDict): def owe(self, mask: IrcString, target: IrcString, args: DocOptDict):
"""Shows how much a nick owes. """Shows how much a nick owes.
%%owe [<nick>...] %%owe [<nick>]
""" """
nick = ' '.join(args.get('<nick>')) or mask.nick nick = args.get('<nick>') or mask.nick
if nick.startswith('@'): # Fetch result from database
nick = nick.lstrip('@').strip() self.cur.execute('select amount from owes where nick = %s', [nick])
self.cur.execute('select amount from owes where nick = ?', [nick])
owes = self.cur.fetchone() owes = self.cur.fetchone()
if owes: # Colorize owe amount and return string
total = '4${total}'.format(total=owes['amount']) total = '4${total}'.format(total=owes['amount']) if owes else '3$0'
else:
total = '3$0'
return '{nick} owes: \x03{total}\x03'.format(nick=nick, total=total) return '{nick} owes: \x03{total}\x03'.format(nick=nick, total=total)
@command @command
def rape(self, mask: IrcString, channel: IrcString, args: DocOptDict): def rape(self, mask: IrcString, target: IrcString, args: DocOptDict):
"""Rapes a nick and eventually charge for it. """Rapes a nick and eventually charge for it.
%%rape <nick> %%rape <nick>
@ -44,19 +41,12 @@ class Rape(DatabasePlugin):
rand = random.randint(0, 3) rand = random.randint(0, 3)
if rand in (0, 1): if rand in (0, 1):
self.cur.execute('select amount from owes where nick = ?', [nick]) self.cur.execute('''insert into owes (nick, amount) values (%s, %s)
owe = self.cur.fetchone() on conflict (nick) do update set amount = owes.amount +
excluded.amount returning amount''', [nick, fine])
if owe:
total = owe['amount'] + fine
self.cur.execute('update owes set amount = ? where nick = ?',
[total, nick])
else:
total = fine
self.cur.execute('insert into owes (nick, amount) values '
'(?, ?)', [nick, total])
self.con.commit() self.con.commit()
total = self.cur.fetchone()['amount']
reason = ('raping', 'being too lewd and getting raped')[rand] reason = ('raping', 'being too lewd and getting raped')[rand]
action = 'fines {nick} \x02${fine}\x02 for {reason}. You owe: ' \ action = 'fines {nick} \x02${fine}\x02 for {reason}. You owe: ' \
'\x0304${total}\x03'.format(nick=nick, '\x0304${total}\x03'.format(nick=nick,
@ -65,4 +55,4 @@ class Rape(DatabasePlugin):
total=total) total=total)
else: else:
action = 'rapes {nick}'.format(nick=nick) action = 'rapes {nick}'.format(nick=nick)
self.bot.action(channel, action) self.bot.action(target, action)