mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-22 21:06:32 +00:00
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import re
|
|
import requests
|
|
import httpx
|
|
|
|
from app.core.logger import logger
|
|
|
|
|
|
class IpLocalUtil:
|
|
"""
|
|
获取IP归属地工具类
|
|
"""
|
|
|
|
@classmethod
|
|
def is_valid_ip(cls, ip: str) -> bool:
|
|
"""
|
|
校验IP格式是否合法
|
|
|
|
:param ip: IP地址
|
|
:return: 是否合法
|
|
"""
|
|
ip_pattern = r'^((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$'
|
|
return bool(re.match(ip_pattern, ip))
|
|
|
|
@classmethod
|
|
async def get_ip_location(cls, ip: str) -> str:
|
|
"""
|
|
获取IP归属地信息
|
|
|
|
:param ip: IP地址
|
|
:return: IP归属地信息
|
|
"""
|
|
# 校验IP格式
|
|
if not cls.is_valid_ip(ip):
|
|
logger.error(f"IP格式不合法: {ip}")
|
|
return "未知"
|
|
|
|
logger.info(f"获取IP归属地: {ip}, 类型: {type(ip)}")
|
|
|
|
# 内网IP直接返回
|
|
if ip == '127.0.0.1' or ip == 'localhost':
|
|
return '内网IP'
|
|
|
|
try:
|
|
# 百度API失败，使用其他API
|
|
# async with httpx.AsyncClient() as client:
|
|
# response = await client.get(
|
|
# f'https://qifu-api.baidubce.com/ip/geo/v1/district?ip={ip}',
|
|
# timeout=5
|
|
# )
|
|
# if response.status_code == 200:
|
|
# data = response.json().get('data', {})
|
|
# return f"【{data.get('owner','')}】-{data.get('country','')}-{data.get('prov','')}-{data.get('city','')}-{data.get('district','')}"
|
|
|
|
# 使用ip-api.com API获取IP归属地信息
|
|
async with httpx.AsyncClient() as client:
|
|
response = await client.get(
|
|
f'http://ip-api.com/json/{ip}?lang=zh-CN',
|
|
timeout=10
|
|
)
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
return f"{result.get('country','')}-{result.get('regionName','')}-{result.get('city','')}"
|
|
|
|
except Exception as e:
|
|
logger.error(f"获取IP归属地失败: {e}")
|
|
return "未知"
|