royrga/bachemap
1
0
Fork 0
forked from orson/bachemap
Code Pull Requests Activity

simplified invite link gen

Browse Source
This commit is contained in:
Orson 2024-09-12 16:18:25 -06:00
parent 78a98b85fb
commit 61f970c56d
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app.py
View File

@ -212,18 +212,16 @@ 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)
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())
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)
@app.cli.command('add_user')