Support both pycryptodome and pycryptodomex

2017-07-29 19:20:43 +09:00
parent 1bff584465
commit 4ad9428dfa
3 changed files with 20 additions and 7 deletions
--- a/pywii/pywii-tools/tmdupdatecr.py
+++ b/pywii/pywii-tools/tmdupdatecr.py
@@ -2,7 +2,10 @@

 import sys, os, os.path
 import pywii as wii
-from Cryptodome.Hash import SHA
+try:
+    from Cryptodome.Hash import SHA
+except ImportError:
+    from Crypto.Hash import SHA

 wii.loadkeys()