1
0
forked from orson/bachemap

fixed uuid/qr gen

This commit is contained in:
Orson 2024-09-06 13:16:25 -06:00
parent 65c0d401cb
commit 0c9d6d094a

10
app.py
View File

@ -202,15 +202,17 @@ def create_app(config=Config):
return redirect(url_for('thelogin')) return redirect(url_for('thelogin'))
if not current_user.is_admin: if not current_user.is_admin:
pins = list(mongo.db.pins.find({"added_by": current_user.id})) 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) 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: if current_user.is_admin:
users = list(mongo.db.users.find()) users = list(mongo.db.users.find())
pins = list(mongo.db.pins.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) 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': #if request.method == 'POST':
# pass # pass