nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

whois: add @lookup keyword to perform research in the list

Browse Source
This commit is contained in:
nemunaire 2017-07-05 22:54:07 +02:00
parent be9492c151
commit 08a7701238
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
modules/whois.py
View File

@ -30,7 +30,7 @@ def load(context):
context.data.getNode("pics").setIndex("login", "pict") context.data.getNode("pics").setIndex("login", "pict")
import nemubot.hooks import nemubot.hooks
context.add_hook(nemubot.hooks.Command(cmd_whois, "whois"), context.add_hook(nemubot.hooks.Command(cmd_whois, "whois", keywords={"lookup": "Perform a lookup of the begining of the login instead of an exact search."}),
"in","Command") "in","Command")
class Login: class Login:
@ -60,31 +60,38 @@ class Login:
return None return None
def found_login(login): def found_login(login, search=False):
if login in context.data.getNode("aliases").index: if login in context.data.getNode("aliases").index:
login = context.data.getNode("aliases").index[login]["to"] login = context.data.getNode("aliases").index[login]["to"]
login_ = login + ":" login_ = login + (":" if not search else "")
lsize = len(login_) lsize = len(login_)
with open(PASSWD_FILE, encoding="iso-8859-15") as f: with open(PASSWD_FILE, encoding="iso-8859-15") as f:
for l in f.readlines(): for l in f.readlines():
if l[:lsize] == login_: if l[:lsize] == login_:
return Login(l.strip()) yield Login(l.strip())
return None
def cmd_whois(msg): def cmd_whois(msg):
if len(msg.args) < 1: if len(msg.args) < 1:
raise IMException("Provide a name") raise IMException("Provide a name")
res = Response(channel=msg.channel, count=" (%d more logins)") def format_response(t):
for srch in msg.args: srch, l = t
l = found_login(srch) if type(l) is Login:
if l is not None:
pic = l.get_photo() pic = l.get_photo()
res.append_message("%s is %s (%s %s): %s%s" % (srch, l.cn.capitalize(), l.login, l.uid, l.get_promo(), " and looks like %s" % pic if pic is not None else "")) return "%s is %s (%s %s): %s%s" % (srch, l.cn.capitalize(), l.login, l.uid, l.get_promo(), " and looks like %s" % pic if pic is not None else "")
else: else:
res.append_message("Unknown %s :(" % srch) return l % srch
res = Response(channel=msg.channel, count=" (%d more logins)", line_treat=format_response)
for srch in msg.args:
found = False
for l in found_login(srch, "lookup" in msg.kwargs):
found = True
res.append_message((srch, l))
if not found:
res.append_message((srch, "Unknown %s :("))
return res return res
@hook.command("nicks") @hook.command("nicks")