54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from pecan import hooks
|
|
|
|
|
|
class UserFilterHook(hooks.PecanHook):
|
|
|
|
# we do this at the very end to ensure user-defined filters
|
|
# don't impact things like pagination and notification hooks
|
|
priority = 90
|
|
|
|
def after(self, state):
|
|
user_fields = state.request.params.getall('fields')
|
|
if not user_fields:
|
|
return
|
|
try:
|
|
data = state.response.json
|
|
except ValueError:
|
|
return
|
|
resource = state.request.context.get('resource')
|
|
collection = state.request.context.get('collection')
|
|
if collection not in data and resource not in data:
|
|
return
|
|
is_single = resource in data
|
|
key = resource if resource in data else collection
|
|
if is_single:
|
|
data[key] = self._filter_item(
|
|
state.response.json[key], user_fields)
|
|
else:
|
|
data[key] = [
|
|
self._filter_item(i, user_fields)
|
|
for i in state.response.json[key]
|
|
]
|
|
state.response.json = data
|
|
|
|
def _filter_item(self, item, fields):
|
|
return {
|
|
field: value
|
|
for field, value in item.items()
|
|
if field in fields
|
|
}
|