From d0362a3ad111214a591354274113d2a320bbf72b Mon Sep 17 00:00:00 2001
From: Dan Helfman <dan@luminotes.com>
Date: Thu, 10 Jan 2008 21:11:45 +0000
Subject: [PATCH] Upgrade page now tells you what rate plan you're at.

---
 controller/Root.py   |  7 ++++++-
 view/Upgrade_note.py | 21 +++++++++++++--------
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/controller/Root.py b/controller/Root.py
index ae691ae..a148b05 100644
--- a/controller/Root.py
+++ b/controller/Root.py
@@ -252,6 +252,11 @@ class Root( object ):
     else:
       main_notebook = None
 
+    if user_id:
+      user = self.__database.load( User, user_id )
+    else:
+      user = None
+
     https_url = self.__settings[ u"global" ].get( u"luminotes.https_url" )
     result = self.__users.current( user_id )
     result[ "notebook" ] = main_notebook
@@ -263,7 +268,7 @@ class Root( object ):
       contents = unicode( Upgrade_note(
         self.__settings[ u"global" ].get( u"luminotes.rate_plans", [] ),
         https_url,
-        user_id,
+        user,
       ) ),
       notebook_id = main_notebook.object_id,
     ) ]
diff --git a/view/Upgrade_note.py b/view/Upgrade_note.py
index 78dfc87..ebd6d0a 100644
--- a/view/Upgrade_note.py
+++ b/view/Upgrade_note.py
@@ -2,7 +2,7 @@ from Tags import Div, Span, H3, P, A, Table, Tr, Th, Td, Br, Img
 
 
 class Upgrade_note( Span ):
-  def __init__( self, rate_plans, https_url, user_id ):
+  def __init__( self, rate_plans, https_url, user ):
     MEGABYTE = 1024 * 1024
 
     Span.__init__(
@@ -17,7 +17,7 @@ class Upgrade_note( Span ):
       ),
       P(
         Table(
-          self.fee_row( rate_plans, user_id ),
+          self.fee_row( rate_plans, user ),
           Tr(
             Td( u"included storage space", class_ = u"feature_name" ),
             [ Td(
@@ -53,7 +53,7 @@ class Upgrade_note( Span ):
           border = u"1",
           id = u"upgrade_table",
         ),
-        ( not user_id ) and P(
+        ( not user ) and P(
           u"To upgrade your Luminotes account, please",
           A( u"login", href = https_url + u"/login?after_login=/upgrade", target = u"_top" ),
           u"first!",
@@ -62,6 +62,11 @@ class Upgrade_note( Span ):
         id = u"upgrade_table_area",
       ),
 
+      user and user.rate_plan > 0 and P(
+        u"You're currently subscribed to Luminotes %s." % 
+        rate_plans[ user.rate_plan ][ u"name" ].capitalize(),
+      ) or None,
+
       H3( u"share your notebook" ),
       P(
         A(
@@ -143,7 +148,7 @@ class Upgrade_note( Span ):
       ),
       P(
         Table(
-          self.fee_row( rate_plans, user_id, include_blank = False ),
+          self.fee_row( rate_plans, user, include_blank = False ),
           Tr(
             [ Td(
               plan[ u"storage_quota_bytes" ] // MEGABYTE, " MB",
@@ -152,7 +157,7 @@ class Upgrade_note( Span ):
           border = u"1",
           id = u"upgrade_table_small",
         ),
-        ( not user_id ) and P(
+        ( not user ) and P(
           u"Please",
           A( u"login", href = https_url + u"/login?after_login=/upgrade", target = u"_top" ),
           u"to upgrade your wiki!",
@@ -162,7 +167,7 @@ class Upgrade_note( Span ):
       ),
     )
 
-  def fee_row( self, rate_plans, user_id, include_blank = True ):
+  def fee_row( self, rate_plans, user, include_blank = True ):
     return Tr(
       include_blank and Th( u"&nbsp;" ) or None,
       [ Th(
@@ -174,8 +179,8 @@ class Upgrade_note( Span ):
             class_ = u"price_text",
             separator = u"",
           ),
-          user_id and plan.get( u"button" ) % user_id or None,
+          user and user.rate_plan != index and plan.get( u"button" ) % user.object_id or None,
         ) or None,
         class_ = u"plan_name",
-      ) for plan in rate_plans ],
+      ) for ( index, plan ) in enumerate( rate_plans ) ],
     )