mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-25 01:25:04 +08:00
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
|
import json
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
from lxml import etree
|
|||
|
|
|
|||
|
|
from command import CommandRegistry
|
|||
|
|
from commands import core
|
|||
|
|
|
|||
|
|
__registry__ = cr = CommandRegistry()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@cr.register('ip')
|
|||
|
|
def ip(args_text, ctx_msg):
|
|||
|
|
query = args_text.strip()
|
|||
|
|
if not query:
|
|||
|
|
core.echo('请指定要查询的 IP 或域名', ctx_msg)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
core.echo('正在查询，请稍等……', ctx_msg)
|
|||
|
|
|
|||
|
|
chinaz_url = 'http://ip.chinaz.com/%s'
|
|||
|
|
ipcn_url = 'http://ip.cn/?ip=%s'
|
|||
|
|
ipipnet_url = 'http://freeapi.ipip.net/%s'
|
|||
|
|
|
|||
|
|
found = False
|
|||
|
|
|
|||
|
|
# Get data from ChinaZ.com
|
|||
|
|
resp = requests.get(chinaz_url % query)
|
|||
|
|
if resp.status_code == 200:
|
|||
|
|
html = etree.HTML(resp.text)
|
|||
|
|
p_elems = html.xpath('//*[@id="leftinfo"]/div[3]/div[2]/p')
|
|||
|
|
if len(p_elems) > 1:
|
|||
|
|
p_elems = p_elems[1:]
|
|||
|
|
reply = 'ChinaZ:'
|
|||
|
|
for p_elem in p_elems:
|
|||
|
|
span_elems = p_elem.getchildren()
|
|||
|
|
reply += '\n' + span_elems[1].text.replace('.', '_') + ', ' + span_elems[3].text
|
|||
|
|
core.echo(reply, ctx_msg)
|
|||
|
|
found = True
|
|||
|
|
|
|||
|
|
# Get data from ip.cn
|
|||
|
|
resp = requests.get(ipcn_url % query, headers={'User-Agent': 'curl/7.47.0'})
|
|||
|
|
if resp.status_code == 200:
|
|||
|
|
# Example: 'IP：123.125.114.144 来自：北京市 联通'
|
|||
|
|
items = resp.text.strip().split('：')
|
|||
|
|
if len(items) == 3:
|
|||
|
|
reply = 'IP_CN:\n' + items[1].split(' ')[0].replace('.', '_') + ', ' + items[2]
|
|||
|
|
core.echo(reply, ctx_msg)
|
|||
|
|
found = True
|
|||
|
|
|
|||
|
|
# Get data from ipip.net
|
|||
|
|
resp = requests.get(ipipnet_url % query, headers={'User-Agent': 'curl/7.47.0'})
|
|||
|
|
if resp.status_code == 200 and resp.text.strip():
|
|||
|
|
# Example: '["中国","江苏","常州","","教育网"]'
|
|||
|
|
parts = json.loads(resp.text)
|
|||
|
|
reply = 'IPIP_NET\n' + query.replace('.', '_') + ' ' + ''.join(parts)
|
|||
|
|
core.echo(reply, ctx_msg)
|
|||
|
|
found = True
|
|||
|
|
|
|||
|
|
core.echo('以上' if found else '查询失败', ctx_msg)
|