pep.py/locationHelper.py

49 lines
1.0 KiB
Python

import urllib.request
import json
import consoleHelper
import bcolors
# API URL
URL = "http://ip.zxq.co/"
def getCountry(ip):
"""
Get country from IP address
ip -- IP Address
return -- Country code (2 letters)
"""
# Default value, sent if API is memeing
country = "XX"
try:
# Try to get country from Pikolo Aul's Go-Sanic ip API
country = json.loads(urllib.request.urlopen("{}/{}".format(URL, ip)).read().decode())["country"]
except:
consoleHelper.printColored("[!] Error in get country", bcolors.RED)
return country.upper()
def getLocation(ip):
"""
Get latitude and longitude from IP address
ip -- IP address
return -- [latitude, longitude]
"""
# Default value, sent if API is memeing
data = [0,0]
try:
# Try to get position from Pikolo Aul's Go-Sanic ip API
data = json.loads(urllib.request.urlopen("{}/{}".format(URL, ip)).read().decode())["loc"].split(",")
except:
consoleHelper.printColored("[!] Error in get position", bcolors.RED)
return [float(data[0]), float(data[1])]