فهرست منبع

fix(hugo-encryptor.py): fix the bug when encrypt Chinese

Li4n0 6 سال پیش
والد
کامیت
fb330deebb
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  1. 2 2
      hugo-encryptor.py

+ 2 - 2
hugo-encryptor.py

@@ -17,7 +17,7 @@ class AESCrypt(object):
         cryptor = AES.new(self.key, self.mode, self.key[16:])
         padlen = AESCrypt.LEN - len(text) % AESCrypt.LEN
         padlen = padlen if padlen != 0 else AESCrypt.LEN
-        text += chr(padlen)*padlen
+        text += (chr(padlen)*padlen).encode('utf8')
 
         return cryptor.encrypt(text)
 
@@ -43,7 +43,7 @@ if __name__ == '__main__':
                 key = md5.hexdigest()
                 cryptor = AESCrypt(key)
                 text = ''.join(map(str, block.contents))
-                written = base64.b64encode(cryptor.encrypt(text))
+                written = base64.b64encode(cryptor.encrypt(text.encode('utf8')))
 
                 del block['data-password']
                 block.string = written.decode()