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

18
app.py
View File

@ -212,19 +212,17 @@ def create_app(config=Config):
if request.method == 'GET': if request.method == 'GET':
if not current_user.is_authenticated: if not current_user.is_authenticated:
return redirect(url_for('thelogin')) return redirect(url_for('thelogin'))
if not current_user.is_admin: else:
pins = list(mongo.db.pins.find({"added_by": current_user.id}))
invite_code = str(uuid4()) invite_code = str(uuid4())
qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': invite_code}}) 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, invite_code=invite_code) if not current_user.is_admin:
if current_user.is_admin: pins = list(mongo.db.pins.find({"added_by": current_user.id}))
users = list(mongo.db.users.find()) return render_template('dashboard.html', pins=pins, invite_code=invite_code)
pins = list(mongo.db.pins.find()) if current_user.is_admin:
invite_code = str(uuid4()) users = list(mongo.db.users.find())
qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': invite_code}}) pins = list(mongo.db.pins.find())
print(qr_update) return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code)
return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code)
@app.cli.command('add_user') @app.cli.command('add_user')
def add_user(): def add_user():