def warn_list(var, wrapper, args): if args.help: wrapper.reply(messages["warn_list_syntax"]) return acc = wrapper.source.account if not acc: return warnings = db.list_warnings(acc, expired=args.all, skip=(args.page - 1) * 10, show=11) points = db.get_warning_points(acc) wrapper.pm(messages["warn_list_header"].format(points)) for i, warning in enumerate(warnings): if i == 10: parts = [] if args.all: parts.append(_wall[0]) parts.append(str(args.page + 1)) wrapper.pm(messages["warn_list_footer"].format("warn", parts)) break wrapper.pm(messages["warn_list"].format(**warning)) if not warnings: wrapper.pm(messages["fwarn_list_empty"])
def add_warning(target: Union[str, users.User], amount: int, actor: users.User, reason: str, notes: str = None, expires=None, sanctions=None): if isinstance(target, users.User): tacc = target.account if tacc is None: return False else: tacc = target reason = reason.format() sacc = actor.account # Turn expires into a datetime if we were passed a string; note that no error checking is performed here if not isinstance(expires, datetime): expires = _parse_expires(expires) # determine if we need to automatically add any sanctions if sanctions is None: sanctions = {} prev = db.get_warning_points(tacc) cur = prev + amount if amount > 0: _get_auto_sanctions(sanctions, prev, cur) sid = db.add_warning(tacc, sacc, amount, reason, notes, expires) if "stasis" in sanctions: db.add_warning_sanction(sid, "stasis", sanctions["stasis"]) if "deny" in sanctions: for cmd in sanctions["deny"]: db.add_warning_sanction(sid, "deny command", cmd) if "tempban" in sanctions: # this inserts into the bantrack table too (acclist, hmlist) = db.add_warning_sanction(sid, "tempban", sanctions["tempban"]) cmodes = [] for acc in acclist: cmodes.append(("+b", "{0}{1}".format(var.ACCOUNT_PREFIX, acc))) channels.Main.mode(*cmodes) for user in channels.Main.users: if user.account in acclist: channels.Main.kick( user, messages["tempban_kick"].format(nick=user, botnick=users.Bot.nick, reason=reason)) # Update any tracking vars that may have changed due to this db.init_vars() return sid
def fwarn_list(var, wrapper, args): if args.help: wrapper.reply(messages["fwarn_list_syntax"]) return if args.account or args.nick == "*": acc = args.nick else: m = users.complete_match(args.nick) if m: acc = m.get().account else: acc = args.nick if not acc: wrapper.reply(messages["fwarn_nick_invalid"].format(args.nick)) return if acc == "*": warnings = db.list_all_warnings(list_all=args.all, skip=(args.page - 1) * 10, show=11) else: warnings = db.list_warnings(acc, expired=args.all, deleted=args.all, skip=(args.page - 1) * 10, show=11) points = db.get_warning_points(acc) wrapper.pm(messages["fwarn_list_header"].format(acc, points)) for i, warning in enumerate(warnings): if i == 10: parts = [] if args.all: parts.append(_wall[0]) parts.append(acc) parts.append(str(args.page + 1)) wrapper.pm(messages["warn_list_footer"].format("fwarn", parts)) break wrapper.pm(messages["fwarn_list"].format(**warning)) if not warnings: wrapper.pm(messages["fwarn_list_empty"])
def fwarn(var, wrapper, message): """Issues a warning to someone or views warnings.""" # !fwarn list [-all] [nick] [page] # -all => Shows all warnings, if omitted only shows active (non-expired and non-deleted) ones. # nick => nick to view warnings for. Can also be a hostmask in nick!user@host form. If nick # is not online, interpreted as an account name. To specify an account if nick is online, # use =account. If not specified, shows all warnings on the bot. # !fwarn view <id> - views details on warning id # !fwarn del <id> - deletes warning id # !fwarn set <id> [~expiry] [reason] [| notes] # !fwarn add <nick> <points> [~expiry] [sanctions] [:]<reason> [| notes] # e.g. !fwarn add lykos 1 ~30d deny=goat,gstats stasis=5 Spamming | I secretly just hate him # nick => nick to warn. Can also be a hostmask in nick!user@host form. If nick is not online, # interpreted as an account name. To specify an account if nick is online, use =account. # points => Warning points, must be above 0 # ~expiry => Expiration time, must be suffixed with d (days), h (hours), or m (minutes) # sanctions => list of sanctions. Valid sanctions are: # deny: denies access to the listed commands # stasis: gives the user stasis # reason => Reason, required. If the first word of the reason is also a sanction, prefix it with : # |notes => Secret notes, not shown to the user (only shown if viewing the warning in PM) # If specified, must be prefixed with |. This means | is not a valid character for use # in reasons (no escaping is performed). params = message.split() target = None points = None expires = None sanctions = {} reason = None notes = None try: command = params.pop(0) except IndexError: wrapper.reply(messages["fwarn_usage"]) return if command not in ("list", "view", "add", "del", "set", "help"): # if what follows is a number, assume we're viewing or setting a warning # (depending on number of params) # if it's another string, assume we're adding or listing, again depending # on number of params params.insert(0, command) try: num = int(command) if len(params) == 1: command = "view" else: command = "set" except ValueError: if len(params) < 3 or params[1] == "-all": command = "list" if len(params) > 1 and params[1] == "-all": # fwarn list expects these two params in a different order params.pop(1) params.insert(0, "-all") else: command = "add" if command == "help": try: subcommand = params.pop(0) except IndexError: wrapper.reply(messages["fwarn_usage"]) return if subcommand not in ("list", "view", "add", "del", "set", "help"): wrapper.reply(messages["fwarn_usage"]) return wrapper.reply(messages["fwarn_{0}_syntax".format(subcommand)]) return if command == "list": list_all = False page = 1 try: list_all = params.pop(0) target = params.pop(0) page = int(params.pop(0)) except IndexError: pass except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return try: if list_all and list_all != "-all": if target is not None: page = int(target) target = list_all list_all = False elif list_all == "-all": list_all = True except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return try: page = int(target) target = None except (TypeError, ValueError): pass if target is not None: acc, hm = parse_warning_target(target) if acc is None and hm is None: wrapper.reply(messages["fwarn_nick_invalid"]) return warnings = db.list_warnings(acc, hm, expired=list_all, deleted=list_all, skip=(page - 1) * 10, show=11) points = db.get_warning_points(acc, hm) wrapper.pm(messages["fwarn_list_header"].format( target, points, "" if points == 1 else "s")) else: warnings = db.list_all_warnings(list_all=list_all, skip=(page - 1) * 10, show=11) i = 0 for warn in warnings: i += 1 if (i == 11): parts = [] if list_all: parts.append("-all") if target is not None: parts.append(target) parts.append(str(page + 1)) wrapper.pm(messages["fwarn_list_footer"].format( " ".join(parts))) break start = "" end = "" ack = "" if warn["expires"] is not None: if warn["expired"]: expires = messages["fwarn_list_expired"].format( warn["expires"]) else: expires = messages["fwarn_view_expires"].format( warn["expires"]) else: expires = messages["fwarn_never_expires"] if warn["deleted"]: start = "\u000314" end = " [\u00034{0}\u000314]\u0003".format( messages["fwarn_deleted"]) elif warn["expired"]: start = "\u000314" end = " [\u00037{0}\u000314]\u0003".format( messages["fwarn_expired"]) if not warn["ack"]: ack = "\u0002!\u0002 " wrapper.pm(messages["fwarn_list"].format( start, ack, warn["id"], warn["issued"], warn["target"], warn["sender"], warn["reason"], warn["amount"], "" if warn["amount"] == 1 else "s", expires, end)) if i == 0: wrapper.pm(messages["fwarn_list_empty"]) return if command == "view": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["fwarn_view_syntax"]) return warning = db.get_warning(warn_id) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return if warning["deleted"]: expires = messages["fwarn_view_deleted"].format( warning["deleted_on"], warning["deleted_by"]) elif warning["expired"]: expires = messages["fwarn_view_expired"].format(warning["expires"]) elif warning["expires"] is None: expires = messages["fwarn_view_active"].format( messages["fwarn_never_expires"]) else: expires = messages["fwarn_view_active"].format( messages["fwarn_view_expires"].format(warning["expires"])) wrapper.pm(messages["fwarn_view_header"].format( warning["id"], warning["target"], warning["issued"], warning["sender"], warning["amount"], "" if warning["amount"] == 1 else "s", expires)) reason = [warning["reason"]] if warning["notes"] is not None: reason.append(warning["notes"]) wrapper.pm(" | ".join(reason)) sanctions = [] if not warning["ack"]: sanctions.append(messages["fwarn_view_ack"]) if warning["sanctions"]: sanctions.append(messages["fwarn_view_sanctions"]) if "stasis" in warning["sanctions"]: if warning["sanctions"]["stasis"] != 1: sanctions.append( messages["fwarn_view_stasis_plural"].format( warning["sanctions"]["stasis"])) else: sanctions.append(messages["fwarn_view_stasis_sing"]) if "deny" in warning["sanctions"]: sanctions.append(messages["fwarn_view_deny"].format(", ".join( warning["sanctions"]["deny"]))) if "tempban" in warning["sanctions"]: sanctions.append(messages["fwarn_view_tempban"].format( warning["sanctions"]["tempban"])) if sanctions: wrapper.pm(" ".join(sanctions)) return if command == "del": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["fwarn_del_syntax"]) return warning = db.get_warning(warn_id) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return acc, hm = parse_warning_target(wrapper.source.nick) db.del_warning(warn_id, acc, hm) wrapper.reply(messages["fwarn_done"]) if var.LOG_CHANNEL: msg = messages["fwarn_log_del"].format( warn_id, warning["target"], hm, warning["reason"], (" | " + warning["notes"]) if warning["notes"] else "") channels.get(var.LOG_CHANNEL).send(msg, prefix=var.LOG_PREFIX) return if command == "set": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["fwarn_set_syntax"]) return warning = db.get_warning(warn_id) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return rsp = " ".join(params).split("|", 1) if len(rsp) == 1: rsp.append(None) reason, notes = rsp reason = reason.strip() # check for modified expiry expires = warning["expires"] rsp = reason.split(" ", 1) if rsp[0] and rsp[0][0] == "~": if len(rsp) == 1: rsp.append("") expires, reason = rsp expires = expires[1:] reason = reason.strip() if expires in messages["never_aliases"]: expires = None else: suffix = expires[-1] try: amount = int(expires[:-1]) except ValueError: wrapper.reply(messages["fwarn_expiry_invalid"]) return if amount <= 0: wrapper.reply(messages["fwarn_expiry_invalid"]) return issued = datetime.strptime(warning["issued"], "%Y-%m-%d %H:%M:%S") if suffix == "d": expires = issued + timedelta(days=amount) elif suffix == "h": expires = issued + timedelta(hours=amount) elif suffix == "m": expires = issued + timedelta(minutes=amount) else: wrapper.reply(messages["fwarn_expiry_invalid"]) return round_add = 0 if expires.second >= 30: round_add = 1 expires -= timedelta(seconds=expires.second, microseconds=expires.microsecond) expires += timedelta(minutes=round_add) # maintain existing reason if none was specified if not reason: reason = warning["reason"] # maintain existing notes if none were specified if notes is not None: notes = notes.strip() if not notes: notes = None else: notes = warning["notes"] db.set_warning(warn_id, expires, reason, notes) wrapper.reply(messages["fwarn_done"]) if var.LOG_CHANNEL: changes = [] if expires != warning["expires"]: oldexpiry = warning["expires"] if warning[ "expires"] else messages["fwarn_log_set_noexpiry"] newexpiry = expires if expires else messages[ "fwarn_log_set_noexpiry"] changes.append(messages["fwarn_log_set_expiry"].format( oldexpiry, newexpiry)) if reason != warning["reason"]: changes.append(messages["fwarn_log_set_reason"].format( warning["reason"], reason)) if notes != warning["notes"]: if warning["notes"]: changes.append(messages["fwarn_log_set_notes"].format( warning["notes"], notes)) else: changes.append( messages["fwarn_log_set_notes_new"].format(notes)) if changes: log_msg = messages["fwarn_log_set"].format( warn_id, warning["target"], wrapper.source.nick, " | ".join(changes)) channels.get(var.LOG_CHANNEL).send(log_msg, prefix=var.LOG_PREFIX) return # command == "add" while params: p = params.pop(0) if target is None: # figuring out what target actually is is handled in add_warning target = p elif points is None: try: points = int(p) except ValueError: wrapper.reply(messages["fwarn_points_invalid"]) return if points < 0: wrapper.reply(messages["fwarn_points_invalid"]) return elif notes is not None: notes += " " + p elif reason is not None: rsp = p.split("|", 1) if len(rsp) > 1: notes = rsp[1] reason += " " + rsp[0] elif p[0] == ":": if p == ":": reason = "" else: reason = p[1:] elif p[0] == "~": if p == "~": wrapper.reply(messages["fwarn_syntax"]) return expires = p[1:] else: # sanctions are the only thing left here sanc = p.split("=", 1) if sanc[0] == "deny": try: cmds = sanc[1].split(",") normalized_cmds = set() for cmd in cmds: normalized = None for obj in COMMANDS[cmd]: # do not allow denying in-game commands (vote, see, etc.) # this technically traps goat too, so special case that, as we want # goat to be deny-able. Furthermore, the warn command cannot be denied. if (not obj.playing and not obj.roles) or obj.name == "goat": normalized = obj.name if normalized == "warn": normalized = None if normalized is None: wrapper.reply( messages["fwarn_deny_invalid_command"].format( cmd)) return normalized_cmds.add(normalized) sanctions["deny"] = normalized_cmds except IndexError: wrapper.reply(messages["fwarn_deny_invalid"]) return elif sanc[0] == "stasis": try: sanctions["stasis"] = int(sanc[1]) except (IndexError, ValueError): wrapper.reply(messages["fwarn_stasis_invalid"]) return elif sanc[0] == "tempban": try: banamt = sanc[1] suffix = banamt[-1] if suffix not in ("d", "h", "m"): sanctions["tempban"] = int(banamt) else: banamt = int(banamt[:-1]) if suffix == "d": sanctions["tempban"] = datetime.utcnow( ) + timedelta(days=banamt) elif suffix == "h": sanctions["tempban"] = datetime.utcnow( ) + timedelta(hours=banamt) elif suffix == "m": sanctions["tempban"] = datetime.utcnow( ) + timedelta(minutes=banamt) except (IndexError, ValueError): wrapper.reply(messages["fwarn_tempban_invalid"]) return else: # not a valid sanction, assume this is the start of the reason reason = p if target is None or points is None or reason is None: wrapper.reply(messages["fwarn_add_syntax"]) return reason = reason.strip() if notes is not None: notes = notes.strip() # convert expires into a proper datetime if expires is None: expires = var.DEFAULT_EXPIRY if expires.lower() in messages["never_aliases"]: expires = None try: warn_id = add_warning(wrapper.client, target, points, wrapper.source.nick, reason, notes, expires, sanctions) # FIXME except ValueError: wrapper.reply(messages["fwarn_expiry_invalid"]) return if warn_id is False: wrapper.reply(messages["fwarn_cannot_add"]) else: wrapper.reply(messages["fwarn_added"].format(warn_id)) # Log to ops/log channel (even if the warning was placed in that channel) if var.LOG_CHANNEL: log_reason = reason if notes is not None: log_reason += " | " + notes if expires is None: log_length = messages["fwarn_log_add_noexpiry"] else: log_length = messages["fwarn_log_add_expiry"].format(expires) log_msg = messages["fwarn_log_add"].format( warn_id, target, wrapper.source.nick, log_reason, points, "" if points == 1 else "s", log_length) channels.get(var.LOG_CHANNEL).send(log_msg, prefix=var.LOG_PREFIX)
def warn(var, wrapper, message): """View and acknowledge your warnings.""" # !warn list [-all] [page] - lists all active warnings, or all warnings if all passed # !warn view <id> - views details on warning id # !warn ack <id> - acknowledges warning id # Default if only !warn is given is to do !warn list. params = re.split(" +", message) try: command = params.pop(0) if command == "": command = "list" except IndexError: command = "list" if command not in ("list", "view", "ack", "help"): wrapper.reply(messages["warn_usage"]) return if command == "help": try: subcommand = params.pop(0) except IndexError: wrapper.reply(messages["warn_help_syntax"]) return if subcommand not in ("list", "view", "ack", "help"): wrapper.reply(messages["warn_usage"]) return wrapper.reply(messages["warn_{0}_syntax".format(subcommand)]) return if command == "list": list_all = False page = 1 try: list_all = params.pop(0) target = params.pop(0) page = int(params.pop(0)) except IndexError: pass except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return try: if list_all and list_all != "-all": page = int(list_all) list_all = False elif list_all == "-all": list_all = True except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return acc, hm = parse_warning_target(wrapper.source.nick) warnings = db.list_warnings(acc, hm, expired=list_all, skip=(page - 1) * 10, show=11) points = db.get_warning_points(acc, hm) wrapper.pm(messages["warn_list_header"].format( points, "" if points == 1 else "s")) i = 0 for warn in warnings: i += 1 if (i == 11): parts = [] if list_all: parts.append("-all") parts.append(str(page + 1)) wrapper.pm(messages["warn_list_footer"].format( " ".join(parts))) break start = "" end = "" ack = "" if warn["expires"] is not None: if warn["expired"]: expires = messages["fwarn_list_expired"].format( warn["expires"]) else: expires = messages["fwarn_view_expires"].format( warn["expires"]) else: expires = messages["fwarn_never_expires"] if warn["expired"]: start = "\u000314" end = " [\u00037{0}\u000314]\u0003".format( messages["fwarn_expired"]) if not warn["ack"]: ack = "\u0002!\u0002 " wrapper.pm(messages["warn_list"].format( start, ack, warn["id"], warn["issued"], warn["reason"], warn["amount"], "" if warn["amount"] == 1 else "s", expires, end)) if i == 0: wrapper.pm(messages["fwarn_list_empty"]) return if command == "view": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["warn_view_syntax"]) return acc, hm = parse_warning_target(wrapper.source.nick) warning = db.get_warning(warn_id, acc, hm) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return if warning["expired"]: expires = messages["fwarn_view_expired"].format(warning["expires"]) elif warning["expires"] is None: expires = messages["fwarn_view_active"].format( messages["fwarn_never_expires"]) else: expires = messages["fwarn_view_active"].format( messages["fwarn_view_expires"].format(warning["expires"])) wrapper.pm(messages["warn_view_header"].format( warning["id"], warning["issued"], warning["amount"], "" if warning["amount"] == 1 else "s", expires)) wrapper.pm(warning["reason"]) sanctions = [] if not warning["ack"]: sanctions.append(messages["warn_view_ack"].format(warning["id"])) if warning["sanctions"]: sanctions.append(messages["fwarn_view_sanctions"]) if "stasis" in warning["sanctions"]: if warning["sanctions"]["stasis"] != 1: sanctions.append( messages["fwarn_view_stasis_plural"].format( warning["sanctions"]["stasis"])) else: sanctions.append(messages["fwarn_view_stasis_sing"]) if "deny" in warning["sanctions"]: sanctions.append(messages["fwarn_view_deny"].format(", ".join( warning["sanctions"]["deny"]))) if sanctions: wrapper.pm(" ".join(sanctions)) return if command == "ack": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["warn_ack_syntax"]) return acc, hm = parse_warning_target(wrapper.source.nick) warning = db.get_warning(warn_id, acc, hm) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return # only add stasis if this is the first time this warning is being acknowledged if not warning["ack"] and warning["sanctions"].get("stasis", 0) > 0: db.set_stasis(warning["sanctions"]["stasis"], acc, hm, relative=True) db.init_vars() db.acknowledge_warning(warn_id) wrapper.reply(messages["fwarn_done"]) return
def add_warning(cli, target, amount, actor, reason, notes=None, expires=None, sanctions=None): tacc, thm = parse_warning_target(target) if tacc is None and thm is None: return False if actor not in var.USERS and actor != users.Bot.nick: return False sacc = None shm = None if actor in var.USERS: sacc = var.USERS[actor]["account"] shm = actor + "!" + var.USERS[actor]["ident"] + "@" + var.USERS[actor][ "host"] # Turn expires into a datetime if we were passed a string; note that no error checking is performed here if isinstance(expires, str): exp_suffix = expires[-1] exp_amount = int(expires[:-1]) if exp_suffix == "d": expires = datetime.utcnow() + timedelta(days=exp_amount) elif exp_suffix == "h": expires = datetime.utcnow() + timedelta(hours=exp_amount) elif exp_suffix == "m": expires = datetime.utcnow() + timedelta(minutes=exp_amount) else: raise ValueError("Invalid expiration string") elif isinstance(expires, int): expires = datetime.utcnow() + timedelta(days=expires) # Round expires to the nearest minute (30s rounds up) if isinstance(expires, datetime): round_add = 0 if expires.second >= 30: round_add = 1 expires -= timedelta(seconds=expires.second, microseconds=expires.microsecond) expires += timedelta(minutes=round_add) # determine if we need to automatically add any sanctions if sanctions is None: sanctions = {} prev = db.get_warning_points(tacc, thm) cur = prev + amount if amount > 0: _get_auto_sanctions(sanctions, prev, cur) sid = db.add_warning(tacc, thm, sacc, shm, amount, reason, notes, expires) if "stasis" in sanctions: db.add_warning_sanction(sid, "stasis", sanctions["stasis"]) if "deny" in sanctions: for cmd in sanctions["deny"]: db.add_warning_sanction(sid, "deny command", cmd) if "tempban" in sanctions: # this inserts into the bantrack table too (acclist, hmlist) = db.add_warning_sanction(sid, "tempban", sanctions["tempban"]) cmodes = [] for acc in acclist: cmodes.append(("+b", "{0}{1}".format(var.ACCOUNT_PREFIX, acc))) for hm in hmlist: cmodes.append(("+b", "*!*@{0}".format(hm.split("@")[1]))) channels.Main.mode(*cmodes) for user in channels.Main.users: if user.account in acclist: channels.Main.kick( user, messages["tempban_kick"].format(nick=user, botnick=users.Bot.nick, reason=reason)) elif user.host in hmlist: channels.Main.kick( user, messages["tempban_kick"].format(nick=user, botnick=users.Bot.nick, reason=reason)) # Update any tracking vars that may have changed due to this db.init_vars() return sid
def fwarn(var, wrapper, message): """Issues a warning to someone or views warnings.""" # !fwarn list [-all] [nick] [page] # -all => Shows all warnings, if omitted only shows active (non-expired and non-deleted) ones. # nick => nick to view warnings for. Can also be a hostmask in nick!user@host form. If nick # is not online, interpreted as an account name. To specify an account if nick is online, # use =account. If not specified, shows all warnings on the bot. # !fwarn view <id> - views details on warning id # !fwarn del <id> - deletes warning id # !fwarn set <id> [~expiry] [reason] [| notes] # !fwarn add <nick> <points> [~expiry] [sanctions] [:]<reason> [| notes] # e.g. !fwarn add lykos 1 ~30d deny=goat,gstats stasis=5 Spamming | I secretly just hate him # nick => nick to warn. Can also be a hostmask in nick!user@host form. If nick is not online, # interpreted as an account name. To specify an account if nick is online, use =account. # points => Warning points, must be above 0 # ~expiry => Expiration time, must be suffixed with d (days), h (hours), or m (minutes) # sanctions => list of sanctions. Valid sanctions are: # deny: denies access to the listed commands # stasis: gives the user stasis # reason => Reason, required. If the first word of the reason is also a sanction, prefix it with : # |notes => Secret notes, not shown to the user (only shown if viewing the warning in PM) # If specified, must be prefixed with |. This means | is not a valid character for use # in reasons (no escaping is performed). params = message.split() target = None points = None expires = None sanctions = {} reason = None notes = None try: command = params.pop(0) except IndexError: wrapper.reply(messages["fwarn_usage"]) return if command not in ("list", "view", "add", "del", "set", "help"): # if what follows is a number, assume we're viewing or setting a warning # (depending on number of params) # if it's another string, assume we're adding or listing, again depending # on number of params params.insert(0, command) try: num = int(command) if len(params) == 1: command = "view" else: command = "set" except ValueError: if len(params) < 3 or params[1] == "-all": command = "list" if len(params) > 1 and params[1] == "-all": # fwarn list expects these two params in a different order params.pop(1) params.insert(0, "-all") else: command = "add" if command == "help": try: subcommand = params.pop(0) except IndexError: wrapper.reply(messages["fwarn_usage"]) return if subcommand not in ("list", "view", "add", "del", "set", "help"): wrapper.reply(messages["fwarn_usage"]) return wrapper.reply(messages["fwarn_{0}_syntax".format(subcommand)]) return if command == "list": list_all = False page = 1 try: list_all = params.pop(0) target = params.pop(0) page = int(params.pop(0)) except IndexError: pass except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return try: if list_all and list_all != "-all": if target is not None: page = int(target) target = list_all list_all = False elif list_all == "-all": list_all = True except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return try: page = int(target) target = None except (TypeError, ValueError): pass if target is not None: acc, hm = parse_warning_target(target) if acc is None and hm is None: wrapper.reply(messages["fwarn_nick_invalid"]) return warnings = db.list_warnings(acc, hm, expired=list_all, deleted=list_all, skip=(page-1)*10, show=11) points = db.get_warning_points(acc, hm) wrapper.pm(messages["fwarn_list_header"].format(target, points, "" if points == 1 else "s")) else: warnings = db.list_all_warnings(list_all=list_all, skip=(page-1)*10, show=11) i = 0 for warn in warnings: i += 1 if (i == 11): parts = [] if list_all: parts.append("-all") if target is not None: parts.append(target) parts.append(str(page + 1)) wrapper.pm(messages["fwarn_list_footer"].format(" ".join(parts))) break start = "" end = "" ack = "" if warn["expires"] is not None: if warn["expired"]: expires = messages["fwarn_list_expired"].format(warn["expires"]) else: expires = messages["fwarn_view_expires"].format(warn["expires"]) else: expires = messages["fwarn_never_expires"] if warn["deleted"]: start = "\u000314" end = " [\u00034{0}\u000314]\u0003".format(messages["fwarn_deleted"]) elif warn["expired"]: start = "\u000314" end = " [\u00037{0}\u000314]\u0003".format(messages["fwarn_expired"]) if not warn["ack"]: ack = "\u0002!\u0002 " wrapper.pm(messages["fwarn_list"].format( start, ack, warn["id"], warn["issued"], warn["target"], warn["sender"], warn["reason"], warn["amount"], "" if warn["amount"] == 1 else "s", expires, end)) if i == 0: wrapper.pm(messages["fwarn_list_empty"]) return if command == "view": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["fwarn_view_syntax"]) return warning = db.get_warning(warn_id) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return if warning["deleted"]: expires = messages["fwarn_view_deleted"].format(warning["deleted_on"], warning["deleted_by"]) elif warning["expired"]: expires = messages["fwarn_view_expired"].format(warning["expires"]) elif warning["expires"] is None: expires = messages["fwarn_view_active"].format(messages["fwarn_never_expires"]) else: expires = messages["fwarn_view_active"].format(messages["fwarn_view_expires"].format(warning["expires"])) wrapper.pm(messages["fwarn_view_header"].format( warning["id"], warning["target"], warning["issued"], warning["sender"], warning["amount"], "" if warning["amount"] == 1 else "s", expires)) reason = [warning["reason"]] if warning["notes"] is not None: reason.append(warning["notes"]) wrapper.pm(" | ".join(reason)) sanctions = [] if not warning["ack"]: sanctions.append(messages["fwarn_view_ack"]) if warning["sanctions"]: sanctions.append(messages["fwarn_view_sanctions"]) if "stasis" in warning["sanctions"]: if warning["sanctions"]["stasis"] != 1: sanctions.append(messages["fwarn_view_stasis_plural"].format(warning["sanctions"]["stasis"])) else: sanctions.append(messages["fwarn_view_stasis_sing"]) if "deny" in warning["sanctions"]: sanctions.append(messages["fwarn_view_deny"].format(", ".join(warning["sanctions"]["deny"]))) if "tempban" in warning["sanctions"]: sanctions.append(messages["fwarn_view_tempban"].format(warning["sanctions"]["tempban"])) if sanctions: wrapper.pm(" ".join(sanctions)) return if command == "del": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["fwarn_del_syntax"]) return warning = db.get_warning(warn_id) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return acc, hm = parse_warning_target(wrapper.source.nick) db.del_warning(warn_id, acc, hm) wrapper.reply(messages["fwarn_done"]) if var.LOG_CHANNEL: msg = messages["fwarn_log_del"].format( warn_id, warning["target"], hm, warning["reason"], (" | " + warning["notes"]) if warning["notes"] else "") channels.get(var.LOG_CHANNEL).send(msg, prefix=var.LOG_PREFIX) return if command == "set": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["fwarn_set_syntax"]) return warning = db.get_warning(warn_id) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return rsp = " ".join(params).split("|", 1) if len(rsp) == 1: rsp.append(None) reason, notes = rsp reason = reason.strip() # check for modified expiry expires = warning["expires"] rsp = reason.split(" ", 1) if rsp[0] and rsp[0][0] == "~": if len(rsp) == 1: rsp.append("") expires, reason = rsp expires = expires[1:] reason = reason.strip() if expires in messages["never_aliases"]: expires = None else: suffix = expires[-1] try: amount = int(expires[:-1]) except ValueError: wrapper.reply(messages["fwarn_expiry_invalid"]) return if amount <= 0: wrapper.reply(messages["fwarn_expiry_invalid"]) return issued = datetime.strptime(warning["issued"], "%Y-%m-%d %H:%M:%S") if suffix == "d": expires = issued + timedelta(days=amount) elif suffix == "h": expires = issued + timedelta(hours=amount) elif suffix == "m": expires = issued + timedelta(minutes=amount) else: wrapper.reply(messages["fwarn_expiry_invalid"]) return round_add = 0 if expires.second >= 30: round_add = 1 expires -= timedelta(seconds=expires.second, microseconds=expires.microsecond) expires += timedelta(minutes=round_add) # maintain existing reason if none was specified if not reason: reason = warning["reason"] # maintain existing notes if none were specified if notes is not None: notes = notes.strip() if not notes: notes = None else: notes = warning["notes"] db.set_warning(warn_id, expires, reason, notes) wrapper.reply(messages["fwarn_done"]) if var.LOG_CHANNEL: changes = [] if expires != warning["expires"]: oldexpiry = warning["expires"] if warning["expires"] else messages["fwarn_log_set_noexpiry"] newexpiry = expires if expires else messages["fwarn_log_set_noexpiry"] changes.append(messages["fwarn_log_set_expiry"].format(oldexpiry, newexpiry)) if reason != warning["reason"]: changes.append(messages["fwarn_log_set_reason"].format(warning["reason"], reason)) if notes != warning["notes"]: if warning["notes"]: changes.append(messages["fwarn_log_set_notes"].format(warning["notes"], notes)) else: changes.append(messages["fwarn_log_set_notes_new"].format(notes)) if changes: log_msg = messages["fwarn_log_set"].format(warn_id, warning["target"], wrapper.source.nick, " | ".join(changes)) channels.get(var.LOG_CHANNEL).send(log_msg, prefix=var.LOG_PREFIX) return # command == "add" while params: p = params.pop(0) if target is None: # figuring out what target actually is is handled in add_warning target = p elif points is None: try: points = int(p) except ValueError: wrapper.reply(messages["fwarn_points_invalid"]) return if points < 0: wrapper.reply(messages["fwarn_points_invalid"]) return elif notes is not None: notes += " " + p elif reason is not None: rsp = p.split("|", 1) if len(rsp) > 1: notes = rsp[1] reason += " " + rsp[0] elif p[0] == ":": if p == ":": reason = "" else: reason = p[1:] elif p[0] == "~": if p == "~": wrapper.reply(messages["fwarn_syntax"]) return expires = p[1:] else: # sanctions are the only thing left here sanc = p.split("=", 1) if sanc[0] == "deny": try: cmds = sanc[1].split(",") normalized_cmds = set() for cmd in cmds: normalized = None for obj in COMMANDS[cmd]: # do not allow denying in-game commands (vote, see, etc.) # this technically traps goat too, so special case that, as we want # goat to be deny-able. Furthermore, the warn command cannot be denied. if (not obj.playing and not obj.roles) or obj.name == "goat": normalized = obj.name if normalized == "warn": normalized = None if normalized is None: wrapper.reply(messages["fwarn_deny_invalid_command"].format(cmd)) return normalized_cmds.add(normalized) sanctions["deny"] = normalized_cmds except IndexError: wrapper.reply(messages["fwarn_deny_invalid"]) return elif sanc[0] == "stasis": try: sanctions["stasis"] = int(sanc[1]) except (IndexError, ValueError): wrapper.reply(messages["fwarn_stasis_invalid"]) return elif sanc[0] == "tempban": try: banamt = sanc[1] suffix = banamt[-1] if suffix not in ("d", "h", "m"): sanctions["tempban"] = int(banamt) else: banamt = int(banamt[:-1]) if suffix == "d": sanctions["tempban"] = datetime.utcnow() + timedelta(days=banamt) elif suffix == "h": sanctions["tempban"] = datetime.utcnow() + timedelta(hours=banamt) elif suffix == "m": sanctions["tempban"] = datetime.utcnow() + timedelta(minutes=banamt) except (IndexError, ValueError): wrapper.reply(messages["fwarn_tempban_invalid"]) return else: # not a valid sanction, assume this is the start of the reason reason = p if target is None or points is None or reason is None: wrapper.reply(messages["fwarn_add_syntax"]) return reason = reason.strip() if notes is not None: notes = notes.strip() # convert expires into a proper datetime if expires is None: expires = var.DEFAULT_EXPIRY if expires.lower() in messages["never_aliases"]: expires = None try: warn_id = add_warning(wrapper.client, target, points, wrapper.source.nick, reason, notes, expires, sanctions) # FIXME except ValueError: wrapper.reply(messages["fwarn_expiry_invalid"]) return if warn_id is False: wrapper.reply(messages["fwarn_cannot_add"]) else: wrapper.reply(messages["fwarn_added"].format(warn_id)) # Log to ops/log channel (even if the warning was placed in that channel) if var.LOG_CHANNEL: log_reason = reason if notes is not None: log_reason += " | " + notes if expires is None: log_length = messages["fwarn_log_add_noexpiry"] else: log_length = messages["fwarn_log_add_expiry"].format(expires) log_msg = messages["fwarn_log_add"].format(warn_id, target, wrapper.source.nick, log_reason, points, "" if points == 1 else "s", log_length) channels.get(var.LOG_CHANNEL).send(log_msg, prefix=var.LOG_PREFIX)
def warn(var, wrapper, message): """View and acknowledge your warnings.""" # !warn list [-all] [page] - lists all active warnings, or all warnings if all passed # !warn view <id> - views details on warning id # !warn ack <id> - acknowledges warning id # Default if only !warn is given is to do !warn list. params = re.split(" +", message) try: command = params.pop(0) if command == "": command = "list" except IndexError: command = "list" if command not in ("list", "view", "ack", "help"): wrapper.reply(messages["warn_usage"]) return if command == "help": try: subcommand = params.pop(0) except IndexError: wrapper.reply(messages["warn_help_syntax"]) return if subcommand not in ("list", "view", "ack", "help"): wrapper.reply(messages["warn_usage"]) return wrapper.reply(messages["warn_{0}_syntax".format(subcommand)]) return if command == "list": list_all = False page = 1 try: list_all = params.pop(0) target = params.pop(0) page = int(params.pop(0)) except IndexError: pass except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return try: if list_all and list_all != "-all": page = int(list_all) list_all = False elif list_all == "-all": list_all = True except ValueError: wrapper.reply(messages["fwarn_page_invalid"]) return acc, hm = parse_warning_target(wrapper.source.nick) warnings = db.list_warnings(acc, hm, expired=list_all, skip=(page-1)*10, show=11) points = db.get_warning_points(acc, hm) wrapper.pm(messages["warn_list_header"].format(points, "" if points == 1 else "s")) i = 0 for warn in warnings: i += 1 if (i == 11): parts = [] if list_all: parts.append("-all") parts.append(str(page + 1)) wrapper.pm(messages["warn_list_footer"].format(" ".join(parts))) break start = "" end = "" ack = "" if warn["expires"] is not None: if warn["expired"]: expires = messages["fwarn_list_expired"].format(warn["expires"]) else: expires = messages["fwarn_view_expires"].format(warn["expires"]) else: expires = messages["fwarn_never_expires"] if warn["expired"]: start = "\u000314" end = " [\u00037{0}\u000314]\u0003".format(messages["fwarn_expired"]) if not warn["ack"]: ack = "\u0002!\u0002 " wrapper.pm(messages["warn_list"].format( start, ack, warn["id"], warn["issued"], warn["reason"], warn["amount"], "" if warn["amount"] == 1 else "s", expires, end)) if i == 0: wrapper.pm(messages["fwarn_list_empty"]) return if command == "view": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["warn_view_syntax"]) return acc, hm = parse_warning_target(wrapper.source.nick) warning = db.get_warning(warn_id, acc, hm) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return if warning["expired"]: expires = messages["fwarn_view_expired"].format(warning["expires"]) elif warning["expires"] is None: expires = messages["fwarn_view_active"].format(messages["fwarn_never_expires"]) else: expires = messages["fwarn_view_active"].format(messages["fwarn_view_expires"].format(warning["expires"])) wrapper.pm(messages["warn_view_header"].format( warning["id"], warning["issued"], warning["amount"], "" if warning["amount"] == 1 else "s", expires)) wrapper.pm(warning["reason"]) sanctions = [] if not warning["ack"]: sanctions.append(messages["warn_view_ack"].format(warning["id"])) if warning["sanctions"]: sanctions.append(messages["fwarn_view_sanctions"]) if "stasis" in warning["sanctions"]: if warning["sanctions"]["stasis"] != 1: sanctions.append(messages["fwarn_view_stasis_plural"].format(warning["sanctions"]["stasis"])) else: sanctions.append(messages["fwarn_view_stasis_sing"]) if "deny" in warning["sanctions"]: sanctions.append(messages["fwarn_view_deny"].format(", ".join(warning["sanctions"]["deny"]))) if sanctions: wrapper.pm(" ".join(sanctions)) return if command == "ack": try: warn_id = params.pop(0) if warn_id[0] == "#": warn_id = warn_id[1:] warn_id = int(warn_id) except (IndexError, ValueError): wrapper.reply(messages["warn_ack_syntax"]) return acc, hm = parse_warning_target(wrapper.source.nick) warning = db.get_warning(warn_id, acc, hm) if warning is None: wrapper.reply(messages["fwarn_invalid_warning"]) return # only add stasis if this is the first time this warning is being acknowledged if not warning["ack"] and warning["sanctions"].get("stasis", 0) > 0: db.set_stasis(warning["sanctions"]["stasis"], acc, hm, relative=True) db.init_vars() db.acknowledge_warning(warn_id) wrapper.reply(messages["fwarn_done"]) return
def add_warning(cli, target, amount, actor, reason, notes=None, expires=None, sanctions=None): tacc, thm = parse_warning_target(target) if tacc is None and thm is None: return False if actor not in var.USERS and actor != users.Bot.nick: return False sacc = None shm = None if actor in var.USERS: sacc = var.USERS[actor]["account"] shm = actor + "!" + var.USERS[actor]["ident"] + "@" + var.USERS[actor]["host"] # Turn expires into a datetime if we were passed a string; note that no error checking is performed here if isinstance(expires, str): exp_suffix = expires[-1] exp_amount = int(expires[:-1]) if exp_suffix == "d": expires = datetime.utcnow() + timedelta(days=exp_amount) elif exp_suffix == "h": expires = datetime.utcnow() + timedelta(hours=exp_amount) elif exp_suffix == "m": expires = datetime.utcnow() + timedelta(minutes=exp_amount) else: raise ValueError("Invalid expiration string") elif isinstance(expires, int): expires = datetime.utcnow() + timedelta(days=expires) # Round expires to the nearest minute (30s rounds up) if isinstance(expires, datetime): round_add = 0 if expires.second >= 30: round_add = 1 expires -= timedelta(seconds=expires.second, microseconds=expires.microsecond) expires += timedelta(minutes=round_add) # determine if we need to automatically add any sanctions if sanctions is None: sanctions = {} prev = db.get_warning_points(tacc, thm) cur = prev + amount if amount > 0: _get_auto_sanctions(sanctions, prev, cur) sid = db.add_warning(tacc, thm, sacc, shm, amount, reason, notes, expires) if "stasis" in sanctions: db.add_warning_sanction(sid, "stasis", sanctions["stasis"]) if "deny" in sanctions: for cmd in sanctions["deny"]: db.add_warning_sanction(sid, "deny command", cmd) if "tempban" in sanctions: # this inserts into the bantrack table too (acclist, hmlist) = db.add_warning_sanction(sid, "tempban", sanctions["tempban"]) cmodes = [] for acc in acclist: cmodes.append(("+b", "{0}{1}".format(var.ACCOUNT_PREFIX, acc))) for hm in hmlist: cmodes.append(("+b", "*!*@{0}".format(hm.split("@")[1]))) channels.Main.mode(*cmodes) for user in channels.Main.users: if user.account in acclist: channels.Main.kick(user, messages["tempban_kick"].format(nick=user, botnick=users.Bot.nick, reason=reason)) elif user.host in hmlist: channels.Main.kick(user, messages["tempban_kick"].format(nick=user, botnick=users.Bot.nick, reason=reason)) # Update any tracking vars that may have changed due to this db.init_vars() return sid
def add_warning(cli, target, amount, actor, reason, notes=None, expires=None, sanctions=None): # make 0-point warnings no-op successfully, otherwise we add warnings when things like PART_PENALTY is 0 if amount == 0: return False tacc, thm = parse_warning_target(target) if tacc is None and thm is None: return False if actor not in var.USERS and actor != botconfig.NICK: return False sacc = None shm = None if actor in var.USERS: sacc = var.USERS[actor]["account"] shm = actor + "!" + var.USERS[actor]["ident"] + "@" + var.USERS[actor]["host"] # Turn expires into a datetime if we were passed a string; note that no error checking is performed here if isinstance(expires, str): exp_suffix = expires[-1] exp_amount = int(expires[:-1]) if exp_suffix == "d": expires = datetime.utcnow() + timedelta(days=exp_amount) elif exp_suffix == "h": expires = datetime.utcnow() + timedelta(hours=exp_amount) elif exp_suffix == "m": expires = datetime.utcnow() + timedelta(minutes=exp_amount) else: raise ValueError("Invalid expiration string") elif isinstance(expires, int): expires = datetime.utcnow() + timedelta(days=expires) # Round expires to the nearest minute (30s rounds up) if isinstance(expires, datetime): round_add = 0 if expires.second >= 30: round_add = 1 expires -= timedelta(seconds=expires.second, microseconds=expires.microsecond) expires += timedelta(minutes=round_add) # determine if we need to automatically add any sanctions if sanctions is None: sanctions = {} prev = db.get_warning_points(tacc, thm) cur = prev + amount for (mn, mx, sanc) in var.AUTO_SANCTION: if (prev < mn and cur >= mn) or (prev >= mn and prev <= mx and cur <= mx): if "stasis" in sanc: if "stasis" not in sanctions: sanctions["stasis"] = sanc["stasis"] else: sanctions["stasis"] = max(sanctions["stasis"], sanc["stasis"]) if "scalestasis" in sanc: (a, b, c) = sanc["scalestasis"] amt = (a * cur * cur) + (b * cur) + c if "stasis" not in sanctions: sanctions["stasis"] = amt else: sanctions["stasis"] = max(sanctions["stasis"], amt) if "deny" in sanc: if "deny" not in sanctions: sanctions["deny"] = set(sanc["deny"]) else: sanctions["deny"].update(sanc["deny"]) if "tempban" in sanc: # tempban's param can either be a fixed expiry time or a number # which indicates the warning point threshold that the ban will be lifted at # if two are set at once, the threshold takes precedence over set times # within each category, a larger set time or a lower threshold takes precedence exp = None ths = None if isinstance(sanc["tempban"], str) and sanc["tempban"][-1] in ("d", "h", "m"): amt = int(sanc["tempban"][:-1]) dur = sanc["tempban"][-1] if dur == "d": exp = datetime.utcnow() + timedelta(days=amt) elif dur == "h": exp = datetime.utcnow() + timedelta(hours=amt) elif dur == "m": exp = datetime.utcnow() + timedelta(minutes=amt) else: ths = int(sanc["tempban"]) if "tempban" in sanctions: if isinstance(sanctions["tempban"], datetime): if ths is not None: sanctions["tempban"] = ths else: sanctions["tempban"] = max(sanctions["tempban"], exp) elif ths is not None: sanctions["tempban"] = min(sanctions["tempban"], ths) elif ths is not None: sanctions["tempban"] = ths else: sanctions["tempban"] = exp sid = db.add_warning(tacc, thm, sacc, shm, amount, reason, notes, expires) if "stasis" in sanctions: db.add_warning_sanction(sid, "stasis", sanctions["stasis"]) if "deny" in sanctions: for cmd in sanctions["deny"]: db.add_warning_sanction(sid, "deny command", cmd) if "tempban" in sanctions: # this inserts into the bantrack table too (acclist, hmlist) = db.add_warning_sanction(sid, "tempban", sanctions["tempban"]) cmodes = [] for acc in acclist: cmodes.append(("+b", "{0}{1}".format(var.ACCOUNT_PREFIX, acc))) for hm in hmlist: cmodes.append(("+b", "*!*@{0}".format(hm))) mass_mode(cli, cmodes, []) for (nick, user) in var.USERS.items(): if user["account"] in acclist: cli.kick(botconfig.CHANNEL, nick, messages["tempban_kick"].format(nick=nick, botnick=botconfig.NICK, reason=reason)) elif user["host"] in hmlist: cli.kick(botconfig.CHANNEL, nick, messages["tempban_kick"].format(nick=nick, botnick=botconfig.NICK, reason=reason)) # Update any tracking vars that may have changed due to this db.init_vars() return sid
def add_warning(cli, target, amount, actor, reason, notes=None, expires=None, sanctions=None): # make 0-point warnings no-op successfully, otherwise we add warnings when things like PART_PENALTY is 0 if amount == 0: return False tacc, thm = parse_warning_target(target) if tacc is None and thm is None: return False if actor not in var.USERS and actor != botconfig.NICK: return False sacc = None shm = None if actor in var.USERS: sacc = var.USERS[actor]["account"] shm = actor + "!" + var.USERS[actor]["ident"] + "@" + var.USERS[actor]["host"] # Turn expires into a datetime if we were passed a string; note that no error checking is performed here if isinstance(expires, str): exp_suffix = expires[-1] exp_amount = int(expires[:-1]) if exp_suffix == "d": expires = datetime.utcnow() + timedelta(days=exp_amount) elif exp_suffix == "h": expires = datetime.utcnow() + timedelta(hours=exp_amount) elif exp_suffix == "m": expires = datetime.utcnow() + timedelta(minutes=exp_amount) else: raise ValueError("Invalid expiration string") elif isinstance(expires, int): expires = datetime.utcnow() + timedelta(days=expires) # Round expires to the nearest minute (30s rounds up) if isinstance(expires, datetime): round_add = 0 if expires.second >= 30: round_add = 1 expires -= timedelta(seconds=expires.second, microseconds=expires.microsecond) expires += timedelta(minutes=round_add) # determine if we need to automatically add any sanctions if sanctions is None: sanctions = {} prev = db.get_warning_points(tacc, thm) cur = prev + amount for (mn, mx, sanc) in var.AUTO_SANCTION: if (prev < mn and cur >= mn) or (prev >= mn and prev <= mx and cur <= mx): if "stasis" in sanc: if "stasis" not in sanctions: sanctions["stasis"] = sanc["stasis"] else: sanctions["stasis"] = max(sanctions["stasis"], sanc["stasis"]) if "scalestasis" in sanc: (a, b, c) = sanc["scalestasis"] amt = (a * cur * cur) + (b * cur) + c if "stasis" not in sanctions: sanctions["stasis"] = amt else: sanctions["stasis"] = max(sanctions["stasis"], amt) if "deny" in sanc: if "deny" not in sanctions: sanctions["deny"] = set(sanc["deny"]) else: sanctions["deny"].update(sanc["deny"]) if "tempban" in sanc: # tempban's param can either be a fixed expiry time or a number # which indicates the warning point threshold that the ban will be lifted at # if two are set at once, the threshold takes precedence over set times # within each category, a larger set time or a lower threshold takes precedence exp = None ths = None if sanc["tempban"][-1] in ("d", "h", "m"): amt = int(sanc["tempban"][:-1]) dur = sanc["tempban"][-1] if dur == "d": exp = datetime.utcnow() + timedelta(days=amt) elif dur == "h": exp = datetime.utcnow() + timedelta(hours=amt) elif dur == "m": exp = datetime.utcnow() + timedelta(minutes=amt) else: ths = int(sanc["tempban"]) if "tempban" in sanctions: if isinstance(sanctions["tempban"], datetime): if ths is not None: sanctions["tempban"] = ths else: sanctions["tempban"] = max(sanctions["tempban"], exp) elif ths is not None: sanctions["tempban"] = min(sanctions["tempban"], ths) elif ths is not None: sanctions["tempban"] = ths else: sanctions["tempban"] = exp sid = db.add_warning(tacc, thm, sacc, shm, amount, reason, notes, expires) if "stasis" in sanctions: db.add_warning_sanction(sid, "stasis", sanctions["stasis"]) if "deny" in sanctions: for cmd in sanctions["deny"]: db.add_warning_sanction(sid, "deny command", cmd) if "tempban" in sanctions: # this inserts into the bantrack table too (acclist, hmlist) = db.add_warning_sanction(sid, "tempban", sanctions["tempban"]) cmodes = [] for acc in acclist: cmodes.append(("+b", "{0}{1}".format(var.ACCOUNT_PREFIX, acc))) for hm in hmlist: cmodes.append(("+b", "*!*@{0}".format(hm))) mass_mode(cli, cmodes, []) for (nick, user) in var.USERS.items(): if user["account"] in acclist: cli.kick(botconfig.CHANNEL, nick, messages["tempban_kick"].format(nick=nick, botnick=botconfig.NICK, reason=reason)) elif user["host"] in hmlist: cli.kick(botconfig.CHANNEL, nick, messages["tempban_kick"].format(nick=nick, botnick=botconfig.NICK, reason=reason)) # Update any tracking vars that may have changed due to this db.init_vars() return sid