import os
import dns.resolver
import requests

url = "http://example.com"
# 1. Parse the URL manually or using urllib.parse to get domain and path
parsed_url = url.split("://")[1].split("/", 1)
domain = parsed_url[0]
path = "/" + parsed_url[1]

# 2. Resolve IP using dnspython
resolver = dns.resolver.Resolver()
answers = resolver.resolve(domain, "A")
resolved_ip = answers[0].address  # Grab the first returned IP

# 3. Create the direct request overriding the URL with the IP
download_url = f"http://{resolved_ip}{path}"
headers = {
    "Host": domain  # Crucial: Tells the server which site you are actually asking for
}

# 4. Stream and save the file
response = requests.get(download_url, headers=headers, stream=True)

if response.status_code == 200:
    local_filename = url.split("/")[-1]
    with open(local_filename, "wb") as f:
        for chunk in response.iter_content(chunk_size=8192):
            if chunk:
                f.write(chunk)
    print(f"File downloaded successfully as {local_filename}")
else:
    print(f"Failed to download. Status Code: {response.status_code}")