Python wrapper for the Have I Been Pwned API

pwnd.py

def check_password(password):
    """
    Returns how many times password appears in the breach data set.
    >>> check_password('qwerty123')
    '592110'
    >>> check_password('Very$ecurePa$$word')
    0
    """
    import requests
    from hashlib import sha1
    hashed = sha1(password.encode()).hexdigest().upper()
    prefix = hashed[:5]
    r = requests.get('https://api.pwnedpasswords.com/range/' + prefix)
    for line in r.text.split('\r\n'):
        suffix, count = line.split(':')
        if hashed[5:] == suffix:
            return count
    return 0