mirror of
https://github.com/Xtao-Labs/iShotaBot.git
synced 2024-11-16 04:35:55 +00:00
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
from urllib.parse import urlparse
|
|
|
|
from pyrogram import Client, filters
|
|
from pyrogram.types import Message
|
|
|
|
from defs.ip import ip_info
|
|
from init import request, bot
|
|
|
|
|
|
@bot.on_message(filters.incoming & filters.command(["ip", f"ip@{bot.me.username}"]))
|
|
async def ip_command(_: Client, message: Message):
|
|
msg = await message.reply("正在查询中...")
|
|
rep_text = ""
|
|
reply = message.reply_to_message
|
|
if reply:
|
|
if reply.entities:
|
|
for num in range(0, len(reply.entities)):
|
|
url = reply.text[
|
|
reply.entities[num].offset : reply.entities[num].offset
|
|
+ reply.entities[num].length
|
|
]
|
|
url = urlparse(url)
|
|
if url.hostname or url.path:
|
|
if url.hostname:
|
|
url = url.hostname
|
|
else:
|
|
url = url.path
|
|
ipinfo_json = (
|
|
await request.get(
|
|
"http://ip-api.com/json/"
|
|
+ url
|
|
+ "?fields=status,message,country,regionName,city,"
|
|
"lat,lon,isp,"
|
|
"org,as,mobile,proxy,hosting,query"
|
|
)
|
|
).json()
|
|
if ipinfo_json["status"] == "success":
|
|
rep_text = ip_info(url, ipinfo_json)
|
|
text = ""
|
|
if message.entities:
|
|
for num in range(0, len(message.entities)):
|
|
url = message.text[
|
|
message.entities[num].offset : message.entities[num].offset
|
|
+ message.entities[num].length
|
|
]
|
|
url = urlparse(url)
|
|
if url.hostname or url.path:
|
|
if url.hostname:
|
|
url = url.hostname
|
|
else:
|
|
url = url.path
|
|
ipinfo_json = (
|
|
await request.get(
|
|
"http://ip-api.com/json/"
|
|
+ url
|
|
+ "?fields=status,message,country,regionName,city,lat,"
|
|
"lon,isp,"
|
|
"org,as,mobile,proxy,hosting,query"
|
|
)
|
|
).json()
|
|
if ipinfo_json["status"] == "success":
|
|
text = ip_info(url, ipinfo_json)
|
|
if text == "":
|
|
url = message.text[4:]
|
|
if url != "":
|
|
ipinfo_json = (
|
|
await request.get(
|
|
"http://ip-api.com/json/"
|
|
+ url
|
|
+ "?fields=status,message,country,regionName,city,lat,"
|
|
"lon,isp,"
|
|
"org,as,mobile,proxy,hosting,query"
|
|
)
|
|
).json()
|
|
if ipinfo_json["status"] == "success":
|
|
text = ip_info(url, ipinfo_json)
|
|
if rep_text == "" and text == "":
|
|
await msg.edit("没有找到要查询的 ip/域名 ...")
|
|
elif rep_text != "" and text != "":
|
|
await msg.edit(f"{rep_text}\n================\n{text}")
|
|
else:
|
|
await msg.edit(f"{rep_text}{text}")
|