from waybackpy import WaybackMachineSaveAPI
import time
import random

from config import ARCHIVE_URLS, USER_AGENT

def archive_url(url):
    print(f"🌐 Archiving {url} ...")
    try:
        save_api = WaybackMachineSaveAPI(url, user_agent=USER_AGENT)
        save_api.save()
        print(f"✅ Archived {url}")
    except Exception as e:
        print(f"⚠️ Exception archiving {url}: {e}")

def archive_all_urls():
    for url in ARCHIVE_URLS:
        delay = 10 + random.uniform(-3, 3)
        time.sleep(delay)
        archive_url(url)

def test_archive():
    test_url = "https://httpbin.org/anything/foo/bar"
    archive_url(test_url)