Spaces:
Running
Running
Update get_gmail_data.py
Browse files- get_gmail_data.py +2 -1
get_gmail_data.py
CHANGED
@@ -188,6 +188,7 @@ class GmailDataExtractor:
|
|
188 |
|
189 |
def encrypt_message_id(self,message_id:str):
|
190 |
key = os.getenv('AES_KEY')
|
|
|
191 |
|
192 |
# Initialize AES cipher with the key and CBC mode
|
193 |
cipher = Cipher(algorithms.AES(key), modes.CBC(), backend=default_backend())
|
@@ -197,7 +198,7 @@ class GmailDataExtractor:
|
|
197 |
|
198 |
# Pad the message_id to be a multiple of 16 bytes (AES block size)
|
199 |
# This is necessary for AES encryption
|
200 |
-
message_id_padded =
|
201 |
|
202 |
# Encrypt the padded message_id
|
203 |
ciphertext = encryptor.update(message_id_padded) + encryptor.finalize()
|
|
|
188 |
|
189 |
def encrypt_message_id(self,message_id:str):
|
190 |
key = os.getenv('AES_KEY')
|
191 |
+
message_id_bytes = message_id.encode('utf-8')
|
192 |
|
193 |
# Initialize AES cipher with the key and CBC mode
|
194 |
cipher = Cipher(algorithms.AES(key), modes.CBC(), backend=default_backend())
|
|
|
198 |
|
199 |
# Pad the message_id to be a multiple of 16 bytes (AES block size)
|
200 |
# This is necessary for AES encryption
|
201 |
+
message_id_padded = message_id_bytes.encode().rjust(32, b'\0')
|
202 |
|
203 |
# Encrypt the padded message_id
|
204 |
ciphertext = encryptor.update(message_id_padded) + encryptor.finalize()
|