Update Crypto.py

robocorp · Nov 17, 2023 · 63098c3 · 63098c3
1 parent 9e181a0
commit 63098c3
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/packages/main/src/RPA/Crypto.py b/packages/main/src/RPA/Crypto.py
@@ -1,7 +1,7 @@
 import base64
 from enum import Enum, auto
-import os
 from pathlib import Path
+from secrets import token_bytes
 from typing import Optional, Union
 
 from cryptography.fernet import Fernet, InvalidToken
@@ -123,8 +123,7 @@ def generate_key(
         if encryption_type == EncryptionType.FERNET:
             return Fernet.generate_key().decode("utf-8")
         elif encryption_type == EncryptionType.AES256:
-            key = os.urandom(32)
-            return base64.urlsafe_b64encode(key)
+            return self._generate_aes256_key()
         else:
             raise UnknownEncryptionTypeError
 
@@ -428,12 +427,12 @@ def decrypt_file(
 
     # Helper methods for AES256 (same as before)
     def _generate_aes256_key(self) -> bytes:
-        key = os.urandom(32)
+        key = token_bytes(32)
         return base64.urlsafe_b64encode(key)
 
     def _encrypt_aes256(self, data: bytes) -> bytes:
         backend = default_backend()
-        iv = os.urandom(12)
+        iv = token_bytes(12)
         cipher = Cipher(
             algorithms.AES(base64.urlsafe_b64decode(self._key)),
             modes.GCM(iv),