mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-23 13:33:08 +00:00
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
import httpx
|
|
from XdbSearchIP.xdbSearcher import XdbSearcher
|
|
from asgiref.sync import sync_to_async
|
|
from httpx import HTTPError
|
|
from fastapi import Request
|
|
|
|
from backend.app.core.path_conf import IP2REGION_XDB
|
|
|
|
|
|
@sync_to_async
|
|
def get_request_ip(request: Request) -> str:
|
|
"""获取请求的 ip 地址"""
|
|
real = request.headers.get('X-Real-IP')
|
|
if real:
|
|
ip = real
|
|
else:
|
|
forwarded = request.headers.get('X-Forwarded-For')
|
|
if forwarded:
|
|
ip = forwarded.split(',')[0]
|
|
else:
|
|
ip = request.client.host
|
|
return ip
|
|
|
|
|
|
async def get_location_online(ipaddr: str, user_agent: str) -> str:
|
|
"""
|
|
在线获取 ip 地址属地，无法保证可用性，准确率较高
|
|
|
|
:param ipaddr:
|
|
:param user_agent:
|
|
:return:
|
|
"""
|
|
async with httpx.AsyncClient(timeout=3) as client:
|
|
ip_api_url = f'http://ip-api.com/json/{ipaddr}?lang=zh-CN'
|
|
whois_url = f'http://whois.pconline.com.cn/ipJson.jsp?ip={ipaddr}&json=true'
|
|
headers = {'User-Agent': user_agent}
|
|
try:
|
|
resp1 = await client.get(ip_api_url, headers=headers)
|
|
city = resp1.json()['city']
|
|
except (HTTPError, KeyError):
|
|
try:
|
|
resp2 = await client.get(whois_url, headers=headers)
|
|
city = resp2.json()['city']
|
|
except (HTTPError, KeyError):
|
|
city = None
|
|
return city or '未知' if city != '' else '未知'
|
|
|
|
|
|
@sync_to_async
|
|
def get_location_offline(ipaddr: str) -> str:
|
|
"""
|
|
离线获取 ip 地址属地，无法保证准确率，100%可用
|
|
|
|
:param ipaddr:
|
|
:return:
|
|
"""
|
|
cb = XdbSearcher.loadContentFromFile(dbfile=IP2REGION_XDB)
|
|
searcher = XdbSearcher(contentBuff=cb)
|
|
data = searcher.search(ipaddr)
|
|
searcher.close()
|
|
location_info = data.split('|')
|
|
country = location_info[0]
|
|
province = location_info[2]
|
|
city = location_info[3]
|
|
return city if city != '0' else province if province != '0' else country if country != '0' else '未知'
|