Pep8 Fixes and layout fix

This commit is contained in:
L1ghtn1ng 2019-02-17 00:27:04 +00:00
parent 42e9eaeac7
commit 243f3b50ea
2 changed files with 9 additions and 6 deletions

View file

@ -4,7 +4,8 @@
import requests
import time
class search_intelx:
class SearchIntelx:
def __init__(self, word, limit):
self.word = word
@ -21,18 +22,21 @@ def do_search(self):
try:
user_agent = Core.get_user_agent()
headers = {'User-Agent': user_agent, 'x-key': self.key}
data = f'{{"term": "{self.word}", "maxresults": {self.limit}, "media": 0, "sort": 2 , "terminate": []}}'
# data is json that corresponds to what we are searching for, sort:2 means sort by most relevant
data = f'{{"term": "{self.word}", "maxresults": {self.limit}, "media": 0, "sort": 2 , "terminate": []}}'
r = requests.post(f'{self.database}phonebook/search', data=data, headers=headers)
if r.status_code == 400:
raise Exception('Invalid json was passed in.')
time.sleep(1)
# grab uuid to send get request to fetch data
uuid = r.json()['id']
url = f'{self.database}phonebook/search/result?id={uuid}&offset=0&limit={self.limit}'
r = requests.get(url, headers=headers)
time.sleep(1)
# to add in future grab status from r.text and check if more results can be gathered
# TODO: add in future grab status from r.text and check if more results can be gathered
if r.status_code != 200:
raise Exception('Error occurred while searching intelx.')
self.results = r.json()
@ -50,4 +54,3 @@ def get_emails(self):
def get_hostnames(self):
return self.info[1]

View file

@ -246,7 +246,7 @@ def start():
from discovery import intelxsearch
# Import locally or won't work.
try:
search = intelxsearch.search_intelx(word, limit)
search = intelxsearch.SearchIntelx(word, limit)
search.process()
emails = filter(search.get_emails())
all_emails.extend(emails)
@ -504,7 +504,7 @@ def start():
from discovery import intelxsearch
# Import locally or won't work.
try:
search = intelxsearch.search_intelx(word, limit)
search = intelxsearch.SearchIntelx(word, limit)
search.process()
emails = filter(search.get_emails())
all_emails.extend(emails)