Handle users with no f2p attribute.

reddit-archive · Apr 17, 2013 · 8cce85a · 8cce85a
1 parent f624c97
commit 8cce85a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/reddit_f2p/steam.py b/reddit_f2p/steam.py
@@ -57,7 +57,7 @@ def make_post_login_url():
     @validate(VUser(),
               error=nop("error"))
     def GET_start(self, error):
-        f2p_status = getattr(c.user, "f2p")
+        f2p_status = getattr(c.user, "f2p", None)
         error = bool(error)
 
         if f2p_status == "participated":
@@ -71,7 +71,7 @@ def GET_start(self, error):
     @validate(VUser(),
               VModhash())
     def POST_auth(self):
-        if getattr(c.user, "f2p") != "participated":
+        if getattr(c.user, "f2p", None) != "participated":
             abort(403)
 
         session = {}
@@ -86,7 +86,7 @@ def POST_auth(self):
 
     @validate(VUser())
     def GET_postlogin(self):
-        if getattr(c.user, "f2p") != "participated":
+        if getattr(c.user, "f2p", None) != "participated":
             return redirect_to("/f2p/steam")
 
         session = g.f2pcache.get("steam_session_%d" % c.user._id)