def dehydrate(self, bundle):
"""
Make sur the user does not see the fields he has no
right to.
:param bundle: the bundle to trim
"""
if not hasattr(bundle, "request") or not models.has_user_related(bundle.request.user):
bundle.data = pick(bundle.data, *self.FIELD_ACCESS_NOT_LOGGED)
else:
user = models.get_user_related(bundle.request.user)
if not user.is_premium:
bundle.data = without(bundle.data, *self.FIELD_DENIED_BASIC)
return bundle
def test_without(self):
_in = {"a": "ok", "b": "ko"}
out = {"a": "ok"}
self.assertEqual(python.without(_in, "b", "c"), out)
