Add per-channel regex filtering.
This commit is contained in:
parent
6169be6da1
commit
6086a7c6da
|
@ -0,0 +1,128 @@
|
|||
from util import hook
|
||||
|
||||
|
||||
# Default value.
|
||||
# If True, all channels without a setting will have regex enabled
|
||||
# If False, all channels without a setting will have regex disabled
|
||||
default_enabled = True
|
||||
|
||||
db_already_initiated = False
|
||||
|
||||
|
||||
def db_init(db):
|
||||
global db_already_initiated
|
||||
if not db_already_initiated:
|
||||
db_already_initiated = True
|
||||
db.execute("CREATE TABLE IF NOT EXISTS regexchans(channel PRIMARY KEY, status)")
|
||||
db.commit()
|
||||
|
||||
|
||||
def get_status(db, channel):
|
||||
row = db.execute("SELECT status FROM regexchans WHERE channel = ?", [channel]).fetchone()
|
||||
if row:
|
||||
return row[0]
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
def set_status(db, channel, status):
|
||||
row = db.execute("REPLACE INTO regexchans (channel, status) VALUES(?, ?)", [channel, status])
|
||||
db.commit()
|
||||
|
||||
|
||||
def delete_status(db, channel):
|
||||
row = db.execute("DELETE FROM regexchans WHERE channel = ?", [channel])
|
||||
db.commit()
|
||||
|
||||
|
||||
def list_status(db):
|
||||
row = db.execute("SELECT * FROM regexchans").fetchall()
|
||||
result = None
|
||||
for values in row:
|
||||
if result:
|
||||
result += u", {}: {}".format(values[0], values[1])
|
||||
else:
|
||||
result = u"{}: {}".format(values[0], values[1])
|
||||
return result
|
||||
|
||||
|
||||
@hook.sieve
|
||||
def sieve_regex(bot, inp, func, kind, args):
|
||||
db = bot.get_db_connection(inp.conn)
|
||||
db_init(db)
|
||||
if kind == 'regex' and inp.chan.startswith("#") and func.__name__ != 'factoid':
|
||||
chanstatus = get_status(db, inp.chan)
|
||||
if chanstatus != "ENABLED" and (chanstatus == "DISABLED" or not default_enabled):
|
||||
print u"Denying input.raw={}, kind={}, args={} from {}".format(inp.raw, kind, args, inp.chan)
|
||||
return None
|
||||
print u"Allowing input.raw={}, kind={}, args={} from {}".format(inp.raw, kind, args, inp.chan)
|
||||
|
||||
return inp
|
||||
|
||||
|
||||
@hook.command(permissions=["botcontrol"])
|
||||
def enableregex(inp, db=None, message=None, notice=None, chan=None, nick=None):
|
||||
db_init(db)
|
||||
inp = inp.strip().lower()
|
||||
if not inp:
|
||||
channel = chan
|
||||
elif inp.startswith("#"):
|
||||
channel = inp
|
||||
else:
|
||||
channel = u"#{}".format(inp)
|
||||
|
||||
message(u"Enabling regex matching (youtube, etc) (issued by {})".format(nick), target=channel)
|
||||
notice(u"Enabling regex matching (youtube, etc) in channel {}".format(channel))
|
||||
set_status(db, channel, "ENABLED")
|
||||
|
||||
|
||||
@hook.command(permissions=["botcontrol"])
|
||||
def disableregex(inp, db=None, message=None, notice=None, chan=None, nick=None):
|
||||
db_init(db)
|
||||
inp = inp.strip().lower()
|
||||
if not inp:
|
||||
channel = chan
|
||||
elif inp.startswith("#"):
|
||||
channel = inp
|
||||
else:
|
||||
channel = u"#{}".format(inp)
|
||||
|
||||
message(u"Disabling regex matching (youtube, etc) (issued by {})".format(nick), target=channel)
|
||||
notice(u"Disabling regex matching (youtube, etc) in channel {}".format(channel))
|
||||
set_status(db, channel, "DISABLED")
|
||||
|
||||
|
||||
@hook.command(permissions=["botcontrol"])
|
||||
def resetregex(inp, db=None, message=None, notice=None, chan=None, nick=None):
|
||||
db_init(db)
|
||||
inp = inp.strip().lower()
|
||||
if not inp:
|
||||
channel = chan
|
||||
elif inp.startswith("#"):
|
||||
channel = inp
|
||||
else:
|
||||
channel = u"#{}".format(inp)
|
||||
|
||||
message(u"Resetting regex matching setting (youtube, etc) (issued by {})".format(nick), target=channel)
|
||||
notice(u"Resetting regex matching setting (youtube, etc) in channel {}".format(channel))
|
||||
delete_status(db, channel)
|
||||
|
||||
|
||||
@hook.command(permissions=["botcontrol"])
|
||||
def regexstatus(inp, db=None, chan=None):
|
||||
db_init(db)
|
||||
inp = inp.strip().lower()
|
||||
if not inp:
|
||||
channel = chan
|
||||
elif inp.startswith("#"):
|
||||
channel = inp
|
||||
else:
|
||||
channel = u"#{}".format(inp)
|
||||
|
||||
return u"Regex status for {}: {}".format(channel, get_status(db, channel))
|
||||
|
||||
|
||||
@hook.command(permissions=["botcontrol"])
|
||||
def listregex(inp, db=None):
|
||||
db_init(db)
|
||||
return list_status(db)
|
Reference in New Issue