kinit/kinit-api/utils/ip_manage.py
ktianc 7fbdcb3b0f 版本升级
1. 新增:微信小程序端新增微信手机号登录功能(必须为企业认证小程序)
2. 新增:加入动态更新常见问题
3. 新增:新增小程序分享功能
4. 新增:小程序新增第一次登录需要修改密码
5. 新增:新增接口权限控制
6. 新增:用户新增is_staff用来判断是否为工作人员
7. 新增:软删除新增is_delete字段来判断,delete_datetime当前主要来记录时间
8. 更新:部分接口删除功能已更新,需要试用软删除的才会试用软删除
9. 更新:更新系统配置缓存功能
10. 更新:接口认证依赖项更新
11. 更新:获取系统基础配置信息与用户协议与隐私协议更新
12. 优化:优化接口与数据库操作
2023-02-27 17:28:27 +08:00

72 lines
2.4 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @version : 1.0
# @Creaet 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: Optional[str] = None
address: Optional[str] = None
country: Optional[str] = None
province: Optional[str] = None
city: Optional[str] = None
county: Optional[str] = None
operator: Optional[str] = None
postal_code: Optional[str] = None
area_code: Optional[str] = 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