open.source/whoogle-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make encrypted string shorter

Browse Source
This commit is contained in:
jacr13 2022-09-08 01:14:12 +02:00
parent 27e16e5e2c
commit 5fa4272484
1 changed files with 20 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
app/models/config.py
View File

@ -3,10 +3,11 @@ from app.utils.misc import read_config_bool
from flask import current_app
import os
import re
from base64 import urlsafe_b64encode
from base64 import urlsafe_b64encode, urlsafe_b64decode
import pickle
from cryptography.fernet import Fernet
import hashlib
import brotli
class Config:
@ -138,9 +139,8 @@ class Config:
if param_val == 'off':
param_val = False
elif isinstance(param_val, bool):
pass
elif param_val.isdigit():
elif isinstance(param_val, str):
if param_val.isdigit():
param_val = int(param_val)
self[param_key] = param_val
@ -168,15 +168,22 @@ class Config:
def _encode_preferences(self) -> str:
if self.preferences_key == '':
return ''
encoded_preferences = pickle.dumps(self.get_attrs())
encoded_preferences = brotli.compress(pickle.dumps(self.get_attrs()))
key = self._get_fernet_key(self.preferences_key)
return Fernet(key).encrypt(encoded_preferences).decode()
encrypted_preferences = Fernet(key).encrypt(encoded_preferences)
return urlsafe_b64encode(
brotli.compress(encrypted_preferences)
).decode()
def _decode_preferences(self, preferences: str) -> dict:
try:
key = self._get_fernet_key(self.preferences_key)
config = Fernet(key).decrypt(preferences.encode())
config = pickle.loads(config)
config = Fernet(key).decrypt(
brotli.decompress(urlsafe_b64decode(preferences.encode()))
)
config = pickle.loads(brotli.decompress(config))
except Exception:
config = {}