diffie hellman algorithm in python code example

Example: python diffie hellman

import pyDH
d1 = pyDH.DiffieHellman()
d2 = pyDH.DiffieHellman()
d1_pubkey = d1.gen_public_key()
d2_pubkey = d2.gen_public_key()
d1_sharedkey = d1.gen_shared_key(d2_pubkey)
d2_sharedkey = d2.gen_shared_key(d1_pubkey)

print(d1_sharedkey == d2_sharedkey)

print(d1_pubkey,'\n',d2_pubkey)
print(d1_sharedkey,' and ',d2_sharedkey)