🐛 Fix ip_risk.py

AirportR · Feb 28, 2024 · 6d18ae0 · 6d18ae0
1 parent 8e887cb
commit 6d18ae0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/addons/builtin/ip_risk.py b/addons/builtin/ip_risk.py
@@ -21,10 +21,14 @@ async def fetch_ip_risk(Collector, session: aiohttp.ClientSession, proxy=None, r
     """
     try:
         ip = ""
-        async with session.get('http://ip-api.com/json/', proxy=proxy, timeout=5) as ipres:
+        async with session.get('http://ip.sb', proxy=proxy, timeout=5, headers={'user-agent': "curl"}) as ipres:
             if ipres.status == 200:
-                ipdata = await ipres.json()
-                ip = ipdata.get('query', '')
+                ip = await ipres.text()
+        if not ip:
+            async with session.get('http://ip-api.com/json/', proxy=proxy, timeout=5) as ipres:
+                if ipres.status == 200:
+                    ipdata = await ipres.json()
+                    ip = ipdata.get('query', '')
         if ip != '':
             url = baseurl + ip
         else: