forked from MarseyWorld/MarseyWorld
135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
from files.__main__ import app
|
|
from files.classes.hats import *
|
|
from files.helpers.alerts import *
|
|
from files.helpers.wrappers import *
|
|
from files.helpers.const import *
|
|
from files.helpers.actions import *
|
|
from flask import g
|
|
|
|
@app.get("/hats")
|
|
@auth_required
|
|
@feature_required('HATS')
|
|
def hats(v):
|
|
owned_hat_ids = [x.hat_id for x in v.owned_hats]
|
|
|
|
if request.values.get("sort") == 'author_asc':
|
|
hats = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.submitter_id == None).order_by(User.username).all()
|
|
elif request.values.get("sort") == 'author_desc':
|
|
hats = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.submitter_id == None).order_by(User.username.desc()).all()
|
|
else:
|
|
if v.equipped_hat_ids:
|
|
equipped = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.submitter_id == None, HatDef.id.in_(owned_hat_ids), HatDef.id.in_(v.equipped_hat_ids)).order_by(HatDef.price, HatDef.name).all()
|
|
not_equipped = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.submitter_id == None, HatDef.id.in_(owned_hat_ids), HatDef.id.notin_(v.equipped_hat_ids)).order_by(HatDef.price, HatDef.name).all()
|
|
owned = equipped + not_equipped
|
|
else:
|
|
owned = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.submitter_id == None, HatDef.id.in_(owned_hat_ids)).order_by(HatDef.price, HatDef.name).all()
|
|
|
|
not_owned = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.submitter_id == None, HatDef.id.notin_(owned_hat_ids)).order_by(HatDef.price, HatDef.name).all()
|
|
hats = owned + not_owned
|
|
|
|
sales = g.db.query(func.sum(User.coins_spent_on_hats)).scalar()
|
|
num_of_hats = g.db.query(HatDef).filter(HatDef.submitter_id == None).count()
|
|
return render_template("hats.html", owned_hat_ids=owned_hat_ids, hats=hats, v=v, sales=sales, num_of_hats=num_of_hats)
|
|
|
|
@app.post("/buy_hat/<hat_id>")
|
|
@auth_required
|
|
@feature_required('HATS')
|
|
def buy_hat(v, hat_id):
|
|
try: hat_id = int(hat_id)
|
|
except: return {"error": "Hat not found!"}, 400
|
|
|
|
hat = g.db.query(HatDef).filter_by(submitter_id=None, id=hat_id).one_or_none()
|
|
if not hat: return {"error": "Hat not found!"}, 400
|
|
|
|
existing = g.db.query(Hat).filter_by(user_id=v.id, hat_id=hat.id).one_or_none()
|
|
if existing: return {"error": "You already own this hat!"}, 400
|
|
|
|
if request.values.get("mb"):
|
|
charged = v.charge_account('procoins', hat.price)
|
|
if not charged:
|
|
return {"error": "Not enough marseybux."}, 400
|
|
|
|
hat.author.procoins += hat.price * 0.1
|
|
currency = "marseybux"
|
|
else:
|
|
charged = v.charge_account('coins', hat.price)
|
|
if not charged:
|
|
return {"error": "Not enough coins."}, 400
|
|
|
|
v.coins_spent_on_hats += hat.price
|
|
hat.author.coins += hat.price * 0.1
|
|
currency = "coins"
|
|
|
|
new_hat = Hat(user_id=v.id, hat_id=hat.id)
|
|
g.db.add(new_hat)
|
|
|
|
g.db.add(v)
|
|
g.db.add(hat.author)
|
|
|
|
send_repeatable_notification(
|
|
hat.author.id,
|
|
f":marseycapitalistmanlet: @{v.username} has just bought `{hat.name}`, you have received your 10% cut ({int(hat.price * 0.1)} {currency}) :!marseycapitalistmanlet:"
|
|
)
|
|
|
|
if v.num_of_owned_hats >= 250:
|
|
badge_grant(user=v, badge_id=154)
|
|
elif v.num_of_owned_hats >= 100:
|
|
badge_grant(user=v, badge_id=153)
|
|
elif v.num_of_owned_hats >= 25:
|
|
badge_grant(user=v, badge_id=152)
|
|
|
|
return {"message": f"'{hat.name}' bought!"}
|
|
|
|
|
|
@app.post("/equip_hat/<hat_id>")
|
|
@auth_required
|
|
@feature_required('HATS')
|
|
def equip_hat(v, hat_id):
|
|
try: hat_id = int(hat_id)
|
|
except: return {"error": "Hat not found!"}, 400
|
|
|
|
hat = g.db.query(Hat).filter_by(hat_id=hat_id, user_id=v.id).one_or_none()
|
|
if not hat: return {"error": "You don't own this hat!"}, 400
|
|
|
|
hat.equipped = True
|
|
g.db.add(hat)
|
|
|
|
return {"message": f"'{hat.name}' equipped!"}
|
|
|
|
@app.post("/unequip_hat/<hat_id>")
|
|
@auth_required
|
|
@feature_required('HATS')
|
|
def unequip_hat(v, hat_id):
|
|
try: hat_id = int(hat_id)
|
|
except: return {"error": "Hat not found!"}, 400
|
|
|
|
hat = g.db.query(Hat).filter_by(hat_id=hat_id, user_id=v.id).one_or_none()
|
|
if not hat: return {"error": "You don't own this hat!"}, 400
|
|
|
|
hat.equipped = False
|
|
g.db.add(hat)
|
|
|
|
return {"message": f"'{hat.name}' unequipped!"}
|
|
|
|
@app.get("/hat_owners/<hat_id>")
|
|
@auth_required
|
|
def hat_owners(v, hat_id):
|
|
|
|
try: hat_id = int(hat_id)
|
|
except: abort(400)
|
|
|
|
try: page = int(request.values.get("page", 1))
|
|
except: page = 1
|
|
|
|
users = [x[1] for x in g.db.query(Hat, User).join(Hat.owners).filter(Hat.hat_id == hat_id).offset(25 * (page - 1)).limit(26).all()]
|
|
|
|
next_exists = (len(users) > 25)
|
|
users = users[:25]
|
|
|
|
return render_template("user_cards.html",
|
|
v=v,
|
|
users=users,
|
|
next_exists=next_exists,
|
|
page=page,
|
|
)
|