Added lookup by country code

Hannah Ward 2016-08-12 14:45:28 +01:00
parent 9ac1b3900a
commit 4f5059fca4
No known key found for this signature in database
GPG Key ID: BA89E572EE1B4C5F
1 changed files with 59 additions and 0 deletions

View File

@ -0,0 +1,59 @@
import json
import requests
misperrors = {'error': 'Error'}
mispattributes = {'input': ['hostname', 'domain'], 'output': ['ip-src',
# possible module-types: 'expansion', 'hover' or both
moduleinfo = {'version': '1', 'author': 'Hannah Ward',
'description': 'Expand Country Codes',
'module-type': ['expansion', 'hover']}
# config fields that your code expects from the site admin
moduleconfig = []
common_tlds = {"com":"Commercial (Worldwide)",
"org":"Organisation (Worldwide)",
"net":"Network (Worldwide)",
"int":"International (Worldwide)",
"edu":"Education (Usually USA)",
"gov":"Government (USA)"
def handler(q=False):
if q is False:
return False
request = json.loads(q)
domain = request["domain"]
# Get the extension
ext = domain.split(".")[-1]
# Check if if's a common, non country one
if ext in common_tlds.keys():
val = common_tlds[ext]
# Retrieve a json full of country info
codes = requests.get("").json()
if not codes["StatusMsg"] == "OK":
val = "Unknown"
# Find our code based on TLD
codes = codes["Results"]
for code in codes.keys():
if codes[code]["CountryCodes"]["tld"] == ext:
val = codes[code]["Name"]
r = {'results': [{'types':['text'], 'values':[val]}]}
return r
def introspection():
return mispattributes
def version():
moduleinfo['config'] = moduleconfig
return moduleinfo