Create encrpyt.py

lwhile · lwhile · commit aa0530ee2fd9 · 2016-01-29T23:47:46.000+08:00
diff --git a/lwh/21/encrpyt.py b/lwh/21/encrpyt.py
@@ -0,0 +1,24 @@
+from hashlib import sha256
+from hmac import HMAC
+import os
+
+
+class Encrypt(object):
+
+    def encrypt(self, password, salt=None):
+        if salt is None:
+            salt = os.urandom(8)
+        result = password.encode('utf-8')
+        for i in range(10):
+            result = HMAC(result, salt, sha256).digest()
+        return salt + result
+
+    def vaildate(self, password, hashed):
+        return hashed == self.encrypt(password, salt=hashed[:8])
+
+if __name__ == '__main__':
+    obj = Encrypt()
+    hashed = obj.encrypt('wh5622')
+    # print(bytes.decode(hashed))
+    ans = obj.vaildate('wh5622', hashed)
+    print(ans)
