Added DatabasePlugin._insert

nxy/plugins/__init__.py

@@ -19,7 +19,18 @@ class Plugin(BasePlugin):
 
 
 class DatabasePlugin(Plugin):
+    table = None
+
     def __init__(self, bot):
         super().__init__(bot)
         self.con = bot.con.db
         self.cur = self.con.cursor()
+
+    def _insert(self, **kwargs):
+        qry = 'insert into {table} ({rows}) values ({placeholders})'.format(
+            table=self.table,
+            rows=', '.join(kwargs),
+            placeholders=', '.join('?' * len(kwargs))
+        )
+        self.cur.execute(qry, kwargs.values())
+        self.cur.commit()

nxy/plugins/database.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import sqlite3
+
 import irc3
 
 from . import Plugin

nxy/plugins/linux.py

@@ -18,8 +18,8 @@ GNU_LINUX = """I'd Just Like To Interject For A Moment. What you're referring
  OS as defined by POSIX."""
 
 
-class Kernel(Plugin):
-    URL = 'https://www.kernel.org/feeds/kdist.xml'
+class Linux(Plugin):
+    KERNEL_FEED = 'https://www.kernel.org/feeds/kdist.xml'
 
     @irc3.event(r'(?i)^:\S+ PRIVMSG (?P<channel>\S+) :'
                 r'.*(?<!gnu[/+])linux(?! kernel).*')
@@ -33,11 +33,14 @@ class Kernel(Plugin):
 
         %%kernel
         """
-        feed = feedparser.parse(self.URL)
+        feed = feedparser.parse(self.KERNEL_FEED)
         releases = []
         for e in feed['entries']:
             version, branch = e['title'].split(': ')
             if '(EOL)' in e['description']:
-                branch = '%s, \x1DEOL\x0F' % branch
-            releases.append('\x02%s\x0F (%s)' % (version, branch))
+                branch = '{branch}, \x1DEOL\x0F'.format(branch=branch)
+            releases.append('\x02{version}\x0F ({branch})'.format(
+                version=version,
+                branch=branch,
+            ))
         return '[Kernel] {releases}'.format(releases=', '.join(releases))

nxy/plugins/mcmaniac.py

@@ -1,8 +1,9 @@
 # -*- coding: utf-8 -*-
+import irc3
+
 from docopt import Dict as DocOptDict
 from irc3.plugins.command import command
 from irc3.utils import IrcString
-import irc3
 
 from . import DatabasePlugin
 from ..utils import parse_int
@@ -11,6 +12,7 @@ from ..utils import parse_int
 class McManiac(DatabasePlugin):
     requires = ['irc3.plugins.command',
                 'nxy.plugins.database']
+    table = 'mcmaniacs'
 
     @command(options_first=True)
     def mcmaniac(self, mask: IrcString, channel: IrcString, args: DocOptDict):
@@ -40,9 +42,9 @@ class McManiac(DatabasePlugin):
         if result:
             return '[{idx}/{len}] {item}'.format(**result)
 
-    @irc3.event(r'(?i)^:(?P<mask>\S+) PRIVMSG \S+ :(?P<msg>.*(?P<item>Mc\S+iaC).*)')
+    @irc3.event(r'(?i)^:(?P<mask>\S+) PRIVMSG \S+ :'
+                r'(?P<msg>.*(?P<item>Mc\S+iaC).*)')
     def save(self, mask: str, msg: str, item: str):
         nick = IrcString(mask).nick
         if nick != self.bot.nick and msg != '.reload mcmaniac':
-            self.cur.execute('insert into mcmaniacs (item) values (?)', [item])
-            self.con.commit()
+            self._insert(item=item)

nxy/plugins/quotes.py

@@ -1,21 +1,19 @@
 # -*- coding: utf-8 -*-
 import irc3
-import re
 
 from docopt import Dict as DocOptDict
 from irc3.plugins.command import command
 from irc3.utils import IrcString
 
 from . import DatabasePlugin
-from ..utils import parse_int
+from ..utils import NICK_REGEX, parse_int
 
 
 @irc3.plugin
 class Quotes(DatabasePlugin):
     requires = ['irc3.plugins.command',
                 'nxy.plugins.database']
-
-    NICK_REGEX = re.compile(r'<?[~&@%+]?([a-zA-Z0-9_\-^`|\\\[\]{}]+)>?')
+    table = 'quotes'
 
     @command(options_first=True)
     def q(self, mask: IrcString, channel: IrcString, args: DocOptDict):
@@ -30,9 +28,10 @@ class Quotes(DatabasePlugin):
         if cmd and item:
             if self.guard.has_permission(mask, 'admin'):
                 if cmd == 'add':
-                    nick = self.NICK_REGEX.match(nick).group(1)
-                    self.cur.execute('insert into quotes (nick, item) '
-                                     'values (?, ?)', [nick, ' '.join(item)])
+                    nick = NICK_REGEX.match(nick).group(1)
+                    if not nick:
+                        return '[Quotes] Error parsing nick'
+                    self._insert(nick=nick, item=' '.join(item))
                 if cmd == 'del':
                     index, order, op = parse_int(''.join(item), select=False)
                     if not index:

nxy/plugins/seen.py

@@ -13,6 +13,7 @@ from . import DatabasePlugin
 class Seen(DatabasePlugin):
     requires = ['irc3.plugins.command',
                 'nxy.plugins.database']
+    table = 'seens'
 
     @command(options_first=True)
     def seen(self, mask: IrcString, channel: IrcString, args: DocOptDict):
@@ -38,7 +39,5 @@ class Seen(DatabasePlugin):
     def save(self, mask: str, channel: str, msg: str):
         nick = IrcString(mask).nick
         if nick != self.bot.nick:
-            self.cur.execute('insert or replace into seens (nick, channel, '
-                             'message, last_seen) values (?, ?, ?, ?)',
-                             [nick, channel.lower(), msg, datetime.now()])
-            self.con.commit()
+            self._insert(nick=nick, channel=channel.lower(), message=msg,
+                         last_seen=datetime.now())

nxy/plugins/tell.py

@@ -11,6 +11,7 @@ from . import DatabasePlugin
 class Tell(DatabasePlugin):
     requires = ['irc3.plugins.command',
                 'nxy.plugins.database']
+    table = 'tells'
 
     def __init__(self, bot: irc3.IrcBot):
         super().__init__(bot)
@@ -34,9 +35,7 @@ class Tell(DatabasePlugin):
             self.tell_queue[nick] = []
         tell = [mask.nick.lower(), nick, ' '.join(args['<message>']).strip()]
         self.tell_queue[nick].append(tell)
-        self.cur.execute('insert into tells (from_user, to_user, message)'
-                         'values (?, ?, ?)', tell)
-        self.con.commit()
+        self._insert(from_user=tell[0], to_user=tell[1], message=tell[2])
 
     @irc3.event(r'(?i)^:(?P<mask>.*) PRIVMSG .* :.*')
     def check(self, mask: str):

nxy/plugins/timer.py

@@ -15,6 +15,7 @@ from ..utils import time_delta
 class Timer(DatabasePlugin):
     requires = ['irc3.plugins.command',
                 'nxy.plugins.database']
+    table = 'timers'
 
     def __init__(self, bot: irc3.IrcBot):
         super().__init__(bot)
@@ -36,12 +37,9 @@ class Timer(DatabasePlugin):
         delay = args['<delay>']
         delta = time_delta(delay)
         if delta:
-            date = datetime.utcnow() + delta
             message = ' '.join(args['<message>'])
-            self.cur.execute('''insert into timers (mask, channel, message,
-            delay, until) values (?, ?, ?, ?, ?)''', [mask.nick, channel,
-                                                      message, delay, date])
-            self.con.commit()
+            self._insert(mask=mask.nick, channel=channel, message=message,
+                         delay=delay, until=datetime.utcnow() + delta)
             asyncio.ensure_future(self._timer(mask, channel, delta, message,
                                               delay, self.cur.lastrowid))
             self.bot.notice(mask.nick, 'Timer in {delay} set: {message}'

nxy/utils/__init__.py

@@ -2,7 +2,7 @@
 from pprint import pprint as pp
 
 from .date import date_from_iso, time_delta, time_since
-from .misc import parse_int
+from .misc import NICK_REGEX, parse_int
 
 __all__ = (
     # pprint alias
@@ -12,5 +12,6 @@ __all__ = (
     'time_delta',
     'time_since',
     # misc
-    'parse_int'
+    'NICK_REGEX',
+    'parse_int',
 )

nxy/utils/misc.py

@@ -1,4 +1,7 @@
 # -*- coding: utf-8 -*-
+import re
+
+NICK_REGEX = re.compile(r'<?[~&@%+]?([a-zA-Z0-9_\-^`|\\\[\]{}]+)>?')
 
 
 def parse_int(val: str, select: bool = True) -> tuple: