1
0
forked from orson/bachemap

sanity check

This commit is contained in:
Orson 2024-09-12 16:13:15 -06:00
parent d6b13864a4
commit 78a98b85fb

50
app.py
View File

@ -132,6 +132,11 @@ def create_app(config=Config):
"is_admin": False "is_admin": False
} }
new_user_id = mongo.db.users.insert_one(new_user_data).inserted_id new_user_id = mongo.db.users.insert_one(new_user_data).inserted_id
str_id = str(new_user_id)
mongo.db.users.update_one(
{'_id': new_user_id},
{'$set': {'str_id': str_id}}
)
invite_link = url_for('registrame', referral_code=new_user_data['referral_code'], _external=True) invite_link = url_for('registrame', referral_code=new_user_data['referral_code'], _external=True)
login_user(load_user(new_user_id)) login_user(load_user(new_user_id))
return redirect(url_for('index')) return redirect(url_for('index'))
@ -220,8 +225,6 @@ def create_app(config=Config):
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', users=users, pins=pins, invite_code=invite_code) return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code)
#if request.method == 'POST':
# pass
@app.cli.command('add_user') @app.cli.command('add_user')
def add_user(): def add_user():
@ -241,40 +244,21 @@ def create_app(config=Config):
@app.route('/leaderboard') @app.route('/leaderboard')
def leaderboard(): def leaderboard():
leaders = mongo.db.pins.aggregate([ leaders = mongo.db.pins.aggregate([
{ {"$group": {"_id": "$added_by", "count": {"$sum": 1}}},
"$group": { {"$sort": {"count": -1}},
"_id":"$added_by", {"$limit": 10},
"count": {"$sum":1} {"$lookup": {
} "from": "users",
}, "localField": "_id",
{ "foreignField": "_id",
"$sort": {"count":-1} "as": "user_info"
}, }},
{ {"$unwind": "$user_info"},
"$limit":10 {"$project": {"_id": 0, "username": "$user_info.username", "count": 1}}
},
{
"$lookup": {
"from": "users",
"localField": "added_by",
"foreignField": "added_by",
"as": "user_info"
}
},
{
"$unwind": "$user_info"
},
{
"$project": {
"_id":0,
"added_by": "$_id",
"count":1,
"username": "$user_info.username"
}
}
]) ])
#print(list(leaders)) #print(list(leaders))
cleaned_leaders=list() cleaned_leaders=list()
print(cleaned_leaders)
for leader in list(leaders): for leader in list(leaders):
leader["username"] = leader["username"][0]+"***"+leader["username"][-1] leader["username"] = leader["username"][0]+"***"+leader["username"][-1]
cleaned_leaders.append(leader) cleaned_leaders.append(leader)