nxy/bot/plugins/tell.py

83 lines
2.3 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
2017-06-30 17:33:26 +00:00
from datetime import datetime
2017-06-30 12:04:50 +00:00
import irc3
from docopt import Dict as DocOptDict
from irc3.plugins.command import command
from irc3.utils import IrcString
from psycopg2 import Error
from . import DatabasePlugin
@irc3.plugin
class Tell(DatabasePlugin):
requires = ['irc3.plugins.command',
2017-07-07 00:11:20 +00:00
'bot.plugins.storage']
2017-05-29 12:40:58 +00:00
def __init__(self, bot: irc3.IrcBot):
super().__init__(bot)
self.tell_queue = {}
2017-06-30 12:04:50 +00:00
self.cur.execute('''
2017-08-22 09:58:32 +00:00
SELECT
2017-06-30 17:33:26 +00:00
to_nick, from_nick, message, created_at
2017-08-22 09:58:32 +00:00
FROM
2017-06-30 12:04:50 +00:00
tells
''')
for res in self.cur.fetchall():
2017-06-30 17:33:26 +00:00
nick = res['to_nick'].lower()
2017-06-30 12:04:50 +00:00
if nick not in self.tell_queue:
self.tell_queue[nick] = []
2017-06-30 17:33:26 +00:00
self.tell_queue[nick].append(res[1:])
@command
2017-06-30 12:04:50 +00:00
def tell(self, mask: IrcString, target: IrcString, args: DocOptDict):
"""Saves a message for nick to forward on activity
%%tell <nick> <message>...
"""
2017-08-22 09:58:32 +00:00
nick = args['<nick>'].lower()
tell = [nick, mask.nick, ' '.join(args['<message>']).strip(), datetime.utcnow()]
2017-06-30 12:04:50 +00:00
2017-08-22 09:58:32 +00:00
if nick not in self.tell_queue:
self.tell_queue[nick] = []
self.tell_queue[nick].append(tell[1:])
2017-06-30 12:04:50 +00:00
try:
self.cur.execute('''
2017-08-22 09:58:32 +00:00
INSERT INTO
2017-06-30 17:33:26 +00:00
tells (to_nick, from_nick, message, created_at)
2017-08-22 09:58:32 +00:00
VALUES
2017-06-30 17:33:26 +00:00
(%s, %s, %s, %s)
''', tell)
self.con.commit()
2017-08-22 09:58:32 +00:00
except Error as ex:
self.log.error(ex)
self.con.rollback()
2017-05-29 12:40:58 +00:00
@irc3.event(r'(?i)^:(?P<mask>.*) PRIVMSG .* :.*')
2017-08-22 09:58:32 +00:00
def on_message(self, mask: str):
2017-06-30 12:04:50 +00:00
"""If activ nick has tells, forward and delete them."""
2017-08-22 09:58:32 +00:00
nick = IrcString(mask).nick.lower()
2017-06-30 12:04:50 +00:00
2017-08-22 09:58:32 +00:00
if nick in self.tell_queue:
2017-06-30 12:04:50 +00:00
# Forward all tells for nick
2017-08-22 09:58:32 +00:00
for tell in self.tell_queue[nick]:
self.bot.privmsg(nick, '[Tell] Message from {nick} at {time}: {message}'.format(
nick=tell[0],
time=tell[1], # TODO: format time
message=tell[2],
))
del self.tell_queue[nick]
self.cur.execute('''
DELETE FROM
tells
WHERE
to_nick = %s
''', [nick])
self.con.commit()