Skip to content

Commit

Permalink
use pickle module instead of secpickle
Browse files Browse the repository at this point in the history
  • Loading branch information
jaedsonpys committed Jan 2, 2023
1 parent d270260 commit c1b918e
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions cookiedb/_document.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,10 @@
# along with this program. If not, see <https://www.gnu.org/licenses/>.

import os
import json
from datetime import datetime
from typing import Union

import secpickle
import pickle
from cryptography import fernet
from secpickle import exceptions as sp_exceptions

Expand All @@ -42,14 +41,14 @@ def exists_document(self, database: str) -> bool:
return os.path.isfile(document_path)

def _encrypt(self, obj: dict) -> str:
pickle_file = secpickle.dumps(obj, self._key)
pickle_file = pickle.dumps(obj)
encrypted_data = self._fernet.encrypt(pickle_file)
return encrypted_data

def _decrypt(self, encrypted: bytes) -> dict:
decrypted_data = self._fernet.decrypt(encrypted)
try:
data = secpickle.loads(decrypted_data, self._key)
data = pickle.loads(decrypted_data)
except sp_exceptions.IntegrityUnconfirmedError:
raise exceptions.InvalidDatabaseKeyError(f'Invalid key to database')

Expand Down

0 comments on commit c1b918e

Please sign in to comment.