瀏覽代碼

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

Jenny Kim 9 年之前
父節點
當前提交
8e6583e569
共有 1 個文件被更改,包括 6 次插入3 次删除
  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):