Garmelon/euphoria-wtf
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Convert into yaboli module

Browse source
This commit is contained in:
Joscha 2018-09-12 21:43:12 +00:00
parent 102b3da887
commit 1c0097a5aa
1 changed files with 16 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

47
wtf.py
View file

@ -64,8 +64,7 @@ class WtfDB(yaboli.Database):
with db: with db:
db.execute("UPDATE acronyms SET deleted = 1 WHERE acronym_id = ?", (acronym_id,)) db.execute("UPDATE acronyms SET deleted = 1 WHERE acronym_id = ?", (acronym_id,))
class Wtf: class Wtf(yaboli.Module):
SHORT_DESCRIPTION = "a database of explanations for words, acronyms and initialisms"
DESCRIPTION = ( DESCRIPTION = (
"'wtf' is a database of explanations for words, acronyms and initialisms." "'wtf' is a database of explanations for words, acronyms and initialisms."
" It is inspired by the linux wtf program and uses its acronyms," " It is inspired by the linux wtf program and uses its acronyms,"
@ -80,6 +79,12 @@ class Wtf:
) )
AUTHOR = "Created by @Garmy using github.com/Garmelon/yaboli\n" AUTHOR = "Created by @Garmy using github.com/Garmelon/yaboli\n"
SHORT_DESCRIPTION = "a database of explanations for words, acronyms and initialisms"
SHORT_HELP = SHORT_DESCRIPTION
LONG_DESCRIPTION = DESCRIPTION + COMMANDS
LONG_HELP = DESCRIPTION + COMMANDS + AUTHOR
RE_IS = r"\s*is\s+(.*)" RE_IS = r"\s*is\s+(.*)"
RE_ADD = r"\s*add\s+(\S+)\s+(.+)" RE_ADD = r"\s*add\s+(\S+)\s+(.+)"
RE_DETAIL = r"\s*detail\s+(.*)" RE_DETAIL = r"\s*detail\s+(.*)"
@ -91,6 +96,12 @@ class Wtf:
def __init__(self, dbfile): def __init__(self, dbfile):
self.db = WtfDB(dbfile) self.db = WtfDB(dbfile)
async def on_send(self, room, message):
await self.trigger_wtf_is(room, message)
async def on_command_general(self, room, message, command, argstr):
await self.command_wtf(room, message, command, argstr)
@yaboli.command("wtf") @yaboli.command("wtf")
async def command_wtf(self, room, message, argstr): async def command_wtf(self, room, message, argstr):
match_is = re.fullmatch(self.RE_IS, argstr) match_is = re.fullmatch(self.RE_IS, argstr)
@ -176,34 +187,6 @@ class Wtf:
lines.append(f"{term!r} not found.") lines.append(f"{term!r} not found.")
return lines return lines
class WtfBot(yaboli.Bot):
SHORT_HELP = Wtf.SHORT_DESCRIPTION
LONG_HELP = Wtf.DESCRIPTION + Wtf.COMMANDS + Wtf.AUTHOR
def __init__(self, nick, wtfdbfile, cookiefile=None):
super().__init__(nick, cookiefile=cookiefile)
self.wtf = Wtf(wtfdbfile)
async def on_send(self, room, message):
await super().on_send(room, message)
await self.wtf.trigger_wtf_is(room, message)
async def on_command_specific(self, room, message, command, nick, argstr):
if similar(nick, room.session.nick) and not argstr:
await self.botrulez_ping(room, message, command)
await self.botrulez_help(room, message, command, text=self.LONG_HELP)
await self.botrulez_uptime(room, message, command)
await self.botrulez_kill(room, message, command)
await self.botrulez_restart(room, message, command)
async def on_command_general(self, room, message, command, argstr):
if not argstr:
await self.botrulez_ping(room, message, command)
await self.botrulez_help(room, message, command, text=self.SHORT_HELP)
await self.wtf.command_wtf(room, message, command, argstr)
def main(configfile): def main(configfile):
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
@ -213,7 +196,9 @@ def main(configfile):
nick = config.get("general", "nick") nick = config.get("general", "nick")
cookiefile = config.get("general", "cookiefile", fallback=None) cookiefile = config.get("general", "cookiefile", fallback=None)
wtfdbfile = config.get("general", "wtfdbfile") wtfdbfile = config.get("general", "wtfdbfile")
bot = WtfBot(nick, wtfdbfile, cookiefile=cookiefile) module = Wtf(wtfdbfile)
bot = yaboli.ModuleBot(module, nick, cookiefile=cookiefile)
for room, password in config.items("rooms"): for room, password in config.items("rooms"):
if not password: if not password: