用python编程密码学Diffie-Hellman算法，在计算会话密钥时需要包含数据加密部分调用密码库的代码是什么呀？

以下是一个使用Python编写Diffie-Hellman算法，并在计算出会话密钥后使用cryptography库进行数据加密的示例代码：

from cryptography.hazmat.primitives.asymmetric import dh
from cryptography.hazmat.primitives import serialization, hashes
from cryptography.hazmat.primitives.ciphers.aead import AESGCM

# 生成 Diffie-Hellman 密钥对
parameters = dh.generate_parameters(generator=2, key_size=2048)
private_key = parameters.generate_private_key()
public_key = private_key.public_key()

# 将公钥序列化为字节串，并发送给对方
public_key_bytes = public_key.public_bytes(
    encoding=serialization.Encoding.PEM,
    format=serialization.PublicFormat.SubjectPublicKeyInfo
)
# 接收对方的公钥，并反序列化为 PublicKey 对象
peer_public_key = serialization.load_pem_public_key(
    peer_public_key_bytes,
    backend=default_backend()
)

# 计算会话密钥
shared_key = private_key.exchange(peer_public_key)
session_key = hashes.Hash(hashes.SHA256(), backend=default_backend())
session_key.update(shared_key)
session_key = session_key.finalize()

# 使用会话密钥加密数据
plaintext = b"Hello, world!"
aad = b"authenticated but unencrypted data"
nonce = b"\x00" * 12  # 必须是 12 字节长的随机数
cipher = AESGCM(session_key)
ciphertext = cipher.encrypt(nonce, plaintext, aad)

需要注意的是，上述代码中仅给出了加密数据的部分，实际使用时，还需要考虑如何将密文传输给对方、如何进行解密等问题。