update ip.py

Teahouse-Studios · Jan 23, 2025 · a62f287 · a62f287
1 parent efa0528
commit a62f287
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/core/utils/ip.py b/core/utils/ip.py
@@ -6,9 +6,10 @@
 
 
 def append_ip(ip_info: dict):
-    Secret.add(ip_info["ip"])
-    Secret.ip_address = ip_info["country"]
-    Secret.ip_country = ip_info["ip"]
+    if ip_info and ip_info.get("ip"):
+        Secret.add(ip_info["ip"])
+    Secret.ip_address = ip_info.get("country")
+    Secret.ip_country = ip_info.get("ip")
 
 
 async def fetch_ip_info() -> dict:
@@ -20,3 +21,4 @@ async def fetch_ip_info() -> dict:
     except Exception:
         Logger.error("Failed to get IP information.")
         Logger.error(traceback.format_exc())
+        return {}