nova-api/api/helpers/network.py

import os
import time
from dotenv import load_dotenv
from slowapi.util import get_remote_address

load_dotenv()

async def get_ip(request) -> str:
    """Get the IP address of the incoming request."""

    xff = None
    if request.headers.get('x-forwarded-for'):
        xff, *_ = request.headers['x-forwarded-for'].split(', ')

    possible_ips = [
        xff,
        request.headers.get('cf-connecting-ip'),
        request.client.host
    ]

    detected_ip = next((i for i in possible_ips if i), None)

    return detected_ip

def get_ratelimit_key(request) -> str:
    """Get the IP address of the incoming request."""
    custom = os.environ('NO_RATELIMIT_IPS')
    ip = get_remote_address(request)

    if ip in custom:
        return f'enterprise_{ip}'

    return ip