Re-added old karma plugin, now with more features

plugins/karma.py

@@ -0,0 +1,131 @@
+from util import hook, timesince
+
+import time
+import re
+
+db_ready = False
+
+
+def db_init(db):
+    db.execute("""CREATE TABLE if not exists karma(
+               nick_vote TEXT PRIMARY KEY,
+               up_karma INTEGER,
+               down_karma INTEGER,
+               total_karma INTEGER)""")
+
+    db.execute("""CREATE TABLE if not exists karma_voters(
+               voter TEXT,
+               votee TEXT,
+               epoch FLOAT,
+               PRIMARY KEY(voter, votee))""")
+    db_ready = True
+
+
+def up(db, nick_vote):
+    db.execute("""UPDATE karma SET
+               up_karma = up_karma + 1,
+               total_karma = total_karma + 1 WHERE nick_vote=?""", (nick_vote.lower(),))
+    db.commit()
+
+
+def down(db, nick_vote):
+    db.execute("""UPDATE karma SET
+               down_karma = down_karma+1,
+               total_karma = total_karma+1 WHERE nick_vote=?""", (nick_vote.lower(),))
+    db.commit()
+
+
+def allowed(db, nick, nick_vote):
+    time_restriction = 3600
+    db.execute("""DELETE FROM karma_voters WHERE ? - epoch >= 3600""",
+            (time.time(),))
+    db.commit()
+    check = db.execute("""SELECT epoch FROM karma_voters WHERE voter=? AND votee=?""",
+            (nick.lower(), nick_vote.lower())).fetchone()
+
+    if check:
+        check = check[0]
+        if time.time() - check >= time_restriction:
+            db.execute("""INSERT OR REPLACE INTO karma_voters(
+                       voter,
+                       votee,
+                       epoch) values(?,?,?)""", (nick.lower(), nick_vote.lower(), time.time()))
+            db.commit()
+            return True, 0
+        else:
+            return False, timesince.timeuntil(check, now=time.time()-time_restriction)
+    else:
+        db.execute("""INSERT OR REPLACE INTO karma_voters(
+                   voter,
+                   votee,
+                   epoch) values(?,?,?)""", (nick.lower(), nick_vote.lower(), time.time()))
+        db.commit()
+        return True, 0
+
+
+# TODO Make this work on multiple matches in a string, right now it'll only
+# work on one match. 
+# karma_re = ('((\S+)(\+\+|\-\-))+', re.I)
+karma_re = ('(.+)(\+\+|\-\-)$', re.I)
+
+@hook.regex(*karma_re)
+def karma_add(match, nick='', chan='', db=None, notice=None):
+
+    if not db_ready:
+        db_init(db)
+
+    nick_vote = match.group(1).strip().replace("+", "")
+    if nick.lower() == nick_vote.lower():
+        return
+    if len(nick_vote) < 3:
+        return # ignore anything below 3 chars in length
+
+    vote_allowed, when = allowed(db, nick, nick_vote)
+    print time
+    if vote_allowed:
+        if match.group(2) == '++':
+            db.execute("""INSERT or IGNORE INTO karma(
+                       nick_vote,
+                       up_karma,
+                       down_karma,
+                       total_karma) values(?,?,?,?)""", (nick_vote.lower(),0,0,0))
+            up(db, nick_vote)
+            notice("Gave {} +1 karma!".format(nick_vote))
+        if match.group(2) == '--':
+            db.execute("""INSERT or IGNORE INTO karma(
+                       nick_vote,
+                       up_karma,
+                       down_karma,
+                       total_karma) values(?,?,?,?)""", (nick_vote.lower(),0,0,0))
+            down(db, nick_vote)
+            notice("Took away 1 karma from {}.".format(nick_vote))
+        else:
+            return
+    else:
+        notice("You are trying to vote too often. You can vote again in {}!".format(when))
+
+    return
+
+
+@hook.command('k')
+@hook.command
+def karma(inp, nick='', chan='', db=None):
+    """.k/.karma <nick> -- returns karma stats for <nick>"""
+
+    if not db_ready:
+        db_init(db)
+
+    if not chan.startswith('#'):
+        return
+
+    nick_vote = inp
+    out = db.execute("""SELECT * FROM karma WHERE nick_vote=?""",
+            (nick_vote.lower(),)).fetchall()
+
+    if not out:
+        return "no karma"
+    else:
+        out = out[0]
+        return "'%s' has %s karma" % (nick_vote, out[1]-out[2])
+
+    return