nemunaire
/
nemubot
1
0
Fork 0
Code Releases Activity

whois: update module

This commit is contained in:
nemunaire 2018-12-30 10:59:10 +01:00
parent b349d22370
commit fa0f2e93ef
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
modules/whois.py
View File

@ -14,7 +14,7 @@ from nemubot.module.more import Response
from nemubot.module.networking.page import headers from nemubot.module.networking.page import headers
PASSWD_FILE = None PASSWD_FILE = None
# You can get one with: curl -b "sessionid=YOURSESSIONID" 'https://accounts.cri.epita.net/api/users/' > users.json # You can get one with: curl -b "sessionid=YOURSESSIONID" 'https://accounts.cri.epita.net/api/users/?limit=10000' > users.json
APIEXTRACT_FILE = None APIEXTRACT_FILE = None
def load(context): def load(context):
@ -49,7 +49,7 @@ def load(context):
class Login: class Login:
def __init__(self, line=None, login=None, uidNumber=None, cn=None, promo=None, **kwargs): def __init__(self, line=None, login=None, uidNumber=None, firstname=None, lastname=None, promo=None, **kwargs):
if line is not None: if line is not None:
s = line.split(":") s = line.split(":")
self.login = s[0] self.login = s[0]
@ -61,19 +61,25 @@ class Login:
self.login = login self.login = login
self.uid = uidNumber self.uid = uidNumber
self.promo = promo self.promo = promo
self.cn = cn self.cn = firstname + " " + lastname
self.gid = "epita" + promo try:
self.gid = "epita" + str(int(promo))
except:
self.gid = promo
def get_promo(self): def get_promo(self):
if hasattr(self, "promo"): if hasattr(self, "promo"):
return self.promo return self.promo
if hasattr(self, "home"): if hasattr(self, "home"):
return self.home.split("/")[2].replace("_", " ") try:
return self.home.split("/")[2].replace("_", " ")
except:
return self.gid
def get_photo(self): def get_photo(self):
if self.login in context.data.getNode("pics").index: if self.login in context.data.getNode("pics").index:
return context.data.getNode("pics").index[self.login]["url"] return context.data.getNode("pics").index[self.login]["url"]
for url in [ "https://photos.cri.epita.net/%s", "https://static.acu.epita.fr/photos/%s", "https://static.acu.epita.fr/photos/%s/%%s" % self.gid, "https://intra-bocal.epitech.eu/trombi/%s.jpg", "http://whois.23.tf/p/%s/%%s.jpg" % self.gid ]: for url in [ "https://photos.cri.epita.fr/%s", "https://intra-bocal.epitech.eu/trombi/%s.jpg" ]:
url = url % self.login url = url % self.login
try: try:
_, status, _, _ = headers(url) _, status, _, _ = headers(url)
@ -91,7 +97,7 @@ def login_lookup(login, search=False):
if APIEXTRACT_FILE: if APIEXTRACT_FILE:
with open(APIEXTRACT_FILE, encoding="utf-8") as f: with open(APIEXTRACT_FILE, encoding="utf-8") as f:
api = json.load(f) api = json.load(f)
for l in api: for l in api["results"]:
if (not search and l["login"] == login) or (search and (("login" in l and l["login"].find(login) != -1) or ("cn" in l and l["cn"].find(login) != -1) or ("uid" in l and str(l["uid"]) == login))): if (not search and l["login"] == login) or (search and (("login" in l and l["login"].find(login) != -1) or ("cn" in l and l["cn"].find(login) != -1) or ("uid" in l and str(l["uid"]) == login))):
yield Login(**l) yield Login(**l)