mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-11 06:00:59 -05:00
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
# coding=utf8
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
|
|
# sopel imports
|
|
import sopel.module
|
|
|
|
import spicemanip
|
|
|
|
import sopel_modules.SpiceBot as SpiceBot
|
|
|
|
|
|
@SpiceBot.prerun.prerun('module')
|
|
@sopel.module.commands('sherlock', 'username')
|
|
def bot_command_sherlock(bot, trigger):
|
|
|
|
data = SpiceBot.sherlock.dict
|
|
netlist = list(SpiceBot.sherlock.dict.keys())
|
|
|
|
username = spicemanip.main(trigger.sb["args"], 1) or trigger.nick
|
|
trigger.sb["args"] = spicemanip.main(trigger.sb["args"], "2+", 'list')
|
|
|
|
checklist = netlist
|
|
checklistname = 'all'
|
|
if SpiceBot.inlist(username, netlist):
|
|
checklistname = SpiceBot.inlist_match(username, netlist)
|
|
checklist = [checklistname]
|
|
username = spicemanip.main(trigger.sb["args"], 1) or trigger.nick
|
|
|
|
bot.osd("Checking username " + username + " in " + checklistname + " network.")
|
|
|
|
inlist = []
|
|
notinlist = []
|
|
|
|
for social_network in checklist:
|
|
|
|
url = data.get(social_network).get("url").format(username)
|
|
error_type = data.get(social_network).get("errorType")
|
|
cant_have_period = data.get(social_network).get("noPeriod")
|
|
|
|
if ("." in username) and (cant_have_period == "True"):
|
|
while ("." in username):
|
|
username = username.replace(".", '')
|
|
|
|
r, error_type = SpiceBot.sherlock.make_request(url=url, error_type=error_type, social_network=social_network)
|
|
|
|
if error_type == "message":
|
|
error = data.get(social_network).get("errorMsg")
|
|
# Checks if the error message is in the HTML
|
|
if error not in r.text:
|
|
inlist.append(social_network)
|
|
else:
|
|
notinlist.append(social_network)
|
|
|
|
elif error_type == "status_code":
|
|
# Checks if the status code of the repsonse is 404
|
|
if not r.status_code == 404:
|
|
inlist.append(social_network)
|
|
else:
|
|
notinlist.append(social_network)
|
|
|
|
elif error_type == "response_url":
|
|
error = data.get(social_network).get("errorUrl")
|
|
# Checks if the redirect url is the same as the one defined in data.json
|
|
if error not in r.url:
|
|
inlist.append(social_network)
|
|
else:
|
|
notinlist.append(social_network)
|
|
|
|
if inlist != []:
|
|
if checklistname != 'all':
|
|
bot.osd(["The username " + username + " is in " + checklist[0]])
|
|
else:
|
|
bot.osd(["The username " + username + " is in the following:", spicemanip.main(inlist, "andlist")])
|
|
if notinlist != []:
|
|
bot.osd(["The username " + username + " is NOT in the following:", spicemanip.main(notinlist, "andlist")])
|