blob: a6521c22f365a887b17919ea3f826fa09194c3d2 (
plain)
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
|
from nonebot.params import ArgStr
import geoip2.webservice
from ATRI.service import Service
from ATRI import conf
geoip = Service("GEOIP查询").document("search ip in MaxMind GEOIP databases")
query_geoip = geoip.on_command("ip查询", "查询IP的地理位置", aliases={"IP查询", "查询IP"})
LANG = "zh-CN"
@query_geoip.got("ip_address", prompt="地址是?(支持ipv4/ipv6)")
async def _(ip_address: str = ArgStr()):
with geoip2.webservice.Client(conf.GeoIP.account_id,
conf.GeoIP.license_key,
host="geolite.info") as client:
response = client.city(ip_address)
country = response.country.names[LANG]
city = response.city.names[LANG]
subdivision = ""
if response.subdivisions:
subdivision = response.subdivisions[0].names[LANG]
await query_geoip.finish(
f"IP: {ip_address}\n"
f"{country}{subdivision}{city}"
)
|
