def change_batch(argv): '''Takes the system arguments as parameter. argv[2] is id of the student to change, argv[3] is the batch to change the student to. Keyword arguments: argv -- An array of command line arguments passed to the program. ''' student_found = False batch_found = False for student in Student.select(): if str(student.id) == str(argv[2]): student_found = True for batch in Batch.select(): if str(batch.id) == str(argv[3]): batch_found = True if student_found is False: print "Student not found" return if batch_found is False: print "Batch not found" return for student in Student.select(): if str(student.id) == str(argv[2]): if str(student.batch.id) == str(argv[3]): student_found = True print student, print "already in " + str(student.batch) else: '''If the student is not already in the batch, then reassign.''' tmp_id = student.id tmp_age = student.age tmp_first_name = student.first_name tmp_last_name = student.last_name d = Student.delete().where(Student.id == tmp_id) d.execute() new = Student.create(id=tmp_id, batch=argv[3], age=tmp_age, last_name=tmp_last_name, first_name=tmp_first_name) print student, print "has been move to", print new.batch
exercise = Exercise.insert(student=sys.argv[3], subject=sys.argv[4],note=sys.argv[5]).execute() print "New exercise: " + str(Exercise.select().where(Exercise.id==exercise).get()) elif sys.argv[1] == 'delete': if sys.argv[2] == 'school': if School.select().where(School.id==sys.argv[3]).count() == 0: print "Nothing to delete" else: print_school = "Delete: " + str(School.select().where(School.id==sys.argv[3]).get()) school = School.delete().where(School.id == sys.argv[3]).execute() print print_school elif sys.argv[2] == 'student': if Student.select().where(Student.id==sys.argv[3]).count() == 0: print "Nothing to delete" else: print_student = "Delete: " + str(Student.select().where(Student.id==sys.argv[3]).get()) student = Student.delete().where(Student.id == sys.argv[3]).execute() print print_student elif sys.argv[2] == 'batch': if Batch.select().where(Batch.id==sys.argv[3]).count() == 0: print "Nothing to delete" else: print_batch = "Delete: " + str(Batch.select().where(Batch.id==sys.argv[3]).get()) batch = Batch.delete().where(Batch.id == sys.argv[3]).execute() print print_batch elif sys.argv[2] == 'exercise': if Exercise.select().where(Exercise.id==sys.argv[3]).count() == 0: print "Nothing to delete" else: print_exercise = "Delete: " + str(Exercise.select().where(Exercise.id==sys.argv[3]).get()) exercise = Exercise.delete().where(Exercise.id == sys.argv[3]).execute() print print_exercise