geoip lookup added. had to downgrade to python2 :(

master
panki27 5 years ago
parent c8c3b69740
commit f7be02bf0a

38
ip.py

@ -1,7 +1,7 @@
#!/usr/bin/env python3
#!/usr/bin/env python
import cgi
import os
import html
#import html
F_PLAIN = 0
F_XML = 1
@ -9,31 +9,43 @@ F_JSON = 2
def header():
print("Content-type: text/plain")
print()
print("")
def main():
params = cgi.FieldStorage()
out = {
"ip": None
"ip": None,
"country": None,
"continent": None,
"location": None
}
header()
remote_ip = html.escape(os.environ["REMOTE_ADDR"])
out["ip"] = cgi.escape(os.environ["REMOTE_ADDR"])
format = F_PLAIN
if(len(params) > 0):
for key in params:
if key == "format":
r_format = params[key].value
if r_format == "xml":
format = F_PLAIN
print("lol xml are you serious? get with the times")
elif r_format == "json":
format = F_JSON
elif key == "and":
extras = params[key].value.split(',')
if "geo" in extras:
from geoip import geolite2
match = geolite2.lookup(out["ip"].encode("UTF-8"))
if match is not None:
out["country"] = match.country
out["continent"] = match.continent
out["location"] = match.location
out["ip"] = remote_ip
for item in out:
if format == F_PLAIN:
print(out[item])
elif format == F_JSON:
import json
print(json.dumps(out))
if format == F_PLAIN:
for item in out:
if out[item] is not None:
print(out[item])
elif format == F_JSON:
import json
print(json.dumps(dict((k, v) for k, v in out.items() if v is not None)))
if __name__ == "__main__":
main()
Loading…
Cancel
Save