diff options
author | AL-LCL <alvin@alvinhavel.com> | 2023-05-19 11:06:25 +0200 |
---|---|---|
committer | AL-LCL <alvin@alvinhavel.com> | 2023-05-19 11:06:25 +0200 |
commit | 18a3d3bc354e667bc58385e59745b82b53695139 (patch) | |
tree | fef56722868ae768b16382aae2f3559ec4bbf370 /Specific/encrypt.py |
Diffstat (limited to 'Specific/encrypt.py')
-rw-r--r-- | Specific/encrypt.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/Specific/encrypt.py b/Specific/encrypt.py new file mode 100644 index 0000000..6f88280 --- /dev/null +++ b/Specific/encrypt.py @@ -0,0 +1,26 @@ +import base64 + +from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC +from cryptography.hazmat.backends import default_backend +from cryptography.hazmat.primitives import hashes +from cryptography.fernet import Fernet + + +class Encryption: + def __init__(self, password='ksxgyRuBRJLKxjFeHD4nmxbE', salt=b'v4CuHZFzmTedBY2EBGrLRXsm'): + self.password = password + self.salt = salt + self.key = Fernet(self.generate_key()) + + + def generate_key(self): + kdf = PBKDF2HMAC(algorithm=hashes.SHA256(), length=32, salt=self.salt, iterations=100000, backend=default_backend()) + return base64.urlsafe_b64encode(kdf.derive(self.password.encode())) + + + def do_encrypt(self, message): + return self.key.encrypt(message) + + + def do_decrypt(self, ciphertext): + return self.key.decrypt(ciphertext)
\ No newline at end of file |