Update app.py

app.py

@@ -21,6 +21,7 @@ import qr
 import stegan
 import json
 import re
+import os
 
 
 
@@ -38,48 +39,29 @@ def rsa_encrypt_eth():
 
 
 def rsa_generate():
-
-    keyPair = RSA.generate(3072)
-    print(dir(keyPair))
+    key_pair = RSA.generate(1024)
     
-    pubKey = keyPair.publickey()
-    print(dir(pubKey))
-    print(f"Public key:  (n={hex(pubKey.n)}, e={hex(pubKey.e)})")
-    pubKeyPEM = pubKey.exportKey()
-    print(pubKeyPEM.decode('ascii'))
-    print(f"Private key: (n={hex(pubKey.n)}, d={hex(keyPair.d)})")
-    privKeyPEM = keyPair.exportKey()
-    print(privKeyPEM.decode('ascii'))
-    output = re.search('[0-9a-fA-F]{32}', str(pubKeyPEM))
+    public_key = key_pair.publickey().exportKey()
+    private_key = key_pair.exportKey()
+    output = re.search('[0-9a-fA-F]{32}', str(public_key))
     qr_wal=qrx.make(output)
     qr_wal.save("qr_wal_tmp.png")
-    #privcipher = PKCS1_OAEP.new(keyPair)
-    #pubcipher = PKCS1_OAEP.new(pubKey)
-    out1 = f'{pubKey.n}{keyPair.d}'
-    out2 = f'{pubKey.n}{keyPair.e}'
-    return privKeyPEM,pubKeyPEM,output,"qr_wal_tmp.png",output
 
+    return private_key,public_key,output,"qr_wal_tmp.png",output
 
 
 
-    
-def rsa_encrypt(msg,pubKey):
-    pubKey = pubKey.strip('"')
-    key = b64decode(pubKey)
-    private_key = RSA.importKey(key)
-    #cipher = PKCS1_v1_5.new(private_key)
-    cipher = PKCS1_OAEP.new(private_key)
 
-    cipher.encrypt(msg)
+    
+def rsa_encrypt(msg,public_key):
+    key = RSA.importKey(public_key)
 
+    message = msg.encode()
+    print(message)
     
-    #pubKey=pubKey.encode("utf-8")
-    print (pubKey)
-    msg = b'A message for encryption'
-    encryptor = PKCS1_OAEP.new(cipher)
-    encrypted = encryptor.encrypt(msg)
-    print("Encrypted:", binascii.hexlify(encrypted))
-    return binascii.hexlify(encrypted)
+    cipher = PKCS1_OAEP.new(key)
+    ciphertext = cipher.encrypt(message)
+    return (ciphertext)
     
 def rsa_decrypt(encrypted,keyPair):
     decryptor = PKCS1_OAEP.new(keyPair)