elif sys.argv[2] == 'student': if len(sys.argv) == 6: student = Student.insert(batch=sys.argv[3], age=sys.argv[4], last_name='', first_name=sys.argv[5]).execute() else: student = Student.insert(batch=sys.argv[3], age=sys.argv[4], last_name=sys.argv[5], first_name=sys.argv[6]).execute() print "New student: " + str(Student.select().where(Student.id==student).get()) elif sys.argv[2] == 'exercise': 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