aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xsrc/mojicrypt12
1 files changed, 8 insertions, 4 deletions
diff --git a/src/mojicrypt b/src/mojicrypt
index 75b4e6b..4f5c048 100755
--- a/src/mojicrypt
+++ b/src/mojicrypt
@@ -11,10 +11,14 @@ import secrets
import time
import uuid
-# requires pycryptodome
-from Crypto.Cipher import AES
-from Crypto.Protocol.KDF import scrypt
-from Crypto.Random import get_random_bytes
+try:
+ from Crypto.Cipher import AES
+ from Crypto.Protocol.KDF import scrypt
+ from Crypto.Random import get_random_bytes
+except ModuleNotFoundError:
+ from Cryptodome.Cipher import AES
+ from Cryptodome.Protocol.KDF import scrypt
+ from Cryptodome.Random import get_random_bytes
APP_NAME = "mojicrypt"
VERSION = "20260303"