dh.py

# Prime number and generator for Diffie-Hellman
p = 23
g = 5

# Alice and Bob generate their own secret keys
a = 6
b = 15

# Alice and Bob calculate their public keys
A = (g ** a) % p
B = (g ** b) % p

# Alice and Bob exchange their public keys
# (this could be done over an insecure channel)
# In this example, we'll just assign the values manually
A_received = B
B_received = A

# Alice and Bob calculate the shared secret key
alice_shared_secret = (B_received ** a) % p
bob_shared_secret = (A_received ** b) % p

# Both shared secrets should be the same
print("Shared secret calculated by Alice: ", alice_shared_secret)
print("Shared secret calculated by Bob: ", bob_shared_secret)