bot: encrypt access_key with cryptography.fernet library

1 files changed, 18 insertions, 0 deletions

diff --git a/util.py b/util.py
new file mode 100644
index 0000000..61efa0b
--- /dev/null
+++ b/util.py
@@ -0,0 +1,18 @@
+from cryptography.fernet import Fernet
+
+
+def encrypt(input: str, key: str) -> str:
+    f = Fernet(key)
+    return f.encrypt(bytes(input, 'utf-8')).decode('utf-8')
+
+
+def decrypt(input: str, key: str) -> str:
+    f = Fernet(key)
+    return f.decrypt(input).decode('utf-8')
+
+
+# if __name__ == "__main__":
+#     key = Fernet.generate_key().decode('utf-8')
+#     print(key)
+#     print(encrypt("Hello World!", key))
+#     print(decrypt(encrypt("Hello World!", key), key))