updated crypto

app.py

@@ -13,6 +13,7 @@ import random
 import string
 import json
 import base64
+import crypto
 
 auth = [
     (os.getenv("USERNAME1"), os.getenv("PASSWORD1")),
@@ -346,9 +347,7 @@ def handle_url(url, From_volume: int, To_Volume: int, Gmail: str, FolderId: str,
     details = []
     final_output = None
     link = None
-    credit = os.getenv("GOOGLE_AUTH_CREDENTIALS") # Get the google auth from the env
-    credit = base64.b64decode(credit).decode()
-    print(credit)
+    credit = crypto.decrypt(os.getenv("KEY"), os.getenv("GOOGLE_AUTH_CREDENTIALS"))
     credit = json.loads(credit)
 
     for key, result in enumerate(filterUrlandRun(url, From_volume, To_Volume, Reverse, Output, MakeOwner, Gmail, FolderId, credit=credit, usegooglesheet=UseGoogleSheet)):

crypto.py

@@ -0,0 +1,30 @@
+from cryptography.fernet import Fernet
+
+def decrypt(key: str, data_to_decrypt: str) -> str:
+    '''
+    here key is the decryption key and data will be encrypted data
+    '''
+    try:
+        key = Fernet(key)
+        decrypted_datas = key.decrypt(data_to_decrypt.encode())
+        return decrypted_datas.decode()
+    except:
+        return None
+
+def encrypt(key: str, data_to_encrypt: str) -> str:
+    '''
+    here key is the decryption key and data will be text which needs to be decrypted
+    '''
+    try:
+        key = Fernet(key)
+        encrypted_data = key.encrypt(data_to_encrypt.encode())
+        return encrypted_data.decode()
+    except:
+        return None
+
+def create_new_key() -> str:
+    '''
+    this section is use to create a new key
+    '''
+    key = Fernet.generate_key()
+    return key.decode()