def test_encrypt_dict_keep_type():
encrypted_dict = security.encrypt({'token': 123123123},
secure_keys=['token'])
assert encrypted_dict['token']
decrypted_dict = security.decrypt(encrypted_dict, secure_keys=['token'])
assert decrypted_dict['token'] == 123123123
def get_all_settings() -> Dict[str, Dict[str, dict]]:
"""
Get all settings.
"""
all_data = configstore.get_all_data()
if settings.ENCRYPT_ENABLED:
return security.decrypt(data=all_data,
secure_keys=settings.SECURE_KEYS)
return all_data
def test_encrypt_dict_inner_list():
encrypted_dict = security.encrypt({'tokens': [1, 2, 3]},
secure_keys=['tokens'])
assert security.is_encrypted(encrypted_dict['tokens'][0])
assert security.is_encrypted(encrypted_dict['tokens'][1])
assert security.is_encrypted(encrypted_dict['tokens'][2])
decrypted_dict = security.decrypt(encrypted_dict, secure_keys=['tokens'])
assert decrypted_dict['tokens'] == [1, 2, 3]
def test_encrypt_dict_nested():
encrypted_dict = security.encrypt(
{'db': {
'user': '******',
'pass': '******',
}}, secure_keys=['db.pass'])
assert encrypted_dict['db']['user'] == 'admin'
assert security.is_encrypted(encrypted_dict['db']['pass'])
decrypted_dict = security.decrypt(encrypted_dict, secure_keys=['db.pass'])
assert decrypted_dict['db']['user'] == 'admin'
assert decrypted_dict['db']['pass'] == 'secret'