def setUp(self):
"""
Define test variables and initialize app
Has to be in this awkward camelCase to overload method in TestCase
"""
self.app = create_app()
self.client = self.app.test_client
self.database_path = os.environ['HEROKU_POSTGRESQL_AQUA_URL']
setup_db(self.app, self.database_path)
# binds the app to the current context
with self.app.app_context():
# populate tables with test values
e1 = Exercise(
name='kb swing',
equipment='kettlebell',
target='reps',
link='dummy link'
)
e2 = Exercise(
name='kb goblet squat',
equipment='kettlebell',
target='reps',
link='another dummy link here'
)
e1.insert()
e2.insert()
w = Workout(
name='test workout',
focus='legs',
repeat=False
)
w.insert()
m = WorkoutExercise(
workout_id=1,
exercise_id=1,
sets=3,
reps=8,
weight=4
)
m.insert()
def create_exercise():
try:
all_data = request.get_json()
name = all_data['name']
equipment = all_data['equipment']
target = all_data['target']
link = all_data['link']
for value in [name, equipment, target, link]:
if value is None:
abort(400)
exercise = Exercise(name=name,
equipment=equipment,
target=target,
link=link)
# could check whether we expect this to fail
# based on inputs, and abort(500) otherwise
exercise.insert()
except Exception:
abort(400)
return jsonify({'success': True, 'id': exercise.id})
print exercise elif sys.argv[1] == 'insert': if sys.argv[2] == 'school': school = School.insert(name=sys.argv[3]).execute() print "New school: " + str(School.select().where(School.id==school).get()) elif sys.argv[2] == 'batch': batch = Batch.insert(school=sys.argv[3], name=sys.argv[4]).execute() print "New batch: " + str(Batch.select().where(Batch.id==batch).get()) 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()
