Minor cleanup

This commit is contained in:
Peter Stockings
2023-03-14 20:16:59 +11:00
parent 65a6ac8287
commit fc26a89793

40
app.py
View File

@@ -64,22 +64,6 @@ class CadenceReading(db.Model):
power = db.Column(db.Integer, nullable=False)
def render_users_and_workouts():
users = User.query.all()
users_data = []
for user in users:
workouts = get_workouts_for_user(user.id)
user_data = {
'id': user.id,
'name': user.name,
'bike_id': user.bike_id,
'workouts_count': len(workouts),
'workouts': workouts
}
users_data.append(user_data)
return render_template('users_and_workouts.html', users=users_data, bikes=Bike.query.all())
@app.route('/', methods=['GET'])
def get_workouts():
return render_users_and_workouts()
@@ -93,14 +77,6 @@ def new_workout(user_id):
@app.route('/users', methods=['GET', 'POST'])
def users():
if request.method == 'GET':
# get a list of all users in the database
users = User.query.all()
users_list = [{'id': user.id, 'name': user.name,
'workouts': len(user.workouts)} for user in users]
return jsonify(users_list), 200
elif request.method == 'POST':
# create a new user
data = request.form
name = data['name']
@@ -217,6 +193,22 @@ def update_users_bike(user_id):
return jsonify({'error': 'User or bike not found.'}), 404
def render_users_and_workouts():
users = User.query.all()
users_data = []
for user in users:
workouts = get_workouts_for_user(user.id)
user_data = {
'id': user.id,
'name': user.name,
'bike_id': user.bike_id,
'workouts_count': len(workouts),
'workouts': workouts
}
users_data.append(user_data)
return render_template('users_and_workouts.html', users=users_data, bikes=Bike.query.all())
def get_workouts_for_user(user_id):
workouts_data = []
workouts = Workout.query.filter_by(user_id=user_id).order_by(