Add ip location cache

[downdawn]

No description provided.

[downdawn]

PLR.

[wu-clan]

please update.

[downdawn]

@sync_to_async
def get_location_offline(ip: str) -> dict | None:
    """
    离线获取 ip 地址属地，无法保证准确率，100%可用

    :param ip:
    :return:
    """
    try:
        cb = XdbSearcher.loadContentFromFile(dbfile=IP2REGION_XDB)
        searcher = XdbSearcher(contentBuff=cb)
        data = searcher.search(ip)
        searcher.close()
        data = data.split('|')
        return {'country': data[0], 'regionName': data[2], 'city': data[3]}
    except Exception as e:
        log.error(f'离线获取 ip 地址属地失败，错误信息：{e}')
        return None


async def parse_ip_info(request: Request) -> tuple[str, str, str, str]:
    country, region, city = None, None, None
    ip = await get_request_ip(request)
    location = await redis_client.get(f'{settings.IP_LOCATION_REDIS_PREFIX}:{ip}')
    if location:
        country, region, city = location.split(' ')
        return ip, country, region, city
    if settings.LOCATION_PARSE == 'online':
        location_info = await get_location_online(ip, request.headers.get('User-Agent'))
    elif settings.LOCATION_PARSE == 'offline':
        location_info = await get_location_offline(ip)
    else:
        location_info = None
    if location_info:
        country = location_info.get('country')
        region = location_info.get('regionName')
        city = location_info.get('city')
    await redis_client.set(f'{settings.IP_LOCATION_REDIS_PREFIX}:{ip}', f'{country} {region} {city}',
                           ex=settings.IP_LOCATION_EXPIRE_SECONDS)
    return ip, country, region, city

[wu-clan]

Is judgment being passed here?

# return {'country': data[0], 'regionName': data[2], 'city': data[3]}

country = location_info[0] if location_info[0] != '0' else None
region = location_info[2] if location_info[2] != '0' else None
city = location_info[3] if location_info[3] != '0' else None
return {'country': country, 'regionName': region, 'city': city}

Also, await redis_client.set() should be in if location_info:.

[downdawn]

Update.

[wu-clan]

LGTM.