remove Crypto.Cipher

xZetsubou · Jan 18, 2024 · 4afac6c · 4afac6c
1 parent 6d23f9d
commit 4afac6c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/custom_components/localtuya/discovery.py b/custom_components/localtuya/discovery.py
@@ -13,7 +13,6 @@
 
 from cryptography.hazmat.backends import default_backend
 from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
-from Crypto.Cipher import AES
 
 from .common import pytuya
 
@@ -32,7 +31,9 @@ def decrypt(msg, key):
     def _unpad(data):
         return data[: -ord(data[len(data) - 1 :])]
 
-    return _unpad(AES.new(key, AES.MODE_ECB).decrypt(msg)).decode()
+    cipher = Cipher(algorithms.AES(key), modes.ECB(), default_backend())
+    decryptor = cipher.decryptor()
+    return _unpad(decryptor.update(msg) + decryptor.finalize()).decode()
 
 
 def decrypt_udp(message):