Update crypt.py

crypt.py

@@ -296,8 +296,60 @@ def generate_keys():
     pub_key = conv_im("address_im.png",data=public_key)   
     
     return public_key,private_key,address_im,address,priv_key,pub_key
+########********************************###########################
 
+def encrypt_trans(data,pub_key):
+    #pub_key = decode(pub_im)
+    data = data.encode("utf-8")
+    recipient_key = RSA.import_key(pub_key)
+    session_key = get_random_bytes(16)
+    
+    # Encrypt the session key with the public RSA key
+    cipher_rsa = PKCS1_OAEP.new(recipient_key)
+    enc_session_key = cipher_rsa.encrypt(session_key)
+    
+    # Encrypt the data with the AES session key
+    cipher_aes = AES.new(session_key, AES.MODE_EAX)
+    ciphertext, tag = cipher_aes.encrypt_and_digest(data)
 
+    file_out = open("encrypted_data.bin", "wb")
+    
+    [ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ]
+    file_out.close()
+    doc_name = "encrypted_data.bin"
+    with open(doc_name, "rb") as file:
+        file_data =(file.read())
+    file.close()
+    return file_data
+
+
+def decrypt_trans(data,priv_im):
+    secret_code="SECRET PASSWORD"
+    priv_key = decode(priv_im)
+    print(f'priv_key:: {priv_key}')
+    key = RSA.import_key(priv_key,passphrase=secret_code)
+    
+    public_key = key.publickey().export_key('PEM')
+    file_out_pub = open("receiver.pem", "wb")
+    file_out_pub.write(public_key)
+    file_out_pub.close()
+
+
+    hash_1 = calculate_hash(public_key, hash_function="sha256")
+    hash_2 = calculate_hash(hash_1, hash_function="ripemd160")
+    address = base58.b58encode(hash_2)
+    address= str(address)
+    print (address)
+    address = address.strip("b")
+    print (address)
+    address = address.strip("'") if address.startswith("'") else address.strip('"')
+    print (address)
+
+    return (address)
+
+########********************************###########################
+
+    
 def encrypt_text(data,pub_im,priv_im,address):
     pub_key = decode(pub_im)
     data = data.encode("utf-8")