fix: support both cryptodome namespaces

volexity · Jan 24, 2025 · c364aba · c364aba
1 parent 96075ef
commit c364aba
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -85,6 +85,7 @@ source = "vcs"
 
 [[tool.mypy.overrides]]
 module = [
+  "Cryptodome.*",
   "olefile.*",
   "pytest.*",
 ]

diff --git a/src/hwp_extract/encrypt.py b/src/hwp_extract/encrypt.py
@@ -5,7 +5,10 @@
 
 import hashlib
 
-from Crypto.Cipher import AES
+try:
+    from Cryptodome.Cipher import AES
+except ModuleNotFoundError:
+    from Crypto.Cipher import AES
 
 
 def pad(s: bytes) -> bytes: