diff --git a/app.py b/app.py index a575cc6..c72a78f 100644 --- a/app.py +++ b/app.py @@ -212,19 +212,17 @@ def create_app(config=Config): if request.method == 'GET': if not current_user.is_authenticated: return redirect(url_for('thelogin')) - if not current_user.is_admin: - pins = list(mongo.db.pins.find({"added_by": current_user.id})) + else: invite_code = str(uuid4()) qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': invite_code}}) print(qr_update) - return render_template('dashboard.html', pins=pins, invite_code=invite_code) - if current_user.is_admin: - users = list(mongo.db.users.find()) - pins = list(mongo.db.pins.find()) - invite_code = str(uuid4()) - qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': invite_code}}) - print(qr_update) - return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code) + if not current_user.is_admin: + pins = list(mongo.db.pins.find({"added_by": current_user.id})) + return render_template('dashboard.html', pins=pins, invite_code=invite_code) + if current_user.is_admin: + users = list(mongo.db.users.find()) + pins = list(mongo.db.pins.find()) + return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code) @app.cli.command('add_user') def add_user():