refactor: ubuntu lts compatHEADmain

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"