Криптосистема Рабина

import random
 
def q():
    #n = int(input('Введите n:'))
    running = True
    while running:
        d = [random.randint(2, 2 ** 10) for i in range(5)]
        #print(d)
    
        a = 1
        for i in d:
            a *=i
 
        global q
        q = a +1
        
        i = 2
        j = 0  # флаг
        while i ** 2 <= q and j != 1:
            if q % i == 0:
                j = 1
            i += 1
        if j == 1:
            #print(q, "- Составное")
            continue
        else:
            print(q, "- секретный ключ q")
            running = False
 
def p():
    #n = int(input('Введите n:'))
    running = True
    while running:
        d = [random.randint(2, 2 ** 10) for i in range(5)]
        #print(d)
    
        a = 1
        for i in d:
            a *=i
        global p
        p = a + 1
       
        i = 2
        j = 0  # флаг
        while i ** 2 <= p and j != 1:
            if p % i == 0:
                j = 1
            i += 1
        if j == 1:
            #print(p, "- Составное")
            continue
        else:
            print(p, "- секретный ключ p")
            running = False
 
 
q()
p()
 
n = q*p
print("Открытый ключ n -", n)
 
#Создаем сообщение М<n
M = int(input('Введите M:'))
print("Исходное сообщение:", M)
 
#Шифрование
C = pow(M, 2, n)
print('зашифрованное сообщение:', C)
 
#Расшифровка
m1 = pow(C, (p+1)//4, p)
m2 = -pow(C, (p+1)//4, p)
m3 = pow(C, (q+1)//4, q)
m4 = -pow(C, (q+1)//4, q)
print(m1, m2, m3 , m4)
 
#вычислим a и b
a = q * pow(q, p-2, p) 
b = p * pow(p, q-2, q)
print("a=", a, "b=", b)
 
#получаем 4 текста  
M1 = (a * m1 + b * m3) % n
M2 = (a * m1 + b * m4) % n
M3 = (a * m2 + b * m3) % n
M4 = (a * m2 + b * m4) % n
print(M1, M2, M3, M4)

M1 = (a * m1 + b * m3) % n
M2 = (a * m1 + b * m4) % n
M3 = (a * m2 + b * m3) % n
M4 = (a * m2 + b * m4) % n
print(M1, M2, M3, M4)