From 0c9d6d094ac3f8b7cab7f0e35eb825cc81b47dfd Mon Sep 17 00:00:00 2001 From: orson Date: Fri, 6 Sep 2024 13:16:25 -0600 Subject: [PATCH] fixed uuid/qr gen --- app.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/app.py b/app.py index 4495099..54b8858 100644 --- a/app.py +++ b/app.py @@ -202,15 +202,17 @@ def create_app(config=Config): return redirect(url_for('thelogin')) if not current_user.is_admin: pins = list(mongo.db.pins.find({"added_by": current_user.id})) - qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': str(uuid4())}}) + 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) + 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()) - qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': str(uuid4())}}) + 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) + return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code) #if request.method == 'POST': # pass