digest.py 文件源码

def calculate_digest(data, alg):    
    '''
    Calculates digest according to algorithm
    '''
    digest_alg = None
    if (alg == SHA1_NAME):
        digest_alg = hashlib.sha1() 

    if (alg == SHA256_NAME):
        digest_alg = hashlib.sha256()

    if (alg == SHA384_NAME):
        digest_alg = hashlib.sha384()

    if (alg == SHA512_NAME):
        digest_alg = hashlib.sha512()

    if digest_alg is None:
        logger.error("Unknown digest algorithm : %s" % alg)
        return None

    digest_alg.update(data)   
    dg = digest_alg.digest()       

    logger.debug("Calculated hash from input data: %s" % base64.b64encode(dg))    
    return dg