rDrama
/
rDrama
9
10
Fork 7
Code Issues Pull Requests Packages Releases Wiki Activity

always use charge_account()

pull/136/head
Aevann 2023-03-04 20:32:55 +02:00
parent eb59a6a1bc
commit 2d2a1ca28c
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
files/classes/user.py
View File

@ -226,7 +226,9 @@ class User(Base):
g.db.query(User).filter(User.id == self.id).update({ User.marseybux: User.marseybux - amount })
succeeded = True
if succeeded: g.db.flush()
if succeeded:
g.db.add(self)
g.db.flush()
return succeeded

7
files/helpers/alerts.py
View File

@ -157,8 +157,7 @@ def NOTIFY_USERS(text, v, oldtext=None):
if cost > v.coins:
abort(403, f"You need {cost} coins for this!")
g.db.query(User).update({ User.coins: User.coins + 5 })
v.coins -= cost
g.db.add(v)
v.charge_account('coins', cost)
return 'everyone'
else:
group = g.db.get(Group, i.group(2))
@ -178,9 +177,7 @@ def NOTIFY_USERS(text, v, oldtext=None):
g.db.query(User).filter(User.id.in_(members)).update({ User.coins: User.coins + mul })
v.coins -= cost
g.db.add(v)
v.charge_account('coins', cost)