72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Create Time : 2022/11/3 17:23
|
||
# @File : ip_manage.py
|
||
# @IDE : PyCharm
|
||
# @desc : 获取IP地址归属地
|
||
|
||
"""
|
||
文档：https://user.ip138.com/ip/doc
|
||
IP查询第三方服务，有1000次的免费次数
|
||
|
||
JSONP请求示例（IPv4）
|
||
https://api.ip138.com/ip/?ip=58.16.180.3&datatype=jsonp&token=cc87f3c77747bccbaaee35006da1ebb65e0bad57
|
||
|
||
aiohttp 异步请求文档：https://docs.aiohttp.org/en/stable/client_quickstart.html
|
||
"""
|
||
from aiohttp import TCPConnector
|
||
|
||
from application.settings import IP_PARSE_TOKEN, IP_PARSE_ENABLE
|
||
import aiohttp
|
||
from core.logger import logger
|
||
from pydantic import BaseModel
|
||
from typing import Optional
|
||
|
||
|
||
class IPLocationOut(BaseModel):
|
||
ip: str | None = None
|
||
address: str | None = None
|
||
country: str | None = None
|
||
province: str | None = None
|
||
city: str | None = None
|
||
county: str | None = None
|
||
operator: str | None = None
|
||
postal_code: str | None = None
|
||
area_code: str | None = None
|
||
|
||
|
||
class IPManage:
|
||
|
||
def __init__(self, ip: str):
|
||
self.ip = ip
|
||
self.url = f"https://api.ip138.com/ip/?ip={ip}&datatype=jsonp&token={IP_PARSE_TOKEN}"
|
||
|
||
async def parse(self):
|
||
"""
|
||
IP 数据解析
|
||
|
||
接口返回：{'ret': 'ok', 'ip': '114.222.121.253','data': ['中国', '江苏', '南京', '江宁区', '电信', '211100', '025']}
|
||
"""
|
||
out = IPLocationOut()
|
||
out.ip = self.ip
|
||
if not IP_PARSE_ENABLE:
|
||
logger.warning("未开启IP地址数据解析，无法获取到IP所属地，请在application/config/production.py:IP_PARSE_ENABLE中开启！")
|
||
return out
|
||
async with aiohttp.ClientSession(connector=TCPConnector(ssl=False)) as session:
|
||
async with session.get(self.url) as resp:
|
||
body = await resp.json()
|
||
if body.get("ret") != 'ok':
|
||
logger.error(f"获取IP所属地失败：{body}")
|
||
return out
|
||
data = body.get("data")
|
||
out.address = f"{''.join(data[i] for i in range(0, 4))} {data[4]}"
|
||
out.country = data[0]
|
||
out.province = data[1]
|
||
out.city = data[2]
|
||
out.county = data[3]
|
||
out.operator = data[4]
|
||
out.postal_code = data[5]
|
||
out.area_code = data[6]
|
||
return out
|