skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fd

Browse source
This commit is contained in:
Aevann1 2022-01-02 02:05:22 +02:00
parent 84a101b022
commit 98738954a5
7 changed files with 54 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

16
files/routes/login.py
View file

@ -300,6 +300,22 @@ def sign_up_post(v):
ref_id = int(request.values.get("referred_by", 0))
if ref_id:
ref_user = g.db.query(User).filter_by(id=ref_id).one_or_none()
if ref_user:
if ref_user.referral_count > 0:
new_badge = Badge(user_id=ref_user.id, badge_id=10)
g.db.add(new_badge)
if ref_user.referral_count > 9:
new_badge = Badge(user_id=ref_user.id, badge_id=11)
g.db.add(new_badge)
if ref_user.referral_count > 99:
new_badge = Badge(user_id=ref_user.id, badge_id=12)
g.db.add(new_badge)
id_1 = g.db.query(User.id).filter_by(id=7).count()
users_count = g.db.query(User.id).count()
if id_1 == 0 and users_count == 7: admin_level=3