from files.classes import * from files.helpers.config.const import * from files.helpers.get import * from files.routes.wrappers import * from files.__main__ import app @app.post("/vote/post/option/") @limiter.limit('1/second', scope=rpath) @limiter.limit('1/second', scope=rpath, key_func=get_ID) @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400) @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID) @auth_required def vote_option(option_id, v): try: option_id = int(option_id) except: abort(404) option = g.db.get(PostOption, option_id) if not option: abort(404) hole = option.parent.hole if hole in {'furry','vampire','racist','femboy','edgy'} and not v.house.lower().startswith(hole): abort(403, f"You need to be a member of House {hole.capitalize()} to vote on polls in /h/{hole}") if option.exclusive == 2: if option.parent.total_bet_voted(v): abort(403, "You can't participate in a closed bet!") if not v.charge_account('coins/marseybux', POLL_BET_COINS)[0]: abort(400, f"You don't have {POLL_BET_COINS} coins or marseybux!") g.db.add(v) if option.exclusive: vote = g.db.query(PostOptionVote).join(PostOption).filter( PostOptionVote.user_id==v.id, PostOptionVote.post_id==option.parent_id, PostOption.exclusive==option.exclusive).all() if vote: if option.exclusive == 2: abort(400, "You already voted on this bet!") for x in vote: g.db.delete(x) g.db.flush() existing = g.db.query(PostOptionVote).filter_by(option_id=option_id, user_id=v.id).one_or_none() if not existing: vote = PostOptionVote( option_id=option_id, user_id=v.id, post_id=option.parent_id, ) g.db.add(vote) elif existing and not option.exclusive: g.db.delete(existing) return {"message": "Bet successful!"} @app.post("/vote/comment/option/") @limiter.limit('1/second', scope=rpath) @limiter.limit('1/second', scope=rpath, key_func=get_ID) @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400) @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID) @auth_required def vote_option_comment(option_id, v): try: option_id = int(option_id) except: abort(404) option = g.db.get(CommentOption, option_id) if not option: abort(404) if option.parent.parent_post: hole = option.parent.post.hole else: hole = None if hole in {'furry','vampire','racist','femboy','edgy'} and not v.house.lower().startswith(hole): abort(403, f"You need to be a member of House {hole.capitalize()} to vote on polls in /h/{hole}") if option.exclusive == 2: if option.parent.total_bet_voted(v): abort(403, "You can't participate in a closed bet!") if not v.charge_account('coins/marseybux', POLL_BET_COINS)[0]: abort(400, f"You don't have {POLL_BET_COINS} coins or marseybux!") g.db.add(v) if option.exclusive: vote = g.db.query(CommentOptionVote).join(CommentOption).filter( CommentOptionVote.user_id==v.id, CommentOptionVote.comment_id==option.parent_id, CommentOption.exclusive==option.exclusive).all() if vote: if option.exclusive == 2: abort(400, "You already voted on this bet!") for x in vote: g.db.delete(x) g.db.flush() existing = g.db.query(CommentOptionVote).filter_by(option_id=option_id, user_id=v.id).one_or_none() if not existing: vote = CommentOptionVote( option_id=option_id, user_id=v.id, comment_id=option.parent_id, ) g.db.add(vote) elif existing: g.db.delete(existing) return {"message": "Bet successful!"} @app.get("/votes/post/option/") @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400) @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID) @auth_required def option_votes(option_id, v): try: option_id = int(option_id) except: abort(404) option = g.db.get(PostOption, option_id) if not option: abort(404) if option.parent.ghost and v.admin_level < PERMS['SEE_GHOST_VOTES']: abort(403) ups = g.db.query(PostOptionVote).filter_by(option_id=option_id).order_by(PostOptionVote.created_utc).all() user_ids = [x[0] for x in g.db.query(PostOptionVote.user_id).filter_by(option_id=option_id)] total_ts = g.db.query(func.sum(User.truescore)).filter(User.id.in_(user_ids)).scalar() total_ts = format(total_ts, ",") if total_ts else '0' if v.admin_level >= PERMS['VIEW_PATRONS']: patrons = [x[0] for x in g.db.query(User.patron).filter(User.id.in_(user_ids), User.patron > 1)] total_patrons = len(patrons) total_money = 0 for tier in patrons: total_money += TIER_TO_MONEY[tier] else: total_patrons = None total_money = 0 return render_template("poll_votes.html", v=v, option=option, ups=ups, total_ts=total_ts, total_patrons=total_patrons, total_money=total_money, ) @app.get("/votes/comment/option/") @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400) @limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID) @auth_required def option_votes_comment(option_id, v): try: option_id = int(option_id) except: abort(404) option = g.db.get(CommentOption, option_id) if not option: abort(404) if option.parent.ghost and v.admin_level < PERMS['SEE_GHOST_VOTES']: abort(403) ups = g.db.query(CommentOptionVote).filter_by(option_id=option_id).order_by(CommentOptionVote.created_utc).all() user_ids = [x[0] for x in g.db.query(CommentOptionVote.user_id).filter_by(option_id=option_id)] total_ts = g.db.query(func.sum(User.truescore)).filter(User.id.in_(user_ids)).scalar() total_ts = format(total_ts, ",") if total_ts else '0' if v.admin_level >= PERMS['VIEW_PATRONS']: patrons = [x[0] for x in g.db.query(User.patron).filter(User.id.in_(user_ids), User.patron > 1)] total_patrons = len(patrons) total_money = 0 for tier in patrons: total_money += TIER_TO_MONEY[tier] else: total_patrons = None total_money = 0 return render_template("poll_votes.html", v=v, option=option, ups=ups, total_ts=total_ts, total_patrons=total_patrons, total_money=total_money, )