Merge pull request #49 from udacity/login_logout_links

[osqa.git] / forum / authentication / __init__.py

diff --git a/forum/authentication/__init__.py b/forum/authentication/__init__.py
index 8dbf825a525c2ab49c17a38834961ac6135706ce..79c47a8dd2e05908b96e2e7ea438a2cf72372576 100644 (file)
--- a/forum/authentication/__init__.py
+++ b/forum/authentication/__init__.py
@@ -8,7 +8,8 @@ class ConsumerAndContext:
         self.id = id
         self._consumer = consumer
 
-        context.id = id
+        if context:
+            context.id = id
         self.context = context
 
     @property
@@ -27,7 +28,7 @@ contexts = dict([
         ])
 
 AUTH_PROVIDERS = dict([
-            (name, ConsumerAndContext(name, consumers[name], contexts[name])) for name in consumers.keys()
-            if name in contexts
+            (name, ConsumerAndContext(name, consumers[name], contexts.get(name, None))) for name in consumers.keys()
         ])
 
+