rDrama/files/routes/polls.py

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/<int:option_id>")
@limiter.limit('1/second', scope=rpath)
@limiter.limit(DEFAULT_RATELIMIT)
@limiter.limit(DEFAULT_RATELIMIT, key_func=get_ID)
@is_not_permabanned
def vote_option(option_id, v):
	try:
		option_id = int(option_id)
	except:
		abort(404)
	option = g.db.get(SubmissionOption, option_id)
	if not option: abort(404)
	sub = option.post.sub

	if sub in {'furry','vampire','racist','femboy'} and not v.house.lower().startswith(sub):
		abort(403, f"You need to be a member of House {sub.capitalize()} to vote on polls in /h/{sub}")

	if option.exclusive == 2:
		if option.post.total_bet_voted(v):
			abort(403, "You can't participate in a closed bet!")
		if not v.charge_account('coins', POLL_BET_COINS):
			abort(400, f"You don't have {POLL_BET_COINS} coins!")
		g.db.add(v)
		autojanny = get_account(AUTOJANNY_ID)
		autojanny.pay_account('coins', POLL_BET_COINS)
		g.db.add(autojanny)

	if option.exclusive:
		vote = g.db.query(SubmissionOptionVote).join(SubmissionOption).filter(
			SubmissionOptionVote.user_id==v.id,
			SubmissionOptionVote.submission_id==option.parent_id,
			SubmissionOption.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)

	existing = g.db.query(SubmissionOptionVote).filter_by(option_id=option_id, user_id=v.id).one_or_none()
	if not existing:
		vote = SubmissionOptionVote(
			option_id=option_id,
			user_id=v.id,
			submission_id=option.parent_id,
		)
		g.db.add(vote)
	elif existing and not option.exclusive:
		g.db.delete(existing)

	return {"message": "Bet successful!"}

@app.get("/votes/post/option/<int:option_id>")
@limiter.limit(DEFAULT_RATELIMIT)
@limiter.limit(DEFAULT_RATELIMIT, 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(SubmissionOption, option_id)
	if not option: abort(404)
	if option.post.ghost: abort(403)
	ups = g.db.query(SubmissionOptionVote).filter_by(option_id=option_id).order_by(SubmissionOptionVote.created_utc).all()

	user_ids = [x[0] for x in g.db.query(SubmissionOptionVote.user_id).filter_by(option_id=option_id).all()]
	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 >= 3:
		total_patrons = g.db.query(User).filter(User.id.in_(user_ids), User.patron > 0).count()
	else:
		total_patrons = None

	return render_template("poll_votes.html",
						v=v,
						thing=option,
						ups=ups,
						total_ts=total_ts,
						total_patrons=total_patrons,
						)



@app.post("/vote/comment/option/<int:option_id>")
@limiter.limit('1/second', scope=rpath)
@limiter.limit(DEFAULT_RATELIMIT)
@limiter.limit(DEFAULT_RATELIMIT, key_func=get_ID)
@is_not_permabanned
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)
	sub = option.comment.post.sub
	if sub in {'furry','vampire','racist','femboy'} and not v.house.lower().startswith(sub):
		abort(403, f"You need to be a member of House {sub.capitalize()} to vote on polls in /h/{sub}")

	if option.exclusive:
		vote = g.db.query(CommentOptionVote).join(CommentOption).filter(
			CommentOptionVote.user_id==v.id,
			CommentOptionVote.comment_id==option.parent_id,
			CommentOption.exclusive==1).one_or_none()
		if vote:
			g.db.delete(vote)

	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 "", 204

@app.get("/votes/comment/option/<int:option_id>")
@limiter.limit(DEFAULT_RATELIMIT)
@limiter.limit(DEFAULT_RATELIMIT, 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.comment.ghost: 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).all()]
	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 >= 3:
		total_patrons = g.db.query(User).filter(User.id.in_(user_ids), User.patron > 0).count()
	else:
		total_patrons = None

	return render_template("poll_votes.html",
						v=v,
						thing=option,
						ups=ups,
						total_ts=total_ts,
						total_patrons=total_patrons,
						)