nxy/bot/useless.py
2017-08-22 17:43:48 +02:00

335 lines
9.9 KiB
Python

# -*- coding: utf-8 -*-
import random
import re
import irc3
from docopt import Dict
from irc3.plugins.command import command
from irc3.utils import IrcString
from psycopg2 import Error
from . import DatabasePlugin
from .utils import re_generator
class Useless(DatabasePlugin):
RAINBOW = (5, 7, 8, 9, 3, 10, 12, 2, 6, 13)
WOAH = (
'woah',
'woah indeed',
'woah woah woah!',
'keep your woahs to yourself',
)
ISPS = (
't-ipconnect',
'telefonica',
'vodafone',
'kabel',
'unity-media',
)
GENDERS = (
'male',
'female',
'shemale',
'hermaphrodite',
)
@command(permission='admin', show_in_help=False)
def kim(self, mask: IrcString, target: IrcString, args: Dict):
"""Kicks Kim, most useful command.
%%kim
"""
self.bot.kick(target, 'Kim')
@irc3.event(r'(?i)^:(?P<mask>\S+@\S+({})\S+) JOIN :(?P<target>#\S+)$'.format('|'.join(ISPS)))
def bounce(self, mask, target):
nick = IrcString(mask).nick
if target == '#w0bm' and nick != self.bot.nick:
self.bot.privmsg(target, '{}: Du musst bouncen!'.format(nick))
@irc3.event(r'(?i)^:\S+ PRIVMSG (?P<target>\S+) :.*(woah|whoa).*$')
def woah(self, target: str):
"""Colorize words in a sentence with rainbow colors."""
if random.randint(0, 4) is 0:
self.bot.privmsg(target, random.choice(self.WOAH))
@irc3.event(r'(?i)^:\S+ PRIVMSG (?P<target>\S+) :(?P<msg>huehuehue)$')
def huehuehue(self, target: str, msg: str):
"""Returns a huehuehue when someone writes it."""
self.bot.privmsg(target, msg)
@irc3.event(r'(?i)^:\S+ PRIVMSG (?P<target>\S+) :re+$')
def reeeeee(self, target: str):
"""Returns a REEEE."""
self.bot.privmsg(target, re_generator())
@irc3.event(r'(?i)^:\S+ PRIVMSG (?P<target>\S+) :same$')
def same(self, target: str):
"""Returns a plain same when a user writes same."""
self.bot.privmsg(target, 'same')
@irc3.event(r'(?i)^:\S+ PRIVMSG (?P<target>\S+) :\[(?P<msg>.*)\]$')
def intensifies(self, target: str, msg: str):
"""String with brackets around will be returned with INTENSIFIES."""
self.bot.privmsg(target, '\x02[{} INTENSIFIES]'.format(msg.upper()))
@command
def kill(self, mask: IrcString, target: IrcString, args: Dict):
"""Kills a user with a random message
%%kill [<nick>]
"""
self.cur.execute('''
SELECT
item
FROM
kills
ORDER BY
random()
LIMIT
1
''')
self.bot.action(target, self.cur.fetchone()['item'].format(
nick=args.get('<nick>', mask.nick),
))
@command
def yiff(self, mask: IrcString, target: IrcString, args: Dict):
"""Yiffs a user with a random message
%%yiff [<nick>]
"""
self.cur.execute('''
SELECT
item
FROM
yiffs
ORDER BY
random()
LIMIT
1
''')
self.bot.action(target, self.cur.fetchone()['item'].format(
nick=args.get('<nick>', mask.nick),
yiffer=mask.nick,
))
@command
def waifu(self, mask: IrcString, target: IrcString, args: Dict):
"""Get waifu for a user (set with prefixing the user with =)
%%waifu [<nick>...]
"""
return self.husbando_waifu('waifu', mask, target, args)
@command
def husbando(self, mask: IrcString, target: IrcString, args: Dict):
"""Get husbando for a user (set with prefixing the user with =)
%%husbando [<nick>...]
"""
return self.husbando_waifu('husbando', mask, target, args)
@command
def storyofpomfface(self, mask: IrcString, target: IrcString, args: Dict):
"""Story of pomf face
%%storyofpomfface
"""
for face in (':O C==3', ':OC==3', ':C==3', ':C=3', ':C3', ':3'):
self.bot.privmsg(target, face)
@command
def choose(self, mask: IrcString, target: IrcString, args: Dict):
"""Decides between items (separated by comma)
%%choose <items>...
"""
choice = random.choice(' '.join(args['<items>']).split(','))
return '{}: {}'.format(mask.nick, choice.strip())
@command
def jn(self, mask: IrcString, target: IrcString, args: Dict):
"""Decides between yes and no (for a question).
%%jn <question>...
%%jn
"""
choice = random.choice(['3Ja', '4Nein'])
return '{}: \x02\x030{}'.format(mask.nick, choice)
@command
def kiss(self, mask: IrcString, target: IrcString, args: Dict):
"""Kisses a user
%%kiss [<nick>]
"""
return '(づ。◕‿‿◕。)\x0304。。・゜゜・。。・゜❤\x03 {} \x0304❤'.format(args.get('<nick>', mask.nick))
@command
def hug(self, mask: IrcString, target: IrcString, args: Dict):
"""Hugs a user
%%hug [<nick>]
"""
return '\x0304♥♡❤♡♥\x03 {} \x0304♥♡❤♡♥'.format(args.get('<nick>', mask.nick))
@command
def bier(self, mask: IrcString, target: IrcString, args: Dict):
"""Gives a user a beer
%%bier [<nick>]
"""
nick = args.get('<nick>', mask.nick)
self.bot.action(target, 'schenkt ein kühles Blondes an {} aus.'.format(nick))
@command
def fucken(self, mask: IrcString, target: IrcString, args: Dict):
"""Kills and fucks a user
%%fucken [<nick>]
"""
nick = args.get('<nick>', mask.nick)
self.bot.action(target, 'fuckt {0} und tötet {0} anschließend.'.format(nick, nick))
@command(aliases=['anhero', 'sudoku'])
def seppuku(self, mask: IrcString, target: IrcString, args: Dict):
"""Kicks a user
%%seppuku
"""
self.bot.kick(target, mask.nick, 'Sayonara bonzai-chan...')
@command
def hack(self, mask: IrcString, target: IrcString, args: Dict):
"""Hacks (a user)
%%hack [<nick>]
"""
nick = args.get('<nick>')
return 'hacking{}...'.format(' %s' % nick if nick else '')
@command
def gay(self, mask: IrcString, target: IrcString, args: Dict):
"""Make someone gay (alias to rainbow)
%%gay <word>...
"""
return self.rainbow(mask, target, args)
@command
def rainbow(self, mask: IrcString, target: IrcString, args: Dict):
"""Colorize a word in rainbow colors
%%rainbow <word>...
"""
last = 0
word = []
for char in ' '.join(args.get('<word>')):
if char != ' ':
char = self._rainbow(last, char)
last += 1
word.append(char)
return ''.join(word)
@command
def wrainbow(self, mask: IrcString, target: IrcString, args: Dict):
"""Colorize words in a sentence with rainbow colors
%%wrainbow <words>...
"""
return ' '.join([self._rainbow(i, word) for i, word in enumerate(args['<words>'])])
@command(aliases=['hw'])
def halfwidth(self, mask: IrcString, target: IrcString, args: Dict):
"""Turns UPPERCASE words into halfwidth characters
%%halfwidth <words>...
"""
def replace(match):
return ''.join(chr(0xFF20 + (ord(c) - 64)) for c in match.group(0))
return '\x02[HALFWIDTH]\x02 {}'.format(re.sub(r'(?:\b)[A-Z]+(?:\b)', replace, ' '.join(args['<words>'])))
@command
def asshole(self, mask: IrcString, target: IrcString, args: Dict):
"""Checks how much of an asshole you are
%%asshole [<nick>]
"""
nick = args.get('<nick>', mask.nick)
if nick.lower() == 'mrhanky':
perc_min = 100
elif nick.lower() == 'flummi':
perc_min = 90
else:
perc_min = 0
asshole_perc = random.randint(perc_min, 100)
return 'Asshole scan... {} is {}% of an asshole.'.format(nick, asshole_perc)
@command
def assume(self, mask: IrcString, target: IrcString, args: Dict):
"""Assumes the gender of a nick or yourself
%%assume [<nick>]
"""
nick = args.get('<nick>', mask.nick)
gender = random.choice(self.GENDERS)
return 'Assuming {}\'s gender... they\'re a {}.'.format(nick, gender)
@command
def spit(self, mask: IrcString, target: IrcString, args: Dict):
"""Spits a user
%%spit [<nick>]
"""
nick = args.get('<nick>', mask.nick)
self.bot.action(target, 'spits on {} like a dirty whore.'.format(nick))
def husbando_waifu(self, field: str, mask: IrcString, target: IrcString, args: Dict):
nick = args.get('<nick>', mask.nick)
if isinstance(nick, list):
nick = ' '.join(nick)
if nick.startswith('='):
nick = nick[1:]
try:
self.cur.execute('''
INSERT INTO
users (nick, {0})
VALUES
(lower(%s), %s)
ON CONFLICT (nick) DO UPDATE SET
{0} = excluded.{0}
'''.format(field), [mask.nick, nick])
self.con.commit()
self.bot.notice(mask.nick, '{} set to: {}'.format(field.title(), nick))
except Error as ex:
self.log.error(ex)
self.con.rollback()
else:
self.cur.execute('''
SELECT
{}
FROM
users
WHERE
lower(nick) = lower(%s)
'''.format(field), [nick])
result = self.cur.fetchone()
if result and result[field]:
return '\x02[{}]\x02 {}: {}'.format(field.title(), nick, result[field])
# noinspection PyMethodMayBeStatic
def _rainbow(self, i, char):
return '\x03{0:02d}{1}'.format(self.RAINBOW[i % len(self.RAINBOW)], char)