1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#--depends-on config
import typing
from src import ModuleManager, utils
URL_OPENCAGE = "https://api.opencagedata.com/geocode/v1/json"
class LocationSetting(utils.Setting):
_func = None
def parse(self, value: str) -> typing.Any:
return self._func(value)
class Module(ModuleManager.BaseModule):
def on_load(self):
setting = LocationSetting("location", "Set your location",
example="London, GB")
setting._func = self._get_location
self.exports.add("set", setting)
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"]
lat = result["geometry"]["lat"]
lon = result["geometry"]["lng"]
name_parts = []
components = result["components"]
for part in ["town", "city", "state", "country"]:
if part in components:
name_parts.append(components[part])
if not name_parts:
name_parts.append(result["formatted"])
name = ", ".join(name_parts)
return {"timezone": timezone, "lat": lat, "lon": lon, "name": name}
|