Browse Source

HUE-3519 [useradmin] Check if user object exists before augmenting in rewrite_user

Jenny Kim 9 years ago
parent
commit
8e6583e
1 changed files with 6 additions and 3 deletions
  1. 6 3
      desktop/core/src/desktop/auth/backend.py

+ 6 - 3
desktop/core/src/desktop/auth/backend.py

@@ -83,9 +83,12 @@ def rewrite_user(user):
   We currently only re-write specific attributes,
   though this could be generalized.
   """
-  augment = get_user_augmentation_class()(user)
-  for attr in ("get_groups", "get_home_directory", "has_hue_permission"):
-    setattr(user, attr, getattr(augment, attr))
+  if user is None:
+    LOG.warn('Failed to rewrite user, user is None.')
+  else:
+    augment = get_user_augmentation_class()(user)
+    for attr in ("get_groups", "get_home_directory", "has_hue_permission"):
+      setattr(user, attr, getattr(augment, attr))
   return user
 
 class DefaultUserAugmentor(object):