master
Aevann1 2021-12-15 00:06:54 +02:00
parent 4a7a39928c
commit 252c9303d8
2 changed files with 13 additions and 2 deletions

View File

@ -68,6 +68,7 @@ class User(Base):
fail2_utc = Column(Integer, default=0) fail2_utc = Column(Integer, default=0)
admin_level = Column(Integer, default=0) admin_level = Column(Integer, default=0)
coins_spent = Column(Integer, default=0) coins_spent = Column(Integer, default=0)
lootboxes_bought = Column(Integer, default=0)
agendaposter = Column(Boolean, default=False) agendaposter = Column(Boolean, default=False)
agendaposter_expires_utc = Column(Integer, default=0) agendaposter_expires_utc = Column(Integer, default=0)
changelogsub = Column(Boolean, default=False) changelogsub = Column(Boolean, default=False)

View File

@ -128,11 +128,9 @@ def buy(v, award):
g.db.add(v)
g.db.flush() g.db.flush()
if award == "lootbox": if award == "lootbox":
send_notification(995, f"@{v.username} bought a lootbox!")
for i in [1,2,3,4,5]: for i in [1,2,3,4,5]:
thing = g.db.query(AwardRelationship).order_by(AwardRelationship.id.desc()).first().id thing = g.db.query(AwardRelationship).order_by(AwardRelationship.id.desc()).first().id
thing += 1 thing += 1
@ -140,12 +138,24 @@ def buy(v, award):
award = AwardRelationship(id=thing, user_id=v.id, kind=award) award = AwardRelationship(id=thing, user_id=v.id, kind=award)
g.db.add(award) g.db.add(award)
g.db.flush() g.db.flush()
v.lootboxes_bought += 1
if v.lootboxes_bought == 10 and not v.has_badge(76):
new_badge = Badge(badge_id=76, user_id=v.id)
g.db.add(new_badge)
elif v.lootboxes_bought == 50 and not v.has_badge(77):
new_badge = Badge(badge_id=77, user_id=v.id)
g.db.add(new_badge)
elif v.lootboxes_bought == 150 and not v.has_badge(78):
new_badge = Badge(badge_id=78, user_id=v.id)
g.db.add(new_badge)
else: else:
thing = g.db.query(AwardRelationship).order_by(AwardRelationship.id.desc()).first().id thing = g.db.query(AwardRelationship).order_by(AwardRelationship.id.desc()).first().id
thing += 1 thing += 1
award = AwardRelationship(id=thing, user_id=v.id, kind=award) award = AwardRelationship(id=thing, user_id=v.id, kind=award)
g.db.add(award) g.db.add(award)
g.db.add(v)
g.db.commit() g.db.commit()
return {"message": "Award bought!"} return {"message": "Award bought!"}