46 lines
927 B
Python
46 lines
927 B
Python
|
#!/usr/bin/python2.4
|
||
|
|
||
|
import os
|
||
|
import os.path
|
||
|
import sys
|
||
|
from controller.Database import Database
|
||
|
from model.Notebook import Notebook
|
||
|
from model.User import User
|
||
|
|
||
|
|
||
|
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 = rate_plan
|
||
|
|
||
|
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 )
|
||
|
|
||
|
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 )
|