Infra/lemur
5
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

minor fixes and downgrading requests (#535)

Browse Source
This commit is contained in:
kevgliss
2016-11-28 16:50:26 -08:00
committed by GitHub
parent 727bc87ede
commit bfc80f982c
3 changed files with 17 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lemur/common/schema.py
View File

@ -118,12 +118,17 @@ def unwrap_pagination(data, output_schema):
if isinstance(data, dict):
if data.get('total') == 0:
return data
else:
marshaled_data = {'total': data['total']}
marshaled_data['items'] = output_schema.dump(data['items'], many=True).data
return marshaled_data
else:
return output_schema.dump(data).data
marshaled_data = {'total': data['total']}
marshaled_data['items'] = output_schema.dump(data['items'], many=True).data
return marshaled_data
elif isinstance(data, list):
marshaled_data = {'total': len(data)}
marshaled_data['items'] = output_schema.dump(data, many=True).data
return marshaled_data
return output_schema.dump(data).data
def validate_schema(input_schema, output_schema):