From ec42526cdaaad980d820ccd8ce9337b2e61b9072 Mon Sep 17 00:00:00 2001 From: femalves Date: Fri, 29 Aug 2025 16:13:37 -0400 Subject: [PATCH] adding payload.values() check --- vault_service/views/user.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/vault_service/views/user.py b/vault_service/views/user.py index 8929899..23518bb 100644 --- a/vault_service/views/user.py +++ b/vault_service/views/user.py @@ -172,12 +172,12 @@ def store_data(): library_server = payload.pop('link_server', None) library = None - - # limit both number of keys and length of value to keep db clean - if len(max(list(payload.values()), key=len)) > current_app.config['MAX_ALLOWED_JSON_SIZE']: - return json.dumps({'msg': 'You have exceeded the allowed storage limit (length of values), no data was saved'}), 400 - if len(list(payload.keys())) > current_app.config['MAX_ALLOWED_JSON_KEYS']: - return json.dumps({'msg': 'You have exceeded the allowed storage limit (number of keys), no data was saved'}), 400 + if payload.values(): + # limit both number of keys and length of value to keep db clean + if len(max(list(payload.values()), key=len)) > current_app.config['MAX_ALLOWED_JSON_SIZE']: + return json.dumps({'msg': 'You have exceeded the allowed storage limit (length of values), no data was saved'}), 400 + if len(list(payload.keys())) > current_app.config['MAX_ALLOWED_JSON_KEYS']: + return json.dumps({'msg': 'You have exceeded the allowed storage limit (number of keys), no data was saved'}), 400 with current_app.session_scope() as session: user = session.query(User).filter_by(id=user_id).with_for_update(of=User).first()