mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-25 21:33:58 +01:00
33 lines
778 B
Python
33 lines
778 B
Python
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 |