bitbot-3.11-fork/modules/location.py

25 lines
1 KiB
Python

from src import ModuleManager, utils
import pytz
URL_OPENCAGE = "https://api.opencagedata.com/geocode/v1/json"
class Module(ModuleManager.BaseModule):
def on_load(self):
self.exports.add("set", {"setting": "location",
"help": "Set your location", "validate": self._get_location,
"human": lambda x: "%s, %s" % (x["city"], x["country"])})
def _get_location(self, s):
page = utils.http.request(URL_OPENCAGE, get_params={
"q": s, "key": self.bot.config["opencagedata-api-key"], "limit": "1"
}, json=True)
if page and page.data["results"]:
result = page.data["results"][0]
timezone = result["annotations"]["timezone"]["name"]
continent = result["components"]["continent"]
country = result["components"]["country"]
city = result["components"]["city"]
print("yes")
return {"timezone": timezone, "continent": continent,
"country": country, "city": city}