luminotes/tools/set_plan.py

#!/usr/bin/python2.4

import os
import os.path
import sys
import cherrypy
from controller.Database import Database
from controller.Users import Users
from model.Notebook import Notebook
from model.User import User
from config import Common


class Plan_setter( object ):
  """
  Set the rate plan for a particular user.
  """
  def __init__( self, database, user_id, rate_plan ):
    self.database = database
    self.user_id = user_id
    self.rate_plan = int( rate_plan )

    rate_plans = Common.settings[ u"global" ][ u"luminotes.rate_plans" ]
    self.users = Users( database, None, None, None, None, rate_plans )

    self.set_plan()
    self.database.commit()

  def set_plan( self ):
    user = self.database.load( User, self.user_id )

    if not user:
      print "user id %s unknown" % self.user_id
      sys.exit( 1 )

    user.rate_plan = self.rate_plan
    self.database.save( user, commit = False )

    # update a user's group membership as a result of a rate plan change
    self.users.update_groups( user )

def main( args ):
  database = Database()
  ranker = Plan_setter( database, *args )


if __name__ == "__main__":
  args = sys.argv[ 1: ]

  if len( args ) != 2:
    print "usage: %s user_id rate_plan_index" % sys.argv[ 0 ]
    sys.exit( 1 )

  main( args )