1#  Copyright (C) 2012-2016 Christian Heimes (christian@python.org)
2#  Licensed to PSF under a Contributor Agreement.
3#
4
5# monkey patch _hashlib
6import hashlib as _hashlib
7
8from _pysha3 import keccak_224, keccak_256, keccak_384, keccak_512
9from _pysha3 import sha3_224, sha3_256, sha3_384, sha3_512
10from _pysha3 import shake_128, shake_256
11
12
13__all__ = ("sha3_224", "sha3_256", "sha3_384", "sha3_512",
14           "keccak_224", "keccak_256", "keccak_384", "keccak_512",
15           "shake_128", "shake_256")
16
17
18if not hasattr(_hashlib, "sha3_512"):
19    _hashlib.sha3_224 = sha3_224
20    _hashlib.sha3_256 = sha3_256
21    _hashlib.sha3_384 = sha3_384
22    _hashlib.sha3_512 = sha3_512
23    _hashlib.shake_128 = shake_128
24    _hashlib.shake_256 = shake_256
25