python Diffie-Hellman A选择的数a自动获取，B选择的数b为学号08,输入素数，输出A，B的私钥，以及输出A，B的计算数，最后输出A，B的会话密钥

好的，那我来为您重新编写一份代码，实现您的需求。

import random

# 生成素数
def generate_prime():
    while True:
        p = int(input("请输入一个素数："))
        for i in range(2, int(p ** 0.5) + 1):
            if p % i == 0:
                break
        else:
            return p

# A和B选择私钥
a = random.randint(1, 100)
b = 8

# 生成素数
p = generate_prime()

# 计算A和B的计算数
A = pow(2, a, p)
B = pow(2, b, p)

# 输出A和B的私钥和计算数
print("A的私钥为：", a)
print("B的私钥为：", b)
print("A的计算数为：", A)
print("B的计算数为：", B)

# 计算会话密钥
K1 = pow(B, a, p)
K2 = pow(A, b, p)

# 输出会话密钥
print("会话密钥为：", K1)

这份代码与之前的代码唯一的区别在于，输出了A和B的私钥，而不是只输出了A的私钥。同时，在计算出会话密钥后，代码也直接输出了会话密钥。

当您输入素数后，代码会输出A和B的私钥和计算数，最后计算并输出会话密钥。

注意：虽然这份代码中私钥是随机生成的，但在实际应用中，为了保证安全性，应该使用更加安全的随机数生成方式。另外，在实际应用中，私钥应该保密，不能直接输出。