Move all exports to @Utils.export calls
This commit is contained in:
parent
8b0314b190
commit
6eb8b1ba6d
15 changed files with 99 additions and 147 deletions
|
@ -1,11 +1,9 @@
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("serverset", {"setting": "accept-invites",
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
exports.add("serverset", {"setting": "accept-invites",
|
|
||||||
"help": "Set whether I accept invites on this server",
|
"help": "Set whether I accept invites on this server",
|
||||||
"validate": Utils.bool_or_none})
|
"validate": Utils.bool_or_none})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.invite")
|
@Utils.hook("received.invite")
|
||||||
def on_invite(self, event):
|
def on_invite(self, event):
|
||||||
if event["server"].is_own_nickname(event["target_user"].nickname):
|
if event["server"].is_own_nickname(event["target_user"].nickname):
|
||||||
|
|
|
@ -1,11 +1,9 @@
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "automode",
|
||||||
|
"help": "Disable/Enable automode", "validate": Utils.bool_or_none})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
_name = "AutoMode"
|
_name = "AutoMode"
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
exports.add("channelset", {"setting": "automode",
|
|
||||||
"help": "Disable/Enable automode",
|
|
||||||
"validate": Utils.bool_or_none})
|
|
||||||
|
|
||||||
def _check_modes(self, channel, user):
|
def _check_modes(self, channel, user):
|
||||||
identified_account = user.get_identified_account()
|
identified_account = user.get_identified_account()
|
||||||
|
|
|
@ -1,20 +1,18 @@
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "highlight-spam-threshold",
|
||||||
_name = "Channel Op"
|
"help": "Set the number of nicknames in a message that qualifies as spam",
|
||||||
def __init__(self, bot, events, exports):
|
"validate": Utils.int_or_none})
|
||||||
exports.add("channelset", {"setting": "highlight-spam-threshold",
|
@Utils.export("channelset", {"setting": "highlight-spam-protection",
|
||||||
"help": "Set the number of nicknames in a message that "
|
|
||||||
"qualifies as spam", "validate": Utils.int_or_none})
|
|
||||||
exports.add("channelset", {"setting": "highlight-spam-protection",
|
|
||||||
"help": "Enable/Disable highlight spam protection",
|
"help": "Enable/Disable highlight spam protection",
|
||||||
"validate": Utils.bool_or_none})
|
"validate": Utils.bool_or_none})
|
||||||
exports.add("channelset", {"setting": "highlight-spam-ban",
|
@Utils.export("channelset", {"setting": "highlight-spam-ban",
|
||||||
"help": "Enable/Disable banning highlight spammers "
|
"help": "Enable/Disable banning highlight spammers "
|
||||||
"instead of just kicking", "validate": Utils.bool_or_none})
|
"instead of just kicking", "validate": Utils.bool_or_none})
|
||||||
exports.add("channelset", {"setting": "ban-format",
|
@Utils.export("channelset", {"setting": "ban-format",
|
||||||
"help": "Set ban format ($n = nick, $u = username, "
|
"help": "Set ban format ($n = nick, $u = username, $h = hostname)"})
|
||||||
"$h = hostname)"})
|
class Module(ModuleManager.BaseModule):
|
||||||
|
_name = "Channel Op"
|
||||||
|
|
||||||
@Utils.hook("received.command.kick|k", channel_only=True,
|
@Utils.hook("received.command.kick|k", channel_only=True,
|
||||||
require_mode="o", usage="<nickname> [reason]", min_args=1)
|
require_mode="o", usage="<nickname> [reason]", min_args=1)
|
||||||
|
|
|
@ -1,25 +1,21 @@
|
||||||
#--require-config virustotal-api-key
|
#--require-config virustotal-api-key
|
||||||
|
|
||||||
import re
|
import re
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
URL_VIRUSTOTAL = "https://www.virustotal.com/vtapi/v2/url/report"
|
URL_VIRUSTOTAL = "https://www.virustotal.com/vtapi/v2/url/report"
|
||||||
RE_URL = re.compile(r"https?://\S+", re.I)
|
RE_URL = re.compile(r"https?://\S+", re.I)
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "check-urls",
|
||||||
def __init__(self, bot, events, exports):
|
"help": "Enable/Disable automatically checking for malicious URLs",
|
||||||
self.bot = bot
|
"validate": Utils.bool_or_none})
|
||||||
self.events = events
|
@Utils.export("serverset", {"setting": "check-urls",
|
||||||
exports.add("channelset", {"setting": "check-urls",
|
"help": "Enable/Disable automatically checking for malicious URLs",
|
||||||
"help": "Enable/Disable automatically checking for "
|
"validate": Utils.bool_or_none})
|
||||||
"malicious URLs", "validate": Utils.bool_or_none})
|
@Utils.export("channelset", {"setting": "check-urls-kick",
|
||||||
exports.add("serverset", {"setting": "check-urls",
|
|
||||||
"help": "Enable/Disable automatically checking for "
|
|
||||||
"malicious URLs", "validate": Utils.bool_or_none})
|
|
||||||
exports.add("channelset", {"setting": "check-urls-kick",
|
|
||||||
"help": "Enable/Disable automatically kicking users that "
|
"help": "Enable/Disable automatically kicking users that "
|
||||||
"send malicious URLs", "validate": Utils.bool_or_none})
|
"send malicious URLs", "validate": Utils.bool_or_none})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.message.channel")
|
@Utils.hook("received.message.channel")
|
||||||
def message(self, event):
|
def message(self, event):
|
||||||
match = RE_URL.search(event["message"])
|
match = RE_URL.search(event["message"])
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
import re
|
import re
|
||||||
from src import EventManager, Utils
|
from src import EventManager, ModuleManager, Utils
|
||||||
|
|
||||||
STR_MORE = "%s (more...)" % Utils.FONT_RESET
|
STR_MORE = "%s (more...)" % Utils.FONT_RESET
|
||||||
STR_CONTINUED = "(...continued) "
|
STR_CONTINUED = "(...continued) "
|
||||||
|
@ -43,17 +43,13 @@ class StdErr(Out):
|
||||||
def prefix(self):
|
def prefix(self):
|
||||||
return Utils.color(Utils.bold(self.module_name), Utils.COLOR_RED)
|
return Utils.color(Utils.bold(self.module_name), Utils.COLOR_RED)
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "command-prefix",
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
self.events = events
|
|
||||||
|
|
||||||
exports.add("channelset", {"setting": "command-prefix",
|
|
||||||
"help": "Set the command prefix used in this channel"})
|
"help": "Set the command prefix used in this channel"})
|
||||||
exports.add("serverset", {"setting": "command-prefix",
|
@Utils.export("serverset", {"setting": "command-prefix",
|
||||||
"help": "Set the command prefix used on this server"})
|
"help": "Set the command prefix used on this server"})
|
||||||
exports.add("serverset", {"setting": "identity-mechanism",
|
@Utils.export("serverset", {"setting": "identity-mechanism",
|
||||||
"help": "Set the identity mechanism for this server"})
|
"help": "Set the identity mechanism for this server"})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("new.user|channel")
|
@Utils.hook("new.user|channel")
|
||||||
def new(self, event):
|
def new(self, event):
|
||||||
if "user" in event:
|
if "user" in event:
|
||||||
|
|
|
@ -1,13 +1,10 @@
|
||||||
import datetime
|
import datetime
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("serverset", {"setting": "ctcp-responses",
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
self.bot = bot
|
|
||||||
exports.add("serverset", {"setting": "ctcp-responses",
|
|
||||||
"help": "Set whether I respond to CTCPs on this server",
|
"help": "Set whether I respond to CTCPs on this server",
|
||||||
"validate": Utils.bool_or_none})
|
"validate": Utils.bool_or_none})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.message.private")
|
@Utils.hook("received.message.private")
|
||||||
def private_message(self, event):
|
def private_message(self, event):
|
||||||
if event["message"][0] == "\x01" and event["message"][-1] == "\x01":
|
if event["message"][0] == "\x01" and event["message"][-1] == "\x01":
|
||||||
|
|
|
@ -13,24 +13,20 @@ DUCK_MESSAGE_RARE = ["beep boop!", "QUACK QUACK QUACK QUACK QUACK!!", "HONK!",
|
||||||
DUCK_MINIMUM_MESSAGES = 10
|
DUCK_MINIMUM_MESSAGES = 10
|
||||||
DUCK_MINIMUM_UNIQUE = 3
|
DUCK_MINIMUM_UNIQUE = 3
|
||||||
|
|
||||||
|
@Utils.export("channelset", {"setting": "ducks-enabled",
|
||||||
|
"help": "Toggle ducks!", "validate": Utils.bool_or_none})
|
||||||
|
@Utils.export("channelset", {"setting": "ducks-kick",
|
||||||
|
"help": "Should the bot kick if there's no duck?",
|
||||||
|
"validate": Utils.bool_or_none})
|
||||||
|
@Utils.export("channelset", {"setting": "ducks-min-unique",
|
||||||
|
"help": "Minimum unique users required to talk before a duck spawns.",
|
||||||
|
"validate": Utils.int_or_none})
|
||||||
|
@Utils.export("channelset", {"setting": "ducks-min-messages",
|
||||||
|
"help": "Minimum messages between ducks spawning.",
|
||||||
|
"validate": Utils.int_or_none})
|
||||||
class Module(object):
|
class Module(object):
|
||||||
def __init__(self, bot, events, exports):
|
def __init__(self, bot, events, exports):
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
exports.add("channelset", {"setting": "ducks-enabled",
|
|
||||||
"help": "Toggle ducks!", "validate": Utils.bool_or_none})
|
|
||||||
|
|
||||||
exports.add("channelset", {"setting": "ducks-kick",
|
|
||||||
"help": "Should the bot kick if there's no duck?",
|
|
||||||
"validate": Utils.bool_or_none})
|
|
||||||
|
|
||||||
exports.add("channelset", {"setting": "ducks-min-unique",
|
|
||||||
"help": "Minimum unique users required to talk before a "
|
|
||||||
"duck spawns.", "validate": Utils.int_or_none})
|
|
||||||
|
|
||||||
exports.add("channelset", {"setting": "ducks-min-messages",
|
|
||||||
"help": "Minimum messages between ducks spawning.",
|
|
||||||
"validate": Utils.int_or_none})
|
|
||||||
|
|
||||||
for server in self.bot.servers.values():
|
for server in self.bot.servers.values():
|
||||||
for channel in server.channels.values():
|
for channel in server.channels.values():
|
||||||
|
|
|
@ -1,10 +1,8 @@
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "greeting",
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
exports.add("channelset", {"setting": "greeting",
|
|
||||||
"help": "Set a greeting to send to users when they join"})
|
"help": "Set a greeting to send to users when they join"})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.join")
|
@Utils.hook("received.join")
|
||||||
def join(self, event):
|
def join(self, event):
|
||||||
greeting = event["channel"].get_setting("greeting", None)
|
greeting = event["channel"].get_setting("greeting", None)
|
||||||
|
|
|
@ -1,16 +1,13 @@
|
||||||
import re, time
|
import re, time
|
||||||
from src import EventManager, Utils
|
from src import EventManager, ModuleManager, Utils
|
||||||
|
|
||||||
REGEX_KARMA = re.compile("^(.*[^-+])[-+]*(\+{2,}|\-{2,})$")
|
REGEX_KARMA = re.compile("^(.*[^-+])[-+]*(\+{2,}|\-{2,})$")
|
||||||
KARMA_DELAY_SECONDS = 3
|
KARMA_DELAY_SECONDS = 3
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "karma-verbose",
|
||||||
def __init__(self, bot, events, exports):
|
"help": "Disable/Enable automatically responding to karma changes",
|
||||||
self.events = events
|
"validate": Utils.bool_or_none})
|
||||||
exports.add("channelset", {"setting": "karma-verbose",
|
class Module(ModuleManager.BaseModule):
|
||||||
"help": "Disable/Enable automatically responding to "
|
|
||||||
"karma changes", "validate": Utils.bool_or_none})
|
|
||||||
|
|
||||||
@Utils.hook("new.user")
|
@Utils.hook("new.user")
|
||||||
def new_user(self, event):
|
def new_user(self, event):
|
||||||
event["user"].last_karma = None
|
event["user"].last_karma = None
|
||||||
|
|
|
@ -1,18 +1,13 @@
|
||||||
#--require-config lastfm-api-key
|
#--require-config lastfm-api-key
|
||||||
|
|
||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
URL_SCROBBLER = "http://ws.audioscrobbler.com/2.0/"
|
URL_SCROBBLER = "http://ws.audioscrobbler.com/2.0/"
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("set", {"setting": "lastfm", "help": "Set last.fm username"})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
_name = "last.fm"
|
_name = "last.fm"
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
self.bot = bot
|
|
||||||
self.events = events
|
|
||||||
|
|
||||||
exports.add("set", {"setting": "lastfm",
|
|
||||||
"help": "Set username on last.fm"})
|
|
||||||
|
|
||||||
@Utils.hook("received.command.np|listening|nowplaying", usage="[username]")
|
@Utils.hook("received.command.np|listening|nowplaying", usage="[username]")
|
||||||
def np(self, event):
|
def np(self, event):
|
||||||
|
|
|
@ -1,11 +1,9 @@
|
||||||
import base64
|
import base64
|
||||||
from src import EventManager, Utils
|
from src import EventManager, ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("serverset", {"setting": "nickserv-password",
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
exports.add("serverset", {"setting": "nickserv-password",
|
|
||||||
"help": "Set the nickserv password for this server"})
|
"help": "Set the nickserv password for this server"})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.numeric.001", priority=EventManager.PRIORITY_URGENT)
|
@Utils.hook("received.numeric.001", priority=EventManager.PRIORITY_URGENT)
|
||||||
def on_connect(self, event):
|
def on_connect(self, event):
|
||||||
nickserv_password = event["server"].get_setting(
|
nickserv_password = event["server"].get_setting(
|
||||||
|
|
|
@ -1,11 +1,5 @@
|
||||||
import base64
|
import base64
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
class Module(object):
|
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
exports.add("serverset", {"setting": "sasl",
|
|
||||||
"help": "Set the sasl username/password for this server",
|
|
||||||
"validate": self._validate})
|
|
||||||
|
|
||||||
def _validate(self, s):
|
def _validate(self, s):
|
||||||
mechanism = s
|
mechanism = s
|
||||||
|
@ -13,6 +7,10 @@ class Module(object):
|
||||||
mechanism, arguments = s.split(" ", 1)
|
mechanism, arguments = s.split(" ", 1)
|
||||||
return {"mechanism": mechanism, "args": arguments}
|
return {"mechanism": mechanism, "args": arguments}
|
||||||
|
|
||||||
|
@Utils.export("serverset", {"setting": "sasl",
|
||||||
|
"help": "Set the sasl username/password for this server",
|
||||||
|
"validate": _validate})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.cap.ls")
|
@Utils.hook("received.cap.ls")
|
||||||
def on_cap(self, event):
|
def on_cap(self, event):
|
||||||
has_sasl = "sasl" in event["capabilities"]
|
has_sasl = "sasl" in event["capabilities"]
|
||||||
|
|
|
@ -1,20 +1,16 @@
|
||||||
import re, traceback
|
import re, traceback
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
REGEX_SPLIT = re.compile("(?<!\\\\)/")
|
REGEX_SPLIT = re.compile("(?<!\\\\)/")
|
||||||
REGEX_SED = re.compile("^s/")
|
REGEX_SED = re.compile("^s/")
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "sed",
|
||||||
def __init__(self, bot, events, exports):
|
|
||||||
self.events = events
|
|
||||||
|
|
||||||
exports.add("channelset", {"setting": "sed",
|
|
||||||
"help": "Disable/Enable sed in a channel",
|
"help": "Disable/Enable sed in a channel",
|
||||||
"validate": Utils.bool_or_none})
|
"validate": Utils.bool_or_none})
|
||||||
exports.add("channelset", {"setting": "sed-sender-only",
|
@Utils.export("channelset", {"setting": "sed-sender-only",
|
||||||
"help": "Disable/Enable sed only looking at the messages "
|
"help": "Disable/Enable sed only looking at the messages sent by the user",
|
||||||
"sent by the user", "validate": Utils.bool_or_none})
|
"validate": Utils.bool_or_none})
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
@Utils.hook("received.message.channel")
|
@Utils.hook("received.message.channel")
|
||||||
def channel_message(self, event):
|
def channel_message(self, event):
|
||||||
sed_split = re.split(REGEX_SPLIT, event["message"], 3)
|
sed_split = re.split(REGEX_SPLIT, event["message"], 3)
|
||||||
|
|
|
@ -1,16 +1,12 @@
|
||||||
#--require-config trakt-api-key
|
#--require-config trakt-api-key
|
||||||
|
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
URL_TRAKT = "https://api-v2launch.trakt.tv/users/%s/watching"
|
URL_TRAKT = "https://api-v2launch.trakt.tv/users/%s/watching"
|
||||||
URL_TRAKTSLUG = "https://trakt.tv/%s/%s"
|
URL_TRAKTSLUG = "https://trakt.tv/%s/%s"
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("set", {"setting": "trakt", "help": "Set username on trakt.tv"})
|
||||||
def __init__(self, bot, events, exports):
|
class Module(ModuleManager.BaseModule):
|
||||||
self.bot = bot
|
|
||||||
exports.add("set", {"setting": "trakt",
|
|
||||||
"help": "Set username on trakt.tv"})
|
|
||||||
|
|
||||||
@Utils.hook("received.command.nowwatching|nw", usage="[username]")
|
@Utils.hook("received.command.nowwatching|nw", usage="[username]")
|
||||||
def now_watching(self, event):
|
def now_watching(self, event):
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
#--require-config google-api-key
|
#--require-config google-api-key
|
||||||
|
|
||||||
import re
|
import re
|
||||||
from src import Utils
|
from src import ModuleManager, Utils
|
||||||
|
|
||||||
REGEX_YOUTUBE = re.compile(
|
REGEX_YOUTUBE = re.compile(
|
||||||
"https?://(?:www.)?(?:youtu.be/|youtube.com/watch\?[\S]*v=)([\w\-]{11})",
|
"https?://(?:www.)?(?:youtu.be/|youtube.com/watch\?[\S]*v=)([\w\-]{11})",
|
||||||
|
@ -16,15 +16,10 @@ URL_YOUTUBESHORT = "https://youtu.be/%s"
|
||||||
ARROW_UP = "▲"
|
ARROW_UP = "▲"
|
||||||
ARROW_DOWN = "▼"
|
ARROW_DOWN = "▼"
|
||||||
|
|
||||||
class Module(object):
|
@Utils.export("channelset", {"setting": "auto-youtube",
|
||||||
def __init__(self, bot, events, exports):
|
"help": "Disable/Enable automatically getting info from youtube URLs",
|
||||||
self.bot = bot
|
"validate": Utils.bool_or_none})
|
||||||
self.events = events
|
class Module(ModuleManager.BaseModule):
|
||||||
|
|
||||||
exports.add("channelset", {"setting": "auto-youtube",
|
|
||||||
"help": "Disable/Enable automatically getting info from "
|
|
||||||
"youtube URLs", "validate": Utils.bool_or_none})
|
|
||||||
|
|
||||||
def get_video_page(self, video_id, part):
|
def get_video_page(self, video_id, part):
|
||||||
return Utils.get_url(URL_YOUTUBEVIDEO, get_params={"part": part,
|
return Utils.get_url(URL_YOUTUBEVIDEO, get_params={"part": part,
|
||||||
"id": video_id, "key": self.bot.config["google-api-key"]},
|
"id": video_id, "key": self.bot.config["google-api-key"]},
|
||||||
|
|
Loading…
Reference in a new issue