add: Update to include domain & hostname composite attributes

composite_attributes_proposal
chrisr3d 2018-11-19 09:33:49 +01:00
parent d127865139
commit 7da6d41020
No known key found for this signature in database
GPG Key ID: 6BBED1B63A6D639F
1 changed files with 5 additions and 1 deletions

View File

@ -4,7 +4,7 @@ import json
from pyeupi import PyEUPI
misperrors = {'error': 'Error'}
mispattributes = {'input': ['hostname', 'domain', 'url'], 'output': ['freetext']}
mispattributes = {'input': ['hostname', 'hostname|port', 'domain', 'domain|ip', 'url'], 'output': ['freetext']}
moduleinfo = {'version': '0.1', 'author': 'Raphaël Vinot',
'description': 'Query the Phishing Initiative service (https://phishing-initiative.lu)',
'module-type': ['expansion', 'hover']}
@ -18,8 +18,12 @@ def handler(q=False):
request = json.loads(q)
if request.get('hostname'):
toquery = request['hostname']
elif request.get('hostname|port'):
toquery, _ = request['hostname|port'].split('|')
elif request.get('domain'):
toquery = request['domain']
elif request.get('domain|ip'):
toquery, _ = request['domain|ip'].split('|')
elif request.get('url'):
toquery = request['url']
else: