fix(ip): punycode

App/Event.py

@@ -114,6 +114,7 @@ async def handle_ip_ali(bot, message, _config):
 async def handle_ip(bot, message, _config):
     msg = await bot.reply_to(message, f"正在查询 {message.text.split()[1]} ...", disable_web_page_preview=True)
     url = message.text.split()[1]
+    url = convert_to_punycode(url)
     status, data = await ipapi_ip(url)
     if status:
         if url == data["query"]:

Utils/IP.py

@@ -6,6 +6,7 @@
 import ipaddress
 import socket
 import aiohttp
+import idna
 from loguru import logger
 
 
@@ -20,6 +21,15 @@ def check_url(url):
         return get_ip_address(url), None
 
 
+def convert_to_punycode(domain):
+    try:
+        domain.encode('ascii')
+    except UnicodeEncodeError:
+        return idna.encode(domain).decode('ascii')
+    else:
+        return domain
+
+
 def get_ip_address(domain):
     try:
         addr_info = socket.getaddrinfo(domain, None, socket.AF_UNSPEC)

requirements.txt

@@ -4,4 +4,5 @@ pathlib>=1.0.1
 aiohttp>=3.9.1
 loguru>=0.7.2
 requests>=2.31.0
-Pillow>=10.1.0
+Pillow>=10.1.0
+idna>=3.4