from sys import argv
from time import sleep
import requests

if len(argv) != 3:
    print(argv[0] + " emails.txt output_folder")

with open(argv[1], "r") as f:
    for email in f:
        email = email.strip()
        while True:
            url = "https://haveibeenpwned.com/api/v2/breachedaccount/" + email
            r = requests.get(url)

            if r.status_code == 429:
                if "Retry-After" in r.headers:
                    sleep(int(r.headers["Retry-After"]))
                else:
                    sleep(5)
            elif r.status_code == 404:
                print(email + " not pwned")
                break
            elif r.status_code == 200:
                print(email + " pwned")
                with open(argv[2] + "/" + email, "w") as o:
                    o.write(r.text)

                break
            else:
                print(email + " failed")
                break