mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-20 20:39:55 +00:00
zhangtao
c2ca6d19ac
style: 统一代码风格和格式 docs: 完善函数和方法的文档字符串 refactor(base_model): 移除冗余的表名和表参数生成方法 refactor(constant): 更新返回码注释格式 refactor(router_class): 添加路由处理器的详细文档 refactor(database): 完善数据库连接函数的文档 refactor(security): 添加认证类和方法的详细文档 refactor(validator): 更新验证器函数的文档格式 refactor(serialize): 优化序列化工具类的文档 refactor(response): 完善响应类的文档字符串 refactor(dependencies): 添加依赖函数的详细文档 refactor(initialize): 完善初始化脚本的文档 refactor(plugin): 添加生命周期和中间件注册的文档 refactor(service): 完善服务层方法的文档 refactor(controller): 添加控制器方法的详细文档 refactor(crud): 完善CRUD操作的文档字符串 refactor(schema): 简化模型类并移除冗余字段 refactor(param): 更新查询参数类的注释格式 refactor(template): 优化代码生成模板的格式 refactor(console): 添加控制台输出功能的实现 refactor(util): 完善工具函数的文档字符串
107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import re
|
|
import httpx
|
|
|
|
from app.core.logger import logger
|
|
|
|
|
|
class IpLocalUtil:
|
|
"""
|
|
获取IP归属地工具类
|
|
"""
|
|
|
|
@classmethod
|
|
def is_valid_ip(cls, ip: str) -> bool:
|
|
"""
|
|
校验IP格式是否合法。
|
|
|
|
参数:
|
|
- ip (str): IP地址。
|
|
|
|
返回:
|
|
- bool: 是否合法。
|
|
"""
|
|
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
|
|
def is_private_ip(cls, ip: str) -> bool:
|
|
"""
|
|
判断是否为内网IP。
|
|
|
|
参数:
|
|
- ip (str): IP地址。
|
|
|
|
返回:
|
|
- bool: 是否为内网IP。
|
|
"""
|
|
priv_pattern = r'^(10\.|172\.(1[6-9]|2[0-9]|3[01])\.|192\.168\.)'
|
|
return bool(re.match(priv_pattern, ip))
|
|
|
|
@classmethod
|
|
async def get_ip_location(cls, ip: str) -> str | None:
|
|
"""
|
|
获取IP归属地信息。
|
|
|
|
参数:
|
|
- ip (str): IP地址。
|
|
|
|
返回:
|
|
- str | None: IP归属地信息，失败时返回"未知"或None。
|
|
"""
|
|
# 校验IP格式
|
|
if not cls.is_valid_ip(ip):
|
|
logger.error(f"IP格式不合法: {ip}")
|
|
return "未知"
|
|
|
|
# 内网IP直接返回
|
|
if cls.is_private_ip(ip):
|
|
return '内网IP'
|
|
|
|
try:
|
|
# 使用ip-api.com API获取IP归属地信息
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
# 尝试使用 ip9.com.cn API
|
|
url = f'https://ip9.com.cn/get?ip={ip}'
|
|
response = await cls._make_api_request(client, url)
|
|
if response and response.json().get('ret') == 200:
|
|
result = response.json().get('data', {})
|
|
return f"{result.get('country','')}-{result.get('prov','')}-{result.get('city','')}-{result.get('area','')}-{result.get('isp','')}"
|
|
|
|
# 尝试使用百度 API
|
|
url = f'https://qifu-api.baidubce.com/ip/geo/v1/district?ip={ip}'
|
|
response = await cls._make_api_request(client, url)
|
|
if response and response.json().get('code') == "Success":
|
|
data = response.json().get('data', {})
|
|
# 修正原代码中的格式错误
|
|
return f"{data.get('country','')}-{data.get('prov','')}-{data.get('city','')}-{data.get('district','')}-{data.get('isp','')}"
|
|
|
|
except Exception as e:
|
|
logger.error(f"获取IP归属地失败: {e}")
|
|
return "未知"
|
|
|
|
@classmethod
|
|
async def _make_api_request(cls, client, url):
|
|
"""
|
|
单独的 API 请求方法，包含重试机制。
|
|
|
|
参数:
|
|
- client (AsyncClient): httpx 异步客户端。
|
|
- url (str): 请求 URL。
|
|
|
|
返回:
|
|
- Response | None: 响应对象，失败时返回None。
|
|
"""
|
|
max_retries = 3
|
|
for attempt in range(max_retries):
|
|
try:
|
|
response = await client.get(url, timeout=10)
|
|
if response.status_code == 200:
|
|
return response
|
|
except Exception as e:
|
|
if attempt < max_retries - 1:
|
|
continue
|
|
logger.error(f"请求 {url} 失败: {e}")
|
|
return None
|