mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-25 20:33:58 +01:00
37 lines
1 KiB
Python
37 lines
1 KiB
Python
import base64
|
|
import asyncio
|
|
|
|
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
|
|
|
|
async def add_proxy_auth_to_headers(username: str, password: str, headers: dict) -> dict:
|
|
"""Add proxy authentication to the headers. This is useful if the proxy authentication doesn't work as it should."""
|
|
|
|
proxy_auth = base64.b64encode(f'{username}:{password}'.encode()).decode()
|
|
headers['Proxy-Authorization'] = f'Basic {proxy_auth}'
|
|
return headers
|
|
|
|
if __name__ == '__main__':
|
|
print(asyncio.run(add_proxy_auth_to_headers(
|
|
'user',
|
|
'pass',
|
|
{
|
|
'Authorization': 'Bearer demo',
|
|
'Another-Header': '123'
|
|
}
|
|
)))
|