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

fds

Browse source
This commit is contained in:
Aevann1 2021-10-23 21:15:50 +02:00
parent 4b30fb0afd
commit 8a8849948b
3 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
files/routes/login.py
View file

@ -306,6 +306,7 @@ def sign_up_post(v):
if ref_id:
ref_user = g.db.query(User).options(
lazyload('*')).filter_by(id=ref_id).first()
if ref_user:
badge_types = g.db.query(BadgeDef).options(lazyload('*')).filter(BadgeDef.qualification_expr.isnot(None)).all()
for badge in badge_types:
@ -313,6 +314,9 @@ def sign_up_post(v):
if not ref_user.has_badge(badge.id):
new_badge = Badge(user_id=ref_user.id, badge_id=badge.id)
g.db.add(new_badge)
else:
bad_badge = ref_user.has_badge(badge.id)
if bad_badge: g.db.delete(bad_badge)
g.db.add(ref_user)