#!/usr/bin/python

import sys
from hashlib import sha1
from urllib2 import urlopen, Request
from getpass import getpass

if __name__ == "__main__":
    p1 = getpass('password: ')
    p2 = getpass('password again: ')

    if p1 != p2:
        print "Passwords don't match"
        sys.exit(1)

    h = sha1(p1).hexdigest()
    prefix, suffix = h[:5], h[5:]

    for line in urlopen(Request('https://api.pwnedpasswords.com/range/%s' % (prefix,), headers={'User-Agent': 'curl/7.55.1'})):
        s, c = line.strip().split(':')
        if s.lower() == suffix:
            print 'Found with count %d' % (int(c),)
            break
    else:
        print 'Not found'